From 55577031d556c613daae9ed174fd6c02f3d0dcaa Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Wed, 20 Sep 2023 14:52:50 +0200 Subject: Add check_running_loop requirement Add create_async function in ChatCompletion Use SelectorEventLoop on windows --- testing/test_chat_completion.py | 23 ++++++++++++++++++----- 1 file changed, 18 insertions(+), 5 deletions(-) (limited to 'testing/test_chat_completion.py') diff --git a/testing/test_chat_completion.py b/testing/test_chat_completion.py index fbaa3169..d901e697 100644 --- a/testing/test_chat_completion.py +++ b/testing/test_chat_completion.py @@ -3,10 +3,23 @@ from pathlib import Path sys.path.append(str(Path(__file__).parent.parent)) -import g4f +import g4f, asyncio -response = g4f.ChatCompletion.create( +print("create:", end=" ", flush=True) +for response in g4f.ChatCompletion.create( model=g4f.models.gpt_35_turbo, - messages=[{"role": "user", "content": "hello, are you GPT 4?"}] -) -print(response) \ No newline at end of file + provider=g4f.Provider.GptGo, + messages=[{"role": "user", "content": "hello!"}], +): + print(response, end="", flush=True) +print() + +async def run_async(): + response = await g4f.ChatCompletion.create_async( + model=g4f.models.gpt_35_turbo, + provider=g4f.Provider.GptGo, + messages=[{"role": "user", "content": "hello!"}], + ) + print("create_async:", response) + +asyncio.run(run_async()) -- cgit v1.2.3