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' ], )