diff options
Diffstat (limited to 'testing')
-rw-r--r-- | testing/test_chat_completion.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/testing/test_chat_completion.py b/testing/test_chat_completion.py index d901e697..7600e46b 100644 --- a/testing/test_chat_completion.py +++ b/testing/test_chat_completion.py @@ -7,19 +7,21 @@ import g4f, asyncio print("create:", end=" ", flush=True) for response in g4f.ChatCompletion.create( - model=g4f.models.gpt_35_turbo, - provider=g4f.Provider.GptGo, - messages=[{"role": "user", "content": "hello!"}], + model=g4f.models.gpt_4_32k_0613, + provider=g4f.Provider.Aivvm, + messages=[{"role": "user", "content": "send a bunch of emojis. i want to test something"}], + temperature=0.0, + stream=True ): print(response, end="", flush=True) print() async def run_async(): response = await g4f.ChatCompletion.create_async( - model=g4f.models.gpt_35_turbo, - provider=g4f.Provider.GptGo, + model=g4f.models.gpt_35_turbo_16k_0613, + provider=g4f.Provider.Aivvm, messages=[{"role": "user", "content": "hello!"}], ) print("create_async:", response) -asyncio.run(run_async()) +# asyncio.run(run_async()) |