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

fix(main.py): removing print_verbose

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-30 20:37:12 -07:00
parent c04cad208c
commit 3743893e76
3 changed files with 22 additions and 65 deletions
Download patch file Download diff file Expand all files Collapse all files

44
litellm/tests/test_helicone_integration.py
View file

@ -1,30 +1,30 @@
#### What this tests ####
# This tests if logging to the helicone integration actually works
# #### What this tests ####
# # This tests if logging to the helicone integration actually works
import sys, os
import traceback
import pytest
# import sys, os
# import traceback
# import pytest
sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
import litellm
from litellm import embedding, completion
# sys.path.insert(
# 0, os.path.abspath("../..")
# ) # Adds the parent directory to the system path
# import litellm
# from litellm import embedding, completion
litellm.success_callback = ["helicone"]
# litellm.success_callback = ["helicone"]
litellm.set_verbose = True
# litellm.set_verbose = True
user_message = "Hello, how are you?"
messages = [{"content": user_message, "role": "user"}]
# user_message = "Hello, how are you?"
# messages = [{"content": user_message, "role": "user"}]
# openai call
response = completion(
model="gpt-3.5-turbo", messages=[{"role": "user", "content": "Hi 👋 - i'm openai"}]
)
# # openai call
# response = completion(
# model="gpt-3.5-turbo", messages=[{"role": "user", "content": "Hi 👋 - i'm openai"}]
# )
# cohere call
response = completion(
model="command-nightly", messages=[{"role": "user", "content": "Hi 👋 - i'm cohere"}]
)
# # cohere call
# response = completion(
# model="command-nightly", messages=[{"role": "user", "content": "Hi 👋 - i'm cohere"}]
# )