From 93a3a0cc1e54c7850c2b8e82df635fcc54a024d4 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 8 Jun 2024 08:59:56 -0700 Subject: [PATCH] fix(utils.py): fix helicone success logging integration Fixes https://github.com/BerriAI/litellm/issues/4062 --- litellm/tests/test_helicone_integration.py | 3 ++- litellm/utils.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_helicone_integration.py b/litellm/tests/test_helicone_integration.py index 82669d09206..03485600092 100644 --- a/litellm/tests/test_helicone_integration.py +++ b/litellm/tests/test_helicone_integration.py @@ -26,5 +26,6 @@ # # cohere call # response = completion( -# model="command-nightly", messages=[{"role": "user", "content": "Hi 👋 - i'm cohere"}] +# model="command-nightly", +# messages=[{"role": "user", "content": "Hi 👋 - i'm cohere"}], # ) diff --git a/litellm/utils.py b/litellm/utils.py index d80d7d68f82..410f9ad8826 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1787,7 +1787,7 @@ class Logging: if callback == "helicone": print_verbose("reaches helicone for logging!") model = self.model - messages = kwargs["input"] + messages = self.model_call_details["input"] heliconeLogger.log_success( model=model, messages=messages,