summaryrefslogtreecommitdiffstats
path: root/testing/forefront_test.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--testing/forefront_test.py7
1 files changed, 3 insertions, 4 deletions
diff --git a/testing/forefront_test.py b/testing/forefront_test.py
index 4d5f4bc1..8d25ed89 100644
--- a/testing/forefront_test.py
+++ b/testing/forefront_test.py
@@ -1,10 +1,9 @@
-import forefront
+from openai_rev import forefront
# create an account
token = forefront.Account.create(logging=True)
print(token)
# get a response
-for response in forefront.StreamingCompletion.create(token=token,
- prompt='hello world', model='gpt-4'):
- print(response.completion.choices[0].text, end='')
+for response in forefront.StreamingCompletion.create(token=token, prompt='hello world', model='gpt-4'):
+ print(response.text, end='')