gpt4free/etc/testing/test_chat_completion.py

27 lines
677 B
Python

import sys
from pathlib import Path
sys.path.append(str(Path(__file__).parent.parent.parent))
import g4f, asyncio
print("create:", end=" ", flush=True)
for response in g4f.ChatCompletion.create(
model=g4f.models.default,
#provider=g4f.Provider.Bing,
messages=[{"role": "user", "content": "write a poem about a tree"}],
stream=True
):
print(response, end="", flush=True)
print()
async def run_async():
response = await g4f.ChatCompletion.create_async(
model=g4f.models.default,
#provider=g4f.Provider.Bing,
messages=[{"role": "user", "content": "hello!"}],
)
print("create_async:", response)
asyncio.run(run_async())