o ưi4#@sdZddlZddlmZmZmZmZmZmZm Z ddl Z ddl m Z ddl mZmZmZddlmZdd lmZerCdd lmZeZneZGd d d eeZdS) z Support for Snowflake REST API N) TYPE_CHECKINGAnyDictListOptionalTupleUnion)AllMessageValues)ChatCompletionMessageToolCallFunction ModelResponse)OpenAIGPTConfig)SnowflakeBaseConfig)LoggingcsXeZdZdZefddZdeeee fde ee ee ffddZ  d#d ed ejd ed ed edeededede de ede edefddZ d$de ede ed ededede edefddZdeeee fdeeee ffddZdeeeee ffdeeeee fffddZd edeededed edef d!d"ZZS)%SnowflakeConfiga. Reference: https://docs.snowflake.com/en/user-guide/snowflake-cortex/cortex-llm-rest-api Snowflake Cortex LLM REST API supports function calling with specific models (e.g., Claude 3.5 Sonnet). This config handles transformation between OpenAI format and Snowflake's tool_spec format. cs tSN)super get_config)cls __class__a/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/snowflake/chat/transformation.pyr"s zSnowflakeConfig.get_config content_listreturnc Csd}g}t|D]@\}}|ddkr||dd7}q|ddkrH|di}t|dddt|ddt|did d }||q||rO|fSd fS) a Transform Snowflake tool calls to OpenAI format. Args: content_list: Snowflake's content_list array containing text and tool_use items Returns: Tuple of (text_content, tool_calls) Snowflake format in content_list: { "type": "tool_use", "tool_use": { "tool_use_id": "tooluse_...", "name": "get_weather", "input": {"location": "Paris"} } } OpenAI format (returned tool_calls): ChatCompletionMessageToolCall( id="tooluse_...", type="function", function=Function(name="get_weather", arguments='{"location": "Paris"}') ) typetextZtool_useZ tool_use_idfunctionnameinput)r! arguments)idrr N) enumerategetr r jsondumpsappend)selfr text_content tool_callsidxZ content_itemZ tool_use_dataZ tool_callrrr._transform_tool_calls_from_snowflake_to_openai&s$    z>SnowflakeConfig._transform_tool_calls_from_snowflake_to_openaiNmodel raw_responsemodel_response logging_obj request_datamessagesoptional_paramslitellm_paramsencodingapi_key json_modec Cs|} |j|d| d|idd| vrKt| ddkrK| dd} d| vrKd| dvrK| dd}||\}}|| dd<|rF|| dd <| dd=td i| }d |jpWd|_|durc||jd <|S) NrZcomplete_input_dict)r"r8original_responseZadditional_argschoicesrmessagercontentr,z snowflake/r/r)r'Z post_calllenr.r r/Z_hidden_params)r*r/r0r1r2r3r4r5r6r7r8r9Z response_jsonchoicerr+r,Zreturned_responserrrtransform_responseYs0      z"SnowflakeConfig.transform_responseapi_basestreamcCs|||}|dS)zc If api_base is not provided, use the default DeepSeek /chat/completions endpoint. z/cortex/inference:complete)Z _get_api_base)r*rAr8r/r5r6rBrrrget_complete_urls z SnowflakeConfig.get_complete_urltoolsc Csrg}|D]2}|ddkr6|di}dd|d|ddidd i}d |vr1|d |dd <||q|S) a Transform OpenAI tool format to Snowflake tool format. Args: tools: List of tools in OpenAI format Returns: List of tools in Snowflake format OpenAI format: { "type": "function", "function": { "name": "get_weather", "description": "...", "parameters": {...} } } Snowflake format: { "tool_spec": { "type": "generic", "name": "get_weather", "description": "...", "input_schema": {...} } } rr Z tool_specZgenericr! parametersobject)r properties)rr!Z input_schema description)r&r))r*rDZsnowflake_toolstoolr Zsnowflake_toolrrr_transform_toolss$   z SnowflakeConfig._transform_tools tool_choicecCsLt|tr|St|tr$|ddkr$|did}|r$d|gdS|S)a Transform OpenAI tool_choice format to Snowflake format. Args: tool_choice: Tool choice in OpenAI format (str or dict) Returns: Tool choice in Snowflake format OpenAI format: {"type": "function", "function": {"name": "get_weather"}} Snowflake format: {"type": "tool", "name": ["get_weather"]} Note: String values ("auto", "required", "none") pass through unchanged. rr r!rI)rr!) isinstancestrdictr&)r*rKZ function_namerrr_transform_tool_choices  z&SnowflakeConfig._transform_tool_choiceheadersc Csl|ddpd}|di}|dd}|r|||d<|dd} | r,|| |d<|||d||S)NrBF extra_bodyrDrK)r/r4rB)poprJrO) r*r/r4r5r6rPrBrQrDrKrrrtransform_requests    z!SnowflakeConfig.transform_request)NNr)__name__ __module__ __qualname____doc__ classmethodrrrrMrrrr r.httpxResponser LiteLLMLoggingObjrNr boolr@rCrJrrOrS __classcell__rrrrrs >     : *4 #r)rWr'typingrrrrrrrrYZlitellm.types.llms.openair Zlitellm.types.utilsr r r Zopenai_like.chat.transformationrutilsrZ*litellm.litellm_core_utils.litellm_loggingrZ_LiteLLMLoggingObjr[rrrrrs$