phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

formatting improvements

Browse source
This commit is contained in:
ishaan-jaff 2023-08-28 09:20:50 -07:00
parent 3e0a16acf4
commit a69b7ffcfa
17 changed files with 464 additions and 323 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/tests/test_llmonitor_integration.py
View file

@ -4,7 +4,7 @@
import sys
import os
sys.path.insert(0, os.path.abspath('../..'))
sys.path.insert(0, os.path.abspath("../.."))
from litellm import completion, embedding
import litellm
@ -17,11 +17,10 @@ litellm.set_verbose = True
def test_chat_openai():
try:
response = completion(model="gpt-3.5-turbo",
messages=[{
"role": "user",
"content": "Hi 👋 - i'm openai"
}])
response = completion(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": "Hi 👋 - i'm openai"}],
)
print(response)
@ -31,7 +30,7 @@ def test_chat_openai():
def test_embedding_openai():
try:
response = embedding(model="text-embedding-ada-002", input=['test'])
response = embedding(model="text-embedding-ada-002", input=["test"])
# Add any assertions here to check the response
print(f"response: {str(response)[:50]}")
except Exception as e:
@ -39,4 +38,4 @@ def test_embedding_openai():
test_chat_openai()
test_embedding_openai()
test_embedding_openai()