uPje&ddlmZmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZmZdd lmZdd lmZGd de Zy))AnyFinalN)verbose_logger) XAI_API_BASE)AuthenticationError)OpenAIResponsesAPIConfig) XAIModelInfo)get_secret_str)!ResponsesAPIOptionalRequestParams)XAIWebSearchToolXAIXSearchTool)GenericLiteLLMParams) LlmProvidersceZdZdZedefdZdedeffd Z de ee fde e ee fzfdZ de ee fdee ee fzfdZd eded ede fd Zd e ded edzde fdZdedzd e defdZdefdZxZS)XAIResponsesAPIConfiga Configuration for XAI's Responses API. Inherits from OpenAIResponsesAPIConfig since XAI's Responses API is largely compatible with OpenAI's, with a few differences: - Does not support the 'instructions' parameter - Requires code_interpreter tools to have 'container' field removed - Recommends store=false when sending images Reference: https://docs.x.ai/docs/api-reference#create-new-response returnc"tjS)N)rXAIselfs n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/xai/responses/transformation.pycustom_llm_providerz)XAIResponsesAPIConfig.custom_llm_providersmodelcPt||}d|vr|jd|S)z Get supported parameters for XAI Responses API. XAI supports most OpenAI Responses API params except 'instructions'. instructions)superget_supported_openai_paramsremove)rrsupported_params __class__s rrz1XAIResponsesAPIConfig.get_supported_openai_params!s4 #('"Ee"L - -  # #N 3rtoolcddi}d|vrtjdi}d|vr |d}||d<d|vr |d}||d<|r||d<d|vr|d|d<|S) a! Transform web_search tool to XAI format. XAI supports web_search with specific filters: - allowed_domains (max 5) - excluded_domains (max 5) - enable_image_understanding XAI does NOT support search_context_size (OpenAI-specific). type web_searchsearch_context_sizezWXAI does not support 'search_context_size' parameter. Removing it from web_search tool.allowed_domainsexcluded_domainsfiltersenable_image_understanding)rinfo)rr"xai_toolr)r'r(s r_transform_web_search_toolz0XAIResponsesAPIConfig._transform_web_search_tool/s,2<*@ !D (   i    $%)*;% *:G& ' ")HY  (4 /59:V5WH1 2rcddi}d|vr |d}||d<d|vr |d}||d<d|vr|d|d<d|vr|d|d<d|vr|d|d<d|vr|d|d<|S) aX Transform x_search tool to XAI format. XAI supports x_search with specific parameters: - allowed_x_handles (max 10) - excluded_x_handles (max 10) - from_date (ISO8601: YYYY-MM-DD) - to_date (ISO8601: YYYY-MM-DD) - enable_image_understanding - enable_video_understanding r$x_searchallowed_x_handlesexcluded_x_handles from_dateto_dater*enable_video_understanding)rr"r,allowed_handlesexcluded_handless r_transform_x_search_toolz.XAIResponsesAPIConfig._transform_x_search_toolVs,2:*> $ &%)*=%>O,;H( ) 4 '&*+?&@ -=H) * $ $($5H[ !  "&y/HY  (4 /59:V5WH1 2 '4 /59:V5WH1 2rresponse_api_optional_params drop_paramsct|}d|vr&tjd|jdd|vr&tjd|jdd|vr|dr|d}t |t s|g}g}|D]}t |tr|j d}|dk(r)tjd|jddiR|d k(r6tjd |j|j||d k(r6tjd |j|j||j||j|||d<|S) a Map parameters for XAI Responses API. Handles XAI-specific transformations: 1. Drops 'instructions' parameter (not supported) 2. Transforms code_interpreter tools to remove 'container' field 3. Transforms web_search tools to XAI format (removes search_context_size, adds filters) 4. Transforms x_search tools to XAI format 5. Sets store=false when images are detected (recommended by XAI) rzIXAI Responses API does not support 'instructions' parameter. Dropping it.metadatazEXAI Responses API does not support 'metadata' parameter. Dropping it.toolsr$code_interpreterzAXAI: Transforming code_interpreter tool, removing container fieldr%z/XAI: Transforming web_search tool to XAI formatr/z-XAI: Transforming x_search tool to XAI format) dictrdebugpop isinstancelistgetappendr-r8) rr9rr:params tools_listtransformed_toolsr" tool_types rmap_openai_paramsz'XAIResponsesAPIConfig.map_openai_params~si 9: V #  !l m JJ~ &    !h i JJz " f Jj$/(\ 24 " 3dD) $ 0I $66&,,-pq)00&:L1MN"l2&,,-^_)001P1PQU1VW"j0&,,-\])001N1Nt1TU*006%,,T2/ 320F7O rheaderslitellm_paramsNc|xs t}tj|jd}|s6ddlm}m}m}||jr |j}|s td|jdd |i|S#|$r1}t||jjt||d}~wwxYw) z Validate environment and set up headers for XAI API. Uses the shared xAI key resolver with Responses API legacy precedence. Tlegacy_generic_before_envr)XAIOAuthAuthenticator XAIOAuthErrorshould_use_xai_oauth)r llm_providermessageNzkXAI API key is required. Set api_key, litellm.xai_key, litellm.api_key, XAI_API_KEY, or use_xai_oauth=True. AuthorizationzBearer )rr get_api_keyapi_keylitellm.llms.xai.oauthrPrQrR model_dumpget_access_tokenrrvaluestr ValueErrorupdate) rrKrrLrWrPrQrRexcs rvalidate_environmentz*XAIResponsesAPIConfig.validate_environments (A+?+A**>+A+A]ab   $N$=$=$?@35FFHGG  77)!4  %%-#%)%=%=%C%C #C sB C,B>>Capi_basecddlm}m}tj|j dd}||r|s|j }n)|xs%tjxstdxst}|jd}|dS) z Get the complete URL for XAI Responses API endpoint. Returns: str: The full URL for the XAI /responses endpoint r)rPrRrWTrNr/z /responses) rXrPrRr rVrD get_api_baselitellmrar rrstrip)rrarLrPrRrWs rget_complete_urlz&XAIResponsesAPIConfig.get_complete_urls~ W%11.2D2DY2Okop  /,.;;=He7#3#3e~n7UeYeH??3':&&rcy)z7XAI does not support native WebSocket for Responses APIFr5rs rsupports_native_websocketz/XAIResponsesAPIConfig.supports_native_websocketsr)__name__ __module__ __qualname____doc__propertyrrr\rCrr?rr r-r r8r boolrJrr`rgri __classcell__)r!s@rrrs  \        %tCH~%BRUYZ]_bZbUcBc%N&T#s(^&QUVY[^V^Q_@_&P>&G>> >  >@%D%%NbeiNi%nr%N'*'' '04rr)typingrrrelitellm._loggingrlitellm.constantsrlitellm.exceptionsr,litellm.llms.openai.responses.transformationrlitellm.llms.xai.common_utilsr litellm.secret_managers.mainr litellm.types.llms.openair litellm.types.llms.xair r litellm.types.routerrlitellm.types.utilsrrr5rrr|s8+*2Q67GC5,o4or