LiUdZddlmZmZddlZddlmZddlmZddlmZejr ddl m Z dd l mZn/eZej ed <eZ ej ed < ddl m Z dd l mZd e d ejfdZdee d eejfdZdej,d efdZdej,d efdZdeeefd dfdZdej:d ej:fdZy#e$rdZ dZYwxYw)z!Utils for working with MCP tools.)PackageNotFoundErrorversionN)Any)_common)types)Tool) ClientSessionMcpClientSessionMcpTooltoolreturnc tj|j|jtjj tj dit|jdgS)z.Translates an MCP tool to a Google GenAI tool.) json_schema)name description parameters)function_declarations) rr rrSchemafrom_json_schema JSONSchema_filter_to_supported_schema inputSchema)r s U/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/google/genai/_mcp_utils.pymcp_to_gemini_toolr'se )))) 55**/0@0@A6    toolsc>|Dcgc] }t|c}Scc}w)z?Translates a list of MCP tools to a list of Google GenAI tools.)rrr s rmcp_to_gemini_toolsr!6s/4 5t T " 55 5scdty|D]$}t|tst|ts$yy)zDChecks whether the list of tools contains any MCP tools or sessions.FT)r isinstancer r s rhas_mcp_tool_usager$;s6 d$ Jt5E$F  rcDty|D]}t|tsyy)z;Checks whether the list of tools contains any MCP sessions.FT)r r#r s rhas_mcp_session_usager&Es- d$()  rheaderscty td}|jdd}|d|zj |d<y#t$rd}Y9wxYw)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_headers rset_mcp_usage_headerr/Osc ENMKK 3R8/*]O44 FH  Ms A A Aschemac ttjjj }d}d}d}|j D]n\}}||vrt |||<||vr|Dcgc] }t |c}||<:||vs?|j Dcic]\}}|t |c}}||<p|j Dcic] \}}||vs ||c}}Scc}wcc}}wcc}}w)zKFilters the schema to only include fields that are supported by JSONSchema.)items)any_of) properties)setrr model_fieldskeysr2r) r0supported_fieldsschema_field_nameslist_schema_field_namesdict_schema_field_names field_name field_valuevaluekeys rrr]s #5#3#3#@#@#E#E#GH#-))8!' j+''6{CfZ . .:E16 %e ,fZ . .(--/c5 *51 1fZ $*<<> S%SrNs (<  '3"%FKK%'6;;)5 W  6tG}6ejj1A6 e11d!4!4  $sCx.  T      w Gs C"" C.-C.