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

fix(helicone.py): fix logging

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-02 14:29:47 -08:00
parent 9947b0db96
commit 31b19a420f
2 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/integrations/helicone.py
View file

@ -2,6 +2,7 @@
# On success, logs events to Helicone # On success, logs events to Helicone
import dotenv, os import dotenv, os
import requests import requests
import litellm
dotenv.load_dotenv() # Loading env variables using dotenv dotenv.load_dotenv() # Loading env variables using dotenv
import traceback import traceback
@ -56,6 +57,10 @@ class HeliconeLogger:
else "gpt-3.5-turbo" else "gpt-3.5-turbo"
) )
provider_request = {"model": model, "messages": messages} provider_request = {"model": model, "messages": messages}
if isinstance(response_obj, litellm.EmbeddingResponse) or isinstance(
response_obj, litellm.ModelResponse
):
response_obj = response_obj.json()
if "claude" in model: if "claude" in model:
provider_request, response_obj = self.claude_mapping( provider_request, response_obj = self.claude_mapping(

2
litellm/utils.py
View file

@ -1269,7 +1269,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["messages"] messages = kwargs["input"]
heliconeLogger.log_success( heliconeLogger.log_success(
model=model, model=model,
messages=messages, messages=messages,