`i- bddlZddlmZddlZddlmZddlmZddlmZm Z m Z m Z m Z ddl mZddlmZmZddlmZdd lmZerdd lmZmZmZdd lmZ ddlZdd lmZmZmZn#e$r ed wxYwd dZd!dZ d"dZ! d#dZ"ddde#ddddddf dZ$GddeZ%dS)$N) TYPE_CHECKING)consts)record_token_usage)get_start_span_functionset_data_normalizedtruncate_and_annotate_messagestransform_openai_content_part&truncate_and_annotate_embedding_inputs)SPANDATA) DidNotEnable Integration)should_send_default_pii)event_from_exception)AnyDictList)datetime)input_callbacksuccess_callbackfailure_callbackzLiteLLM not installedkwargsDict[str, Any]returncn|di}|d}|i}||d<|S)z,Get the metadata dictionary from the kwargs.litellm_paramsmetadata) setdefaultget)rrrs s/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/sentry_sdk/integrations/litellm.py_get_metadata_dictr sG&&'7<xu U U  ,' 8%$ .48$D%F:.F!,,..88Y 3   i 7 7 7JJz**zz-00%zz*?@@N %**,,FF U   &=&=&=u E E EFFsA!!A21A2completion_responser start_timerend_timec t|d}|dStjt }|dS t |dr t|tj |j tr\|j rTt |drCg}|j D]}t |drt |jdr-||jVt |jdr-||ji}t |jdr|jj|d<t |jd r|jj|d <t |jd r|jj|d <|||rt|tj|t |d rG|j} t/|t1| d dt1| d dt1| dd|ddddS#|dddwxYw)zHandle successful completion.r>Nr3choicesr- model_dumpr)roler% tool_callsusage prompt_tokenscompletion_tokens total_tokens) input_tokens output_tokensr})r rrJrKrLrMhasattrrr GEN_AI_RESPONSE_MODELr3rrWrvr-r,rwr)rxr%ryGEN_AI_RESPONSE_TEXTrzrgetattr__exit__) rrrrsrtrhrcresponse_messageschoicemsgrzs r_success_callbackrs f % % ) ). 9 9D |'))99:LMMK-( & 0 0  h46I6O    # $ $ )D *I66 $&!19::Fvy11:"6><@@ :-44V^5N5N5P5PQQQQ$V^V<< :-44V^5H5H5J5JKKKK#%C&v~v>>B.4n.AF &v~yAAH171GI&v~|DDN4:N4ML 1-44S999$'h;=N & 0 0 '-E $UOTBB%e-@$GG$UNDAA      dD$''''' dD$''''s G2I""I; exceptioncvt|d}|dS t|tjjddd\}}tj|||t||ddS#|t||dwxYw)zHandle request failure.r>NrNF)typehandled)client_options mechanism)hint) r rrrJrKoptions capture_eventrr)rrrsrtrheventrs r_failure_callbackrs f % % ) ). 9 9D | 8* %022:(U;;   t  T2222 d9ooy$77777 d9ooy$7777s AB&B8cNeZdZdZdZdeZd dddeddfd Zed d Z dS) rMa" LiteLLM integration for Sentry. This integration automatically captures LiteLLM API calls and sends them to Sentry for monitoring and error tracking. It supports all 100+ LLM providers that LiteLLM supports, including OpenAI, Anthropic, Google, Cohere, and many others. Features: - Automatic exception capture for all LiteLLM calls - Token usage tracking across all providers - Provider detection and attribution - Input/output message capture (configurable) - Streaming response support - Cost tracking integration Usage: ```python import litellm import sentry_sdk # Initialize Sentry with the LiteLLM integration sentry_sdk.init( dsn="your-dsn", send_default_pii=True integrations=[ sentry_sdk.integrations.LiteLLMIntegration( include_prompts=True # Set to False to exclude message content ) ] ) # All LiteLLM calls will now be monitored response = litellm.completion( model="gpt-3.5-turbo", messages=[{"role": "user", "content": "Hello!"}] ) ``` Configuration: - include_prompts (bool): Whether to include prompts and responses in spans. Defaults to True. Set to False to exclude potentially sensitive data. rNzauto.ai.TselfrWrNc||_dS)N)rW)rrWs r__init__zLiteLLMIntegration.__init__5s.r!ctpgt_ttjvr$tjttpgt_t tjvr$tjt t pgt_ttjvr&tjtdSdS)z(Set up LiteLLM callbacks for monitoring.N)rrNrqr,rrrrr!r setup_oncezLiteLLMIntegration.setup_once8s"0!52 '"8 8 8  " ) )/ : : :#3#9r G$< < <  $ + +,= > > >#3#9r G$< < <  $ + +,= > > > > > = rs 777777'&&&&&========444444111111"&&&&&&&&&&!!!!!!0NNNJJJJJJJJJJJ000 ,. / //0    6eFeFeFeFP=( =(=(=( =(  =(=(=(=(@8 888 8  88880@?@?@?@?@?@?@?@?@?@?s A%%A6