g <ddlZddlZddlmZddlZGddZy)N) BaseModelceZdZdZdZy)PromptLayerLoggerc8tjd|_y)NPROMPTLAYER_API_KEY)osgetenvkey)selfs V/var/www/openai/venv/lib/python3.12/site-packages/litellm/integrations/prompt_layer.py__init__zPromptLayerLogger.__init__ s9923c  i}|d|d<|d|d<|dD] }|d|||<g}i} d|dvrCd|ddvr |ddd}|ddjD cic] \} } | dk7r| | } } } |d|d|t|tr|j}tj j d d ||t|t|jt|j|jd  } | j} | jjd ds td|d| jd| vrK| rHtj j d| d|j| d }|d|jyyycc} } w#t$r |dtj YywxYw)Nmodelmessagesoptional_paramsmetadatalitellm_paramspl_tagszAPrompt Layer Logging - Enters logging function for model kwargs: z , response: z.https://api.promptlayer.com/rest/track-requestzopenai.ChatCompletion.create) function_namekwargstagsrequest_responserequest_start_timerequest_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)rrrz?Prompt Layer Logging: success - metadata post response object: zerror: Prompt Layer Error - )items isinstancer model_dumplitellmmodule_level_clientpostdictint timestampr rget Exceptiontext traceback format_exc)r r response_obj start_timeend_time print_verbose new_kwargsoptional_paramrrkvr response_jsonresponses r log_eventzPromptLayerLogger.log_events]H J"(/Jw %+J%7Jz "#)):";-34E-F~-V >*#<DHV$455'7 8 DD!"23J? JD !''7 8 D J J L L1I~qD L ST^S__mnzm{|  , 2+668 &::??@%C( (,\(:*-j.B.B.D*E(+H,>,>,@(A#xx  @  "-113M#((*..y%@ TUU IJZJ_J_I`a },&::??I*7 *E'+xx(0 @ H"YZbZgZgYhi-Oj  89M9M9O8PQ R  s%A F5"F/4D8F5/F55&GGN)__name__ __module__ __qualname__r r8rr rr s 4Jrr)rr,pydanticrr#rr<rr r>s PPr