o ưiW%@sddlmZmZmZmZmZmZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZddlmZmZdd lmZdd lmZerQdd lmZeZneZGd d d e ZdS)) TYPE_CHECKINGAnyDictListOptionalUnionN)verbose_logger) XAI_API_BASE)OpenAIResponsesAPIConfig)get_secret_str)!ResponsesAPIOptionalRequestParams)XAIWebSearchToolXAIXSearchTool)GenericLiteLLMParams) LlmProviders)LoggingcseZdZdZedefddZdedeffdd Z de ee fde e e ee fffd d Zde ee fde ee ee fffd d Zd ededede fddZdededeedefddZdeededefddZdefddZZS)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 returncCstjS)N)rZXAIselfr`/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/xai/responses/transformation.pycustom_llm_provider"sz)XAIResponsesAPIConfig.custom_llm_providermodelcs"t|}d|vr|d|S)z Get supported parameters for XAI Responses API. XAI supports most OpenAI Responses API params except 'instructions'. instructions)superget_supported_openai_paramsremove)rrZsupported_params __class__rrr&s  z1XAIResponsesAPIConfig.get_supported_openai_paramstoolcCsrddi}d|vr tdi}d|vr|d}||d<d|vr'|d}||d<|r-||d<d|vr7|d|d<|S) a1 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_searchZsearch_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$rrr_transform_web_search_tool4s"  z0XAIResponsesAPIConfig._transform_web_search_toolcCsddi}d|vr|d}||d<d|vr|d}||d<d|vr&|d|d<d|vr0|d|d<d|vr:|d|d<d|vrD|d|d<|S) a` 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_searchZallowed_x_handlesZexcluded_x_handlesZ from_dateZto_dater&Zenable_video_understandingr)rr r(Zallowed_handlesZexcluded_handlesrrr_transform_x_search_tool[s      z.XAIResponsesAPIConfig._transform_x_search_toolresponse_api_optional_params drop_paramsc Cst|}d|vrtd|dd|vr td|dd|vr|dr|d}t|ts4|g}g}|D]L}t|tr|d}|dkrUtd|ddiq8|d krgtd |||q8|d krytd || |q8||q8||q8||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.Ztoolsr!Zcode_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)r+) rr,rr-paramsZ tools_listZtransformed_toolsr Z tool_typerrrmap_openai_paramssL       z'XAIResponsesAPIConfig.map_openai_paramsheaderslitellm_paramscCsB|pt}|jptjptd}|std|dd|i|S)z Validate environment and set up headers for XAI API. Uses XAI_API_KEY from environment or litellm_params. Z XAI_API_KEYzXXAI API key is required. Set XAI_API_KEY environment variable or pass api_key parameter. AuthorizationzBearer )rapi_keylitellmr ValueErrorupdate)rr8rr9r;rrrvalidate_environments  z*XAIResponsesAPIConfig.validate_environmentapi_basecCs*|p tjp tdp t}|d}|dS)z Get the complete URL for XAI Responses API endpoint. Returns: str: The full URL for the XAI /responses endpoint r /z /responses)r<r@r r rstrip)rr@r9rrrget_complete_urls   z&XAIResponsesAPIConfig.get_complete_urlcCsdS)z7XAI does not support native WebSocket for Responses APIFrrrrrsupports_native_websocketsz/XAIResponsesAPIConfig.supports_native_websocket)__name__ __module__ __qualname____doc__propertyrrstrr3rrrrr r)rr+r boolr7r/rrr?rCrD __classcell__rrrrrsB **'( J  r)typingrrrrrrr<Zlitellm._loggingrZlitellm.constantsr Z,litellm.llms.openai.responses.transformationr Zlitellm.secret_managers.mainr Zlitellm.types.llms.openair Zlitellm.types.llms.xair rZlitellm.types.routerrZlitellm.types.utilsrZ*litellm.litellm_core_utils.litellm_loggingrZ_LiteLLMLoggingObjZLiteLLMLoggingObjrrrrrs