o i@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.) json_schema)name description parameters)function_declarationsN) rrrrSchemafrom_json_schema JSONSchema_filter_to_supported_schema inputSchema)r rr]/var/www/html/karishye-ai-python/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;sr"cCs*tdurdS|D] }t|trdSqdS)z;Checks whether the list of tools contains any MCP sessions.NFT)r r r!rrrhas_mcp_session_usageEs r#headerscCsVtdurdSztd}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)r$ version_labelexisting_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)any_of) propertiescSrrr)rvaluerrrrksz/_filter_to_supported_schema..cSsi|] \}}|t|qSrr0rkeyr1rrr osz/_filter_to_supported_schema..csi|] \}}|vr||qSrrr2supported_fieldsrrr4ss)setrr model_fieldskeysr-r)r,schema_field_nameslist_schema_field_namesdict_schema_field_names field_name field_valuerr5rr]s&   r)__doc__importlib.metadatarrtypingrr&rr TYPE_CHECKING mcp.typesrr r%r r Type__annotations__ ImportErrorrlistr ToolListUnionboolr"r#dictstrr+ StringDictrrrrrs8