summaryrefslogtreecommitdiffstats
path: root/etc/testing/test_chat_completion.py
blob: 7600e46b03a97cb29c8d6974cedb0636e36d94ec (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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())
hhhhhh