o ưi @s6ddlZddlZddlmZddlZGdddZdS)N) BaseModelc@seZdZddZddZdS)PromptLayerLoggercCstd|_dS)NZPROMPTLAYER_API_KEY)osgetenvkey)selfrX/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/prompt_layer.py__init__ szPromptLayerLogger.__init__c Cszi}|d|d<|d|d<|dD] }|d|||<qg}i} d|dvrEd|ddvr8|ddd}dd|ddD} |d |d |t|trX|}tjjd d ||t|t| t| |j d d} | } | ddst d|d| jd| vr| rtjjd| d|j | dd} |d| jWdSWdSWdSt y|dtYdSw)NmodelmessagesZoptional_paramsmetadataZlitellm_paramspl_tagscSsi|] \}}|dkr||qS)rr).0kvrrr $s z/PromptLayerLogger.log_event..zAPrompt Layer Logging - Enters logging function for model kwargs: z , response: z.https://api.promptlayer.com/rest/track-requestzopenai.ChatCompletion.create)Z function_namekwargstagsrequest_responseZrequest_start_timeZrequest_end_timeapi_key)jsonsuccessFz2Promptlayer did not successfully log the response!z7Prompt Layer Logging: success - final response object: request_idz/https://api.promptlayer.com/rest/track-metadata)rrr z?Prompt Layer Logging: success - metadata post response object: zerror: Prompt Layer Error - )items isinstancerZ model_dumplitellmZmodule_level_clientpostdictint timestamprrget Exceptiontext traceback format_exc) rrZ response_obj start_timeend_timeZ print_verboseZ new_kwargsZoptional_paramrr rZ response_jsonresponserrr log_eventsl           zPromptLayerLogger.log_eventN)__name__ __module__ __qualname__r r)rrrr r s r)rr$Zpydanticrrrrrrr s