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

fix promptlayer logging

Browse source
This commit is contained in:
ishaan-jaff 2023-09-13 13:22:43 -07:00
parent 508a2c1daf
commit 1913d36e05
3 changed files with 40 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/integrations/prompt_layer.py
View file

@ -17,8 +17,11 @@ class PromptLayerLogger:
def log_event(self, kwargs, response_obj, start_time, end_time, print_verbose):
# Method definition
try:
if 'litellm_logging_obj' in kwargs:
kwargs.pop('litellm_logging_obj')
print_verbose(
f"Prompt Layer Logging - Enters logging function for model {kwargs}"
f"Prompt Layer Logging - Enters logging function for model kwargs: {kwargs}\n, response: {response_obj}"
)
request_response = requests.post(
@ -27,9 +30,7 @@ class PromptLayerLogger:
"function_name": "openai.ChatCompletion.create",
"kwargs": kwargs,
"tags": ["hello", "world"],
"request_response": dict(
response_obj
), # TODO: Check if we need a dict
"request_response": dict(response_obj),
"request_start_time": int(start_time.timestamp()),
"request_end_time": int(end_time.timestamp()),
"api_key": self.key,
@ -39,11 +40,9 @@ class PromptLayerLogger:
# "prompt_version":1,
},
)
print_verbose(
f"Prompt Layer Logging - final response object: {request_response}"
f"Prompt Layer Logging: success - final response object: {request_response}"
)
except:
# traceback.print_exc()
print_verbose(f"Prompt Layer Error - {traceback.format_exc()}")
print_verbose(f"error: Prompt Layer Error - {traceback.format_exc()}")
pass