tPjPUddlmZdZ ddlZddlZddlZddlmZmZddlm Z ejdk\r ddlm Z m Z n ddlm Z dd l m Z ejeZeed <ed e Zed ZGdde eefZGddZddlZGddZy#e$rdZYwxYw))FinalTN)AnyTypeVar) trace_tree))LiteralProtocol)r )r loggerK)boundVcPeZdZUeded<dedefdZd dededzdedzfdZy) OpenAIResponse)chat.completionedittext_completionobjectkeyreturncyN)selfrs h/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/weights_biases.py __getitem__zOpenAIResponse.__getitem__Ndefaultcyrr)rrrs rgetzOpenAIResponse.getrrr) __name__ __module__ __qualname__r __annotations__r rrr!rrrrrs5DEE/1//H1Hq4xH1t8Hrrc eZdZdeeefdededejdzfdZ e deeefdede ejdedejf dZdeeefdededejfd Zdeeefdededejfd Zdeeefdededejfd Zdeeefded ed e ededejf dZy)OpenAIRequestResponseResolverrequestresponse time_elapsedrNc4 |ddk(r|j|||S|ddk(r|j|||S|ddk(r|j|||Stj d|d y#t $r }tj d|Yd}~yd}~wwxYw)Nrrrrz"Unknown OpenAI response object: %sz&Failed to resolve request/response: %s) _resolve_edit_resolve_completion_resolve_chat_completionr debug Exceptionwarning)rr(r)r*es r__call__z&OpenAIRequestResponseResolver.__call__!s LH%/--gxNNh'+<<33GX|TTh'+<<88(LYYLL!ExPXGYZ LGKK Ls'A.A.A.A.. B7BBresultsc ptt|ddz}|tt|dzz}tj|j ddd|dd|j dt |||tj j|}||dd}tj|| S) ajConverts the request, response, and results into a trace tree. params: request: The request dictionary response: The response object results: A list of results object time_elapsed: The time elapsed in seconds returns: A wandb trace tree object. createdimodelopenai_r)name attributes start_time_ms end_time_ms span_kindr4)r(r)_kind) root_span model_dict) introundrSpanr!dictSpanKindLLM WBTraceTree)r(r)r4r*r<r=span model_objs rresults_to_trace_treez3OpenAIRequestResponseResolver.results_to_trace_tree4s"$'uXi-@4-G'H#IM!.U<$;N5O1P!PK$// Wh78(8:L9MQx||\eOfNgh>+'$--11 D,3S[\I))DYO Orcd|dd|dd}|dDcgc] }d|dd }}|j||||| Scc}w) zrs_$ #/ 6!,,".%G%%h/FE/3A AI!Q$I@@PQQ$#$sBBB%$B%