uPj00ddlZddlZddlmZmZmZddlZddlmZddl m Z ddl m Z ddl mZmZddlmZdd lmZdd lmZmZmZerdd lmZGd d e Zy)N) TYPE_CHECKINGAnyFinal)AuthenticationError)AnthropicConfig) OpenAIConfig)AllMessageValuesChatCompletionToolCallChunk) ModelResponse) Authenticator)DEFAULT_GITHUB_COPILOT_API_BASEGetAPIKeyErrorget_copilot_default_headers)LoggingceZdZ d#dedzdedzdeddffd Zdededzdedzdedeedzedzeff dZdefd Z d$d eded e e d ed ededzdedzdeffd Z dede ffd Z d e e defdZ d e e defdZede edeee ee edzffdZededefdZededefdZdedefdZ d$dedej2ddddded e e d ed ed ededzd!edzddffd" ZxZS)%GithubCopilotConfigNapi_keyapi_basecustom_llm_providerreturnc@t|t|_y)N)super__init__r authenticator)selfrrr __class__s t/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/github_copilot/chat/transformation.pyrzGithubCopilotConfig.__init__s *_modelc |xs9|jjxstjdxst} |jj }|||fS#t $r}t||t|d}~wwxYw)NGITHUB_COPILOT_API_BASE)r llm_providermessage) r get_api_baseosgetenvr get_api_keyrrstr)rr rrrdynamic_api_basedynamic_api_keyes r$_get_openai_compatible_provider_infoz8GithubCopilotConfig._get_openai_compatible_provider_info"s  /!!..0 /yy23 //   %)%7%7%C%C%EO 2EEE  %0A  sA B'A>>Bcddl}|jr|Sg}|D]N}|jddk(r'|j}d|d<|j |>|j |P|S)Nrrolesystem assistant)litellm#disable_copilot_system_to_assistantgetcopyappend)rmessagesr r2transformed_messagesr$transformed_messages r_transform_messagesz'GithubCopilotConfig._transform_messages9s}   6 6O') 5G{{6"h.&-lln#.9#F+$++,?@$++G4 5$#rheadersr7optional_paramslitellm_paramsc t ||||||||} |jj} t | } i| |}|j |} | |d<|j|rd|d<|S#t $rY9wxYw)Nz X-InitiatortruezCopilot-Vision-Request)rvalidate_environmentrr(rr_determine_initiator_has_vision_content) rr;r r7r<r=rrvalidated_headerscopilot_api_keycopilot_headers initiatorrs rr@z(GithubCopilotConfig.validate_environmentSs"G8 UHo~wPX   %)%7%7%C%C%EO%@%QO H? H6G H   44X> +4-(  # #H -:@ 6 7     s+A11 A=<A=cddlm}t| |}d|j vrA||j r*d|vr|j dd|vr|j d|S)aK Get supported OpenAI parameters for GitHub Copilot. For Claude models that support extended thinking (Claude 4 family and Claude 3-7), includes thinking and reasoning_effort parameters. For other models, returns standard OpenAI parameters (which may include reasoning_effort for o-series models). r)supports_reasoningclaude)r thinkingreasoning_effort) litellm.utilsrHrget_supported_openai_paramslowerr6)rr rH base_paramsrs rrMz/GithubCopilotConfig.get_supported_openai_paramstso 5#W@G  u{{} $);++-* ,"":.!4""#56rc@|D]}|jd}|dvsyy)z Determine if request is user or agent initiated based on message roles. Returns 'agent' if any message has role 'tool' or 'assistant', otherwise 'user'. r/)toolr1agentuser)r4)rr7r$r/s rrAz(GithubCopilotConfig._determine_initiators2   G;;v&D,, rc|D]\}|jd}t|ts%|D]3}t|tsd|vry|jd}|dk(s2y^y)a Check if any message contains vision content (images). Returns True if any message has content with vision-related types, otherwise False. Checks for: - image_url content type (OpenAI format) - Content items with type 'image_url' content image_urlTtypeF)r4 isinstancelistdict)rr7r$rU content_item content_types rrBz'GithubCopilotConfig._has_vision_contentso (Gkk),G'4($+(L!,5&,6#''3'7'7'? ';6#'( (rcontent_blockscVtjd|i\}}}}}}}}|||fS)z Parse Anthropic-native content blocks into OpenAI-compatible fields. Concatenates all text blocks, extracts tool_use blocks as tool_calls, and preserves thinking blocks when present. rU)completion_response)rextract_response_content) r] text_content _citationsthinking_blocks_reasoning_content tool_calls_web_search_results _tool_results_compaction_blockss r_parse_anthropic_native_contentz3GithubCopilotConfig._parse_anthropic_native_contentsH&   6 6IWeKf 6 g        Z88rusagect|}d|vr d|vr|d|d<d|vr d|vr|d|d<d|vr(|jdd|jddz|d<|S)N input_tokens prompt_tokens output_tokenscompletion_tokens total_tokensr)rZr4)rj normalizeds r_normalize_anthropic_usagez.GithubCopilotConfig._normalize_anthropic_usages~ K U "e'C*/*?J ' e #(;5(H.3O.DJ* +  +)3)Kjnn]prsNt)tJ~ &r response_jsonc|jdr|Sd}g}d}|jd}t|tr|j|\}}}nt|tr|}|jd}ddddd }|rd}n||vr||}n|rd}nd}d |s|s|ndd } |r|| d<|r|| d <i|dd | |dgi} |jd} t| t r|j | | d<| S)a Synthesize a `choices` array from an Anthropic-native Copilot response. Newer Copilot Claude models (e.g. opus-4.7, opus-4.8) return content blocks and `stop_reason` without an OpenAI-style `choices` array, and the max_tokens=1 probe returns no content at all. Returns the response unchanged when it already carries choices. See: https://github.com/BerriAI/litellm/issues/29391 choicesNrU stop_reasonstoplengthre)end_turn max_tokens stop_sequencetool_user1)r/rUrcr)indexr$ finish_reasonrj)r4rXrYrir)rZrr) clsrsrUrerc raw_contentrwfinish_reason_maprr$ synthesizedrjs r(_synthesize_choices_for_anthropic_nativez>(44$,, =188:L w)%)#%+)*  / 7-)-')! /-!#.   sB#CC)NNopenai)NN)__name__ __module__ __qualname__r)rtupler-r:rZrYr r@rMrAboolrB staticmethodrr rirr classmethodrrrrobjectr __classcell__)rs@rrrs###+ -t-*-! -  -FF*Ft F ! F sTz3:s* + F.$$B##!!!'( !  !  !t!*! !B0 T2B-C  D1A,Bt09S 9 sD45tCy47GG H99*$47T7d77rNdNtN$#!%3 3 nn3 ( 3 ) 3  3 '(3 3 3 3 t3 $;3  3 3 rr)rr&typingrrrrlitellm.exceptionsr*litellm.llms.anthropic.chat.transformationrlitellm.llms.openai.openairlitellm.types.llms.openair r litellm.types.utilsr rr common_utilsrrr*litellm.litellm_core_utils.litellm_loggingrrrrrrsF ,, 2F3S-) Wl ,l r