summaryrefslogtreecommitdiffstats
path: root/testing/test_chat_completion.py
diff options
context:
space:
mode:
Diffstat (limited to 'testing/test_chat_completion.py')
-rw-r--r--testing/test_chat_completion.py11
1 files changed, 6 insertions, 5 deletions
diff --git a/testing/test_chat_completion.py b/testing/test_chat_completion.py
index 77774a2f..7600e46b 100644
--- a/testing/test_chat_completion.py
+++ b/testing/test_chat_completion.py
@@ -7,19 +7,20 @@ import g4f, asyncio
print("create:", end=" ", flush=True)
for response in g4f.ChatCompletion.create(
- model=g4f.models.gpt_4,
- provider=g4f.Provider.Vercel,
- 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_4_32k_0613,
+ model=g4f.models.gpt_35_turbo_16k_0613,
provider=g4f.Provider.Aivvm,
messages=[{"role": "user", "content": "hello!"}],
- temperature=0.0
)
print("create_async:", response)