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

updating test

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-22 15:16:08 -07:00
parent 2246bf6264
commit 7bb633e190
1 changed files with 16 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

32
litellm/tests/test_litedebugger_integration.py
View file

@ -1,22 +1,22 @@
#### What this tests #### # #### What this tests ####
# This tests if logging to the litedebugger integration actually works # # This tests if logging to the litedebugger integration actually works
# pytest mistakes intentional bad calls as failed tests -> [TODO] fix this # # pytest mistakes intentional bad calls as failed tests -> [TODO] fix this
import sys, os # import sys, os
import traceback # import traceback
import pytest # import pytest
sys.path.insert(0, os.path.abspath('../..')) # Adds the parent directory to the system path # sys.path.insert(0, os.path.abspath('../..')) # Adds the parent directory to the system path
import litellm # import litellm
from litellm import embedding, completion # from litellm import embedding, completion
litellm.email = "krrish@berri.ai" # litellm.email = "krrish@berri.ai"
user_message = "Hello, how are you?" # user_message = "Hello, how are you?"
messages = [{ "content": user_message,"role": "user"}] # messages = [{ "content": user_message,"role": "user"}]
#openai call # #openai call
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"}])
#bad request call # #bad request call
response = completion(model="chatgpt-test", messages=[{"role": "user", "content": "Hi 👋 - i'm a bad request"}]) # response = completion(model="chatgpt-test", messages=[{"role": "user", "content": "Hi 👋 - i'm a bad request"}])