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

fix(utils.py): fix helicone success logging integration

Browse source 
Fixes https://github.com/BerriAI/litellm/issues/4062
This commit is contained in:
Krrish Dholakia 2024-06-08 08:59:56 -07:00
parent af39e2d2dd
commit 93a3a0cc1e
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_helicone_integration.py
View file

@ -26,5 +26,6 @@
# # cohere call # # cohere call
# response = completion( # response = completion(
# model="command-nightly", messages=[{"role": "user", "content": "Hi 👋 - i'm cohere"}] # model="command-nightly",
# messages=[{"role": "user", "content": "Hi 👋 - i'm cohere"}],
# ) # )

2
litellm/utils.py
View file

@ -1787,7 +1787,7 @@ class Logging:
if callback == "helicone": if callback == "helicone":
print_verbose("reaches helicone for logging!") print_verbose("reaches helicone for logging!")
model = self.model model = self.model
messages = kwargs["input"] messages = self.model_call_details["input"]
heliconeLogger.log_success( heliconeLogger.log_success(
model=model, model=model,
messages=messages, messages=messages,