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

test issues

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-19 20:57:38 -07:00
parent 366ff43761
commit 2350865a47
1 changed files with 20 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

40
litellm/tests/test_supabase_integration.py
View file

@ -1,28 +1,28 @@
#### What this tests #### # #### What this tests ####
# This tests if logging to the helicone integration actually works # # This tests if logging to the helicone 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.input_callback = ["supabase"] # litellm.input_callback = ["supabase"]
litellm.success_callback = ["supabase"] # litellm.success_callback = ["supabase"]
litellm.failure_callback = ["supabase"] # litellm.failure_callback = ["supabase"]
litellm.modify_integration("supabase",{"table_name": "test_table"}) # litellm.modify_integration("supabase",{"table_name": "test_table"})
litellm.set_verbose = True # litellm.set_verbose = True
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"}])