diff options
Diffstat (limited to 'testing/test_needs_auth.py')
| -rw-r--r-- | testing/test_needs_auth.py | 96 |
1 files changed, 0 insertions, 96 deletions
diff --git a/testing/test_needs_auth.py b/testing/test_needs_auth.py deleted file mode 100644 index 26630e23..00000000 --- a/testing/test_needs_auth.py +++ /dev/null @@ -1,96 +0,0 @@ -import sys -from pathlib import Path -import asyncio - -sys.path.append(str(Path(__file__).parent.parent)) - -import g4f -from testing.log_time import log_time, log_time_async, log_time_yield - - -_providers = [ - g4f.Provider.H2o, - g4f.Provider.You, - g4f.Provider.HuggingChat, - g4f.Provider.OpenAssistant, - g4f.Provider.Bing, - g4f.Provider.Bard -] - -_instruct = "Hello, are you GPT 4?." - -_example = """ -OpenaiChat: Hello! How can I assist you today? 2.0 secs -Bard: Hello! How can I help you today? 3.44 secs -Bing: Hello, this is Bing. How can I help? 😊 4.14 secs -Async Total: 4.25 secs - -OpenaiChat: Hello! How can I assist you today? 1.85 secs -Bard: Hello! How can I help you today? 3.38 secs -Bing: Hello, this is Bing. How can I help? 😊 6.14 secs -Stream Total: 11.37 secs - -OpenaiChat: Hello! How can I help you today? 3.28 secs -Bard: Hello there! How can I help you today? 3.58 secs -Bing: Hello! How can I help you today? 3.28 secs -No Stream Total: 10.14 secs -""" - -print("Bing: ", end="") -for response in log_time_yield( - g4f.ChatCompletion.create, - model=g4f.models.default, - messages=[{"role": "user", "content": _instruct}], - provider=g4f.Provider.Bing, - #cookies=g4f.get_cookies(".huggingface.co"), - stream=True, - auth=True -): - print(response, end="", flush=True) -print() -print() - - -async def run_async(): - responses = [ - log_time_async( - provider.create_async, - model=None, - messages=[{"role": "user", "content": _instruct}], - ) - for provider in _providers - ] - responses = await asyncio.gather(*responses) - for idx, provider in enumerate(_providers): - print(f"{provider.__name__}:", responses[idx]) -print("Async Total:", asyncio.run(log_time_async(run_async))) -print() - - -def run_stream(): - for provider in _providers: - print(f"{provider.__name__}: ", end="") - for response in log_time_yield( - provider.create_completion, - model=None, - messages=[{"role": "user", "content": _instruct}], - ): - print(response, end="", flush=True) - print() -print("Stream Total:", log_time(run_stream)) -print() - - -def create_no_stream(): - for provider in _providers: - print(f"{provider.__name__}:", end=" ") - for response in log_time_yield( - provider.create_completion, - model=None, - messages=[{"role": "user", "content": _instruct}], - stream=False - ): - print(response, end="") - print() -print("No Stream Total:", log_time(create_no_stream)) -print()
\ No newline at end of file |