summaryrefslogtreecommitdiffstats
path: root/testing
diff options
context:
space:
mode:
Diffstat (limited to 'testing')
-rw-r--r--testing/test_chat_completion.py14
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())