+jkUdZddlZddlZddlmZmZddlZddlmZddlZ ddl m Z ddl m Z ddl mZdd lmZejrdd lmZdd lmZdd lmZdd lmZnSeZejed<eZejed<dZeed<dZeed< dd lmZdd lmZdd lmZdd lmZn#e$r dZdZdZdZYnwxYwdedejfdZdedejfdZ d#de!ede"de!ejfdZ#dej$de"fdZ%dej$de"fdZ&de'e(e(fddfdZ)de j*de j*fdZ+ej,d ed!e(dej-efd"Z.dS)$z!Utils for working with MCP tools.N)PackageNotFoundErrorversion)Any)Request)_common)types)_MULTI_REGIONAL_LOCATIONS)Tool) ClientSession)streamable_http_client)create_mcp_http_clientMcpClientSessionMcpToolr rtoolreturnc tj|j|jtjtjdit|jdgS)z.Translates an MCP tool to a Google GenAI tool.) json_schema)name description parametersfunction_declarations) r r rrSchemafrom_json_schema JSONSchema_filter_to_supported_schema inputSchemars [/home/wildlama/visual-decline/.venv/lib/python3.11/site-packages/google/genai/_mcp_utils.pymcp_to_gemini_toolr"6sw )) 55*/0@AA6       cTtj|j|j|jdgS)z9Translates an Agent Platform tool to a Google GenAI tool.)rrparameters_json_schemar)r r rrrr s r!agent_platform_to_gemini_toolr&Es= i!-(,(8     r#Ftoolsis_agent_platformc6|r d|DSd|DS)z?Translates a list of MCP tools to a list of Google GenAI tools.c,g|]}t|Sr)r&.0rs r! z'mcp_to_gemini_tools..Xs! B B BD )$ / / B B Br#c,g|]}t|Sr)r"r+s r!r-z'mcp_to_gemini_tools..Ys! 5 5 5t T " " 5 5 5r#r)r'r(s r!mcp_to_gemini_toolsr/Rs5 C B BE B B BB 5 5u 5 5 55r#c|tdS|D]/}t|tst|trdS0dS)zDChecks whether the list of tools contains any MCP tools or sessions.NFT)r isinstancerr'rs r!has_mcp_tool_usager3\sQ 5d$  Jt5E$F$F TT r#cRtdS|D]}t|trdSdS)z;Checks whether the list of tools contains any MCP sessions.NFT)rr1r2s r!has_mcp_session_usager5fs@ 5d$()) TT r#headersctdS td}n#t$rd}YnwxYw|dd}|d|z|d<dS)z;Sets the MCP version label in the Google API client header.Nmcpz0.0.0zx-goog-api-clientz mcp_used/)rrrgetlstrip)r6 version_labelexisting_headers r!set_mcp_usage_headerr>ps FENNMM MMMKK 3R88/4]444 FHH s  **schemacttjj}|gdd}d}d}i}|D]_\}}||vrt|||<||vrd|D||<0||vr"d|D||<V||vr|||<`|S)zKFilters the schema to only include fields that are supported by JSONSchema.)additionalPropertiesanyOfoneOf$defsz$ref)itemsrAadditional_properties)rBany_ofrCone_of) propertiesdefsrDc,g|]}t|Srr)r,values r!r-z/_filter_to_supported_schema..s.%%%16 %e , ,%%%r#c4i|]\}}|t|SrrL)r,keyrMs r! z/_filter_to_supported_schema..s7%%%c5 *511%%%r#)setr r model_fieldskeysupdaterEr)r?supported_fieldsschema_field_nameslist_schema_field_namesdict_schema_field_namesfiltered_schema field_name field_values r!rr~s5 #5#3#@#E#E#G#GHH C;$&/!' 0 0j+'''$? $L$Loj!! . . .%%:E%%%oj!! . . .%%'--//%%%oj!! ' ' '$/oj! r# api_client toolset_namecKttdd}t|dr!t|jdr |jj}|r'|dr |dd}|d|}n7t |dd }|d krd |}n|tvr d |d |}nd |d|}|d{V}t |dd}i}t|dr,|jr%|jj rt|jj }d||d<|r||d<t|tj |d} |4d{Vt||4d{V } | \} } } t| | 4d{V} | d{V | WVnD#t $r7Ydddd{Vdddd{Vdddd{VdSwxYw dddd{Vn#1d{VswxYwYdddd{Vn#1d{VswxYwYdddd{VdS#1d{VswxYwYdS#t"$rd}gdt$ddffd t'|jdvst|dr!|t+d|d|Yd}~dSd}~wwxYw)zGInternal helper to manage the Agent Platform MCP lifecycle per request.Nz@The 'mcp' package is required to use Agent Platform MCP servers. _http_optionsbase_url/z/mcp/locationglobalz&https://aiplatform.googleapis.com/mcp/zhttps://aiplatform.z.rep.googleapis.com/mcp/zhttps://z-aiplatform.googleapis.com/mcp/projectzBearer AuthorizationzX-Goog-User-Project)r6timeout)url http_clientexcrc t|dr|jD] }|dSt|jdt |}t|dr|jt |jdt |jdd}t |jdt |j}t|rt |j}|d|d|d z }|dS) N exceptionsz: response status_codestatusUnknowntextz (HTTP )) hasattrrltype__name__strrmgetattrcallableappend)rjemsgrorq_extract_errorserror_messagess r!r|z4_connect_agent_platform_mcp.._extract_errorss l # ##  A /!      c#11s3xx11 3 # # - (@lclHi88  & vs3httpx AsyncClientr initialize GeneratorExit BaseExceptionrrtru ValueError)r\r]r`mcp_urlrctokenrer6ristreams read_stream write_stream_sessionegr|r}s @@r!_connect_agent_platform_mcprs# H  ( Z))1gj6NPZ.[.[1'0H S#2#h.. ..GGz:x88H8GGGgg . . .VhVV VVggR8RRLRRg..00 0 0 0 0 0 0% J 4 0 0' ' Z))5j.F5:KcKk5:+344G.u..'/ -%,G !"w!'4@@@++        ';         '.$ \1#K>>       '""$$ $ $ $ $ $ $ $ MMMMM                                                                                                                                  N#S#T#######& BxxDDD LIIDob  1w 1 1 . 1 1   EDDDDD/s II -H& H&F,+H, G- 6H7H&I I,G- -H1H& H H&H H&I & H0 0I 3H0 4I 7I IIII K 'AKK )F)/__doc__ contextlibrimportlib.metadatarrtypingr google.authgooglegoogle.auth.transport.requestsrr9rr _api_clientr TYPE_CHECKING mcp.typesr rr8r rmcp.client.streamable_httpr mcp.shared._httpx_utilsrType__annotations__rr"r&listboolr/ ToolListUnionr3r5rrvr> StringDictrasynccontextmanager AsyncIteratorrrr#r!rsP ('' <<<<<<<< 222222222222 "''''''333333??????<<<<<<<"%FK%%%'6; $#$$$ $#$$$ "))))))555555AAAAAA>>>>>>> """G!! " W       EJ    $66 =66 %*6666e1d!4  $sCx.  T     "  " """"J R#RSRVMabeMfRRR RRRsB B*)B*