summaryrefslogtreecommitdiffstats
path: root/testing
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--testing/phind_test.py17
1 files changed, 16 insertions, 1 deletions
diff --git a/testing/phind_test.py b/testing/phind_test.py
index e3148eb6..6b78f02d 100644
--- a/testing/phind_test.py
+++ b/testing/phind_test.py
@@ -2,6 +2,7 @@ import phind
prompt = 'hello world'
+# normal completion
result = phind.Completion.create(
model = 'gpt-4',
prompt = prompt,
@@ -10,4 +11,18 @@ result = phind.Completion.create(
detailed = False,
codeContext = '') # up to 3000 chars of code
-print(result.completion.choices[0].text) \ No newline at end of file
+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-3.5',
+ 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) \ No newline at end of file