phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Added openrouter.ai support

Browse source
This commit is contained in:
Zakhar Kogan 2023-07-28 11:44:08 +03:00
parent 2463dc234b
commit 00c2384a2d
8 changed files with 118 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

38
completion_test.py
View file

@ -1,4 +1,4 @@
from main import completion
from litellm.main import completion
import os
## Configs for Models ##
@ -8,23 +8,27 @@ import os
messages = [{ "content": "Hello, how are you?","role": "user"}]
# openai call
response = completion(model="gpt-3.5-turbo", messages=messages)
print("\nOpenAI call")
print(response)
# # 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)
# # 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)
# # 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)
# # cohere call
# response = completion("command-nightly", messages)
# print("\nCohere call")
# print(response)
# openrouter call
response = completion("google/palm-2-codechat-bison", messages)
print("\OpenRouter call")
print(response)