gpt4free/etc/testing/test_chat_completion.py

28 lines
722 B
Python
Raw Normal View History

2023-07-28 10:07:17 +00:00
import sys
from pathlib import Path
2023-10-07 02:03:36 +00:00
sys.path.append(str(Path(__file__).parent.parent.parent))
2023-07-28 10:07:17 +00:00
import g4f, asyncio
2023-07-28 10:07:17 +00:00
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": "write a poem about a tree"}],
temperature=0.1,
2023-09-29 14:54:46 +00:00
stream=True
):
print(response, end="", flush=True)
print()
async def run_async():
response = await g4f.ChatCompletion.create_async(
2023-09-29 14:54:46 +00:00
model=g4f.models.gpt_35_turbo_16k_0613,
2023-10-07 17:10:26 +00:00
provider=g4f.Provider.GptGod,
messages=[{"role": "user", "content": "hello!"}],
)
print("create_async:", response)
# asyncio.run(run_async())