summaryrefslogtreecommitdiffstats
path: root/testing/phind_test.py
blob: ed8ff65ec0df745c7d2d53986f0f016e0059c8bf (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import phind

# set cf_clearance cookie ( not needed at the moment)
phind.cf_clearance = 'MDzwnr3ZWk_ap8u.iwwMR5F3WccfOkhUy_zGNDpcF3s-1682497341-0-160'
phind.user_agent = 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36'

prompt = 'hello world'

# normal completion
result = phind.Completion.create(
    model='gpt-4',
    prompt=prompt,
    results=phind.Search.create(prompt, actualSearch=False),
    # create search (set actualSearch to False to disable internet)
    creative=False,
    detailed=False,
    codeContext='')  # up to 3000 chars of code

print(result.completion.choices[0].text)

prompt = 'who won the quatar world cup'

# help needed: not getting newlines from the stream, please submit a PR if you know how to fix this
# stream completion
for result in phind.StreamingCompletion.create(
        model='gpt-4',
        prompt=prompt,
        results=phind.Search.create(prompt, actualSearch=True),
        # create search (set actualSearch to False to disable internet)
        creative=False,
        detailed=False,
        codeContext=''):  # up to 3000 chars of code

    print(result.completion.choices[0].text, end='', flush=True)