summaryrefslogtreecommitdiffstats
path: root/etc/testing/test_chat_completion.py
diff options
context:
space:
mode:
Diffstat (limited to 'etc/testing/test_chat_completion.py')
-rw-r--r--etc/testing/test_chat_completion.py27
1 files changed, 27 insertions, 0 deletions
diff --git a/etc/testing/test_chat_completion.py b/etc/testing/test_chat_completion.py
new file mode 100644
index 00000000..7600e46b
--- /dev/null
+++ b/etc/testing/test_chat_completion.py
@@ -0,0 +1,27 @@
+import sys
+from pathlib import Path
+
+sys.path.append(str(Path(__file__).parent.parent))
+
+import g4f, asyncio
+
+print("create:", end=" ", flush=True)
+for response in g4f.ChatCompletion.create(
+ 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_16k_0613,
+ provider=g4f.Provider.Aivvm,
+ messages=[{"role": "user", "content": "hello!"}],
+ )
+ print("create_async:", response)
+
+# asyncio.run(run_async())