summaryrefslogblamecommitdiffstats
path: root/etc/testing/test_chat_completion.py
blob: 6b053b72384300ade28bb0c95d2976ce6715a1eb (plain) (tree)
1
2
3
4
5
6
7
8
9


                        
                                                         
 
                   
 

                                          
                             
                                
                                                                        
               





                                                     
                                 
                                    



                                                         
                        
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())