diff options
Diffstat (limited to 'testing/forefront_test.py')
| -rw-r--r-- | testing/forefront_test.py | 7 |
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='') |