From b04546335524637f800c001d3ffd9deab6e27c35 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 31 Jul 2023 11:18:11 -0700 Subject: [PATCH] run tests --- completion_test.py | 29 ----------------------------- embedding_test.py | 19 ------------------- requirements.txt | 4 ++++ setup.py | 3 ++- 4 files changed, 6 insertions(+), 49 deletions(-) delete mode 100644 completion_test.py delete mode 100644 embedding_test.py create mode 100644 requirements.txt diff --git a/completion_test.py b/completion_test.py deleted file mode 100644 index c9e74932c5..0000000000 --- a/completion_test.py +++ /dev/null @@ -1,29 +0,0 @@ -from litellm.main import completion -import os - -## Configs for Models ## -# OpenAI Configs - -## End Configs for Models ## - - -messages = [{ "content": "Hello, how are you?","role": "user"}] -# openai call -response = completion(model="gpt-3.5-turbo", messages=messages) -print("\nOpenAI call") -print(response) - -# azure openai call -response = completion("chatgpt-test", messages, azure=True) -print("\nAzure call") -print(response) - -# text davinci openai call -response = completion("text-davinci-003", messages) -print("\nDavinci call") -print(response) - -# cohere call -response = completion("command-nightly", messages) -print("\nCohere call") -print(response) \ No newline at end of file diff --git a/embedding_test.py b/embedding_test.py deleted file mode 100644 index e9b085719e..0000000000 --- a/embedding_test.py +++ /dev/null @@ -1,19 +0,0 @@ -from main import embedding -import os - -## Configs for Models ## -# OpenAI Configs - -## End Configs for Models ## - - -input = ["Who is ishaan"] -# openai call -response = embedding(model="text-embedding-ada-002", input=input) -print("\nOpenAI call") -print(response) - -# azure openai call -response = embedding(model="azure-embedding-mode", input=input, azure=True) -print("\nAzure OpenAI call") -print(response) diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000000..73d7f7bf74 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,4 @@ +openai +cohere +func_timeout +anthropic \ No newline at end of file diff --git a/setup.py b/setup.py index a143fbe5bc..f4de6c1816 100644 --- a/setup.py +++ b/setup.py @@ -10,6 +10,7 @@ setup( ], install_requires=[ 'openai', - 'cohere' + 'cohere', + 'func_timeout' ], )