o Yi@sBUdZddlmZmZddlZddlmZddlmZddlmZej r1ddl m Z dd l mZn*eZejed <eZ ejed <zddl m Z dd l mZWn eyZdZ dZYnwd e d ej fddZdee d eej fddZdejd efddZdejd efddZdeeefd dfddZdejd ejfddZdS)z!Utils for working with MCP tools.)PackageNotFoundErrorversionN)Any)_common)types)Tool) ClientSessionMcpClientSessionMcpTooltoolreturnc Cs6tj|j|jtjjtjdit|jddgdS)z.Translates an MCP tool to a Google GenAI tool.)Z json_schema)name description parameters)Zfunction_declarationsN) rrrrZSchemaZfrom_json_schema JSONSchema_filter_to_supported_schemaZ inputSchema)r rrQ/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/_mcp_utils.pymcp_to_gemini_tool'srtoolscCsdd|DS)z?Translates a list of MCP tools to a list of Google GenAI tools.cSg|]}t|qSr)r).0r rrr 8sz'mcp_to_gemini_tools..r)rrrrmcp_to_gemini_tools6srcCs4tdurdS|D]}t|tst|trdSqdS)zDChecks whether the list of tools contains any MCP tools or sessions.NFT)r isinstancer rr rrrhas_mcp_tool_usage;srcCs*tdurdS|D] }t|trdSqdS)z;Checks whether the list of tools contains any MCP sessions.NFT)r rrrrrhas_mcp_session_usageEs rheaderscCsVtdurdSztd}Wn tyd}Ynw|dd}|d||d<dS)z;Sets the MCP version label in the Google API client header.Nmcpz0.0.0zx-goog-api-clientz mcp_used/)r rrgetlstrip)rZ version_labelZexisting_headerrrrset_mcp_usage_headerOs     r$schemacsttjjd}d}d}|D],\}}||vr!t|||<q||vr/dd|D||<q||vr>dd|D||<qfdd|DS) zKFilters the schema to only include fields that are supported by JSONSchema.)items)Zany_of) propertiescSrrr)rvaluerrrrksz/_filter_to_supported_schema..cSsi|] \}}|t|qSrr(rkeyr)rrr osz/_filter_to_supported_schema..csi|] \}}|vr||qSrrr*Zsupported_fieldsrrr,ss)setrrZ model_fieldskeysr&r)r%Zschema_field_namesZlist_schema_field_namesZdict_schema_field_names field_nameZ field_valuerr-rr]s&   r)__doc__importlib.metadatarrtypingrr!rr TYPE_CHECKINGZ mcp.typesrr r r r Type__annotations__ ImportErrorrlistrZ ToolListUnionboolrrdictstrr$Z StringDictrrrrrs8