ciUdZddlZddlZddlZddlmZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZddl Z ddl!Z!ddl"m#Z#ddl$mZ%ddl&m'Z'ddl&m(Z(d dl)mZ*e jVd k\rdd lm,Z,ndd l-m,Z, e j\Z/ee0e1d < ejfZ4ee0ee0fe1d < e jjZ6e7e0e1d<ejprddl9m:Z:e:jvZd?eZyejGd@dAeZzejGdBdCeZ{ejGdDdEeZ|ejprddFl}m~Z~e~Ze,e1dG<n ddFl}m~Z~e~Zee,e1dG<eeewexe|eveyezfZGdHdIedJKZGdLdMe{e|ZGdNdOeZGdPdQeZGdRdSeZee*j e*je*jfZGdTdUeZGdVdWeZddXe0dYe0dZe0fd[Zd\d]dXe0d^ed_edZefd`Zd\d]dXe0d^ed_edZefdaZddbdcee0e0fddeee0ee0edffdeeee0dZefdfZdgedhee0ee0fdZee!jfdiZdjeee0efdZee!jfdkZdlej(dZefdmZdigdndoedpefdqee0dree0e0fdsee0dZee0eff dtZdee0dgedZee0effduZddvdwe0dxe0dye0dzeedZee2f$reZe2f$reZBe=ee1d<YwxYw#e>e2f$rdZDdZEdZFdZJdZKdZGdZHYwxYw#e>e2f$rdZYwxYw)z$Utility functions for agent engines.N)metadata)Any AsyncIteratorCallable CoroutineDictIteratorListMappingOptionalProtocolSequenceSet TypedDictUnion) exceptions)types) struct_pb2) json_format) ) TypeAlias_BUILTIN_MODULE_NAMES_PACKAGE_DISTRIBUTIONS_STDLIB_MODULE_NAMESstorage_StorageBucket _SpecifierSet) AgentCardTransportProtocolMessage TaskIdParamsTaskQueryParams) ClientConfig ClientFactoryactionsappendagent_frameworkasync async_stream bidi_streamzagent_engine.pklcustom)z google-adk langchain langgraphag2z llama-indexr. async_queryzCoroutine[Any]async_stream_queryzAsyncIterable[Any] agent_enginea Runs the Agent Engine to serve the user request. This will be based on the `.{method_name}(...)` of the python object that was passed in when creating the Agent Engine. The method will invoke the `{default_method_name}` API client of the python object. Args: **kwargs: Optional. The arguments of the `.{method_name}(...)` method. Returns: {return_type}: The response from serving the user request. queryzdict[str, Any]z Iterable[Any] cloudpicklepydantic stream_querybidi_stream_queryzdependencies.tar.gzzFailed to register API methods. Please follow the guide to register the API methods: https://cloud.google.com/vertex-ai/generative-ai/docs/agent-engine/develop/custom#custom-methods. Error: {%s}installation_scriptsnameapi_modezrequirements.txtstream a2a_extensiona2a_agent_cardwarningsmissing incompatiblezvertexai_genai.agentenginesc4eZdZdZej dZy) Queryablez/Protocol for Agent Engines that can be queried.c y)z.Runs the Agent Engine to serve the user query.Nselfkwargss K/tmp/pip-target-z3e9_cxr/lib/python/vertexai/_genai/_agent_engines_utils.pyr5zQueryable.queryN)__name__ __module__ __qualname____doc__abcabstractmethodr5rGrMrKrErEs9==rMrEc4eZdZdZej dZy)AsyncQueryablez>Protocol for Agent Engines that can be queried asynchronously.c y)z=Runs the Agent Engine to serve the user query asynchronously.NrGrHs rKr2zAsyncQueryable.async_queryrLrMN)rNrOrPrQrRrSr2rGrMrKrUrUsHLLrMrUc@eZdZdZej deefdZy)AsyncStreamQueryablezDProtocol for Agent Engines that can stream responses asynchronously.returnc Kyw)z8Asynchronously stream responses to serve the user query.NrGrHs rKr3z'AsyncStreamQueryable.async_stream_queryN) rNrOrPrQrRrSrrr3rGrMrKrXrXs,NGM#4FGGrMrXc@eZdZdZej deefdZy)StreamQueryablez5Protocol for Agent Engines that can stream responses.rYc y)z)Stream responses to serve the user query.NrGrHs rKr8zStreamQueryable.stream_queryrLrMN) rNrOrPrQrRrSr rr8rGrMrKr^r^s(?8 88rMr^c^eZdZdZej dejede efdZ y)BidiStreamQueryablezBProtocol for Agent Engines that can stream requests and responses. input_queuerYc Kyw)z8Stream requests and responses to serve the user queries.NrG)rIrbs rKr9z%BidiStreamQueryable.bidi_stream_queryr[r\N) rNrOrPrQrRrSasyncioQueuerrr9rGrMrKraras?LG"==-G s GGrMrac:eZdZdZej defdZy) Cloneablez.Protocol for Agent Engines that can be cloned.rYcy)zReturn a clone of the object.NrGrIs rKclonezCloneable.clone rLrMN)rNrOrPrQrRrSrrjrGrMrKrgrgs$8,s,,rMrgcNeZdZdZej dedeee effdZ y)OperationRegistrablez4Protocol for agents that have registered operations.rJrYc y)z:Register the user provided operations (modes and methods).NrGrHs rKregister_operationsz(OperationRegistrable.register_operations rMN) rNrOrPrQrRrSrdictstrlistrnrGrMrKrlrls8> C Dd3i4H  rMrl BaseAgentADKAgentc\eZdZUeed<eed<eeeefed<eeeed<e ed<y)_ModuleAgentAttributes module_name agent_namern sys_pathsagentN) rNrOrPrq__annotations__rrrr r_AgentEngineInterfacerGrMrKrwrw0s5Oc49n-- &&  rMrwF)totalc zeZdZdZdddededeeeefdeeefdZ dd Z d e d e eeeffd Z dd Zy) ModuleAgentzAgent that is defined by a module and an agent name. This agent is instantiated by importing a module and instantiating an agent from that module. It also allows to register operations that are defined in the agent. N)rzrxryrnrzc||||d|_y)a{Initializes a module-based agent. Args: module_name (str): Required. The name of the module to import. agent_name (str): Required. The name of the agent in the module to instantiate. register_operations (Dict[str, list[str]]): Required. A dictionary of API modes to a list of method names. sys_paths (Sequence[str]): Optional. The system paths to search for the module. It should be relative to the directory where the code will be running. I.e. it should correspond to the directory being passed to `extra_packages=...` in the create method. It will be appended to the system path in the sequence being specified here, and only be appended if it is not already in the system path. rxryrnrzN) _tmpl_attrs)rIrxryrnrzs rK__init__zModuleAgent.__init__@s4'$#6" 4 rMrYct|jjd|jjd|jjd|jjdS)zReturn a clone of the agent.rxryrnrzr)rrgetris rKrjzModuleAgent.clone`s`((,,];''++L9 $ 0 0 4 45J K&&**;7   rMrJc V|jjd}| td|S)NrnzRegister operations is not set.)rr ValueError)rIrJreg_operationss rKrnzModuleAgent.register_operationsis1))--.CD  !>? ?rMc |jjd}t|trTddl}|D]K}t j j|}||j vs1|j j|Mddl }|j|jjd} |j||jjd} t||} | |jd <t#| d r| j%|j'j)D]7} | D]0} t+| | } t-|| t/j0| |29y#t$r>}tjd|jjdd|Yd}~d}~wwxYw#t $r0}t!d|d |jjd|d}~wwxYw) zSets up the agent for execution of queries at runtime. It runs the code to import the agent from the module, and registers the operations of the agent. rzrNrxzFailed to reload module : ryzAgent z not found in module r{set_up)r{ operation)rr isinstancersysospathabspathr) importlib import_modulereload ExceptionloggerwarninggetattrAttributeErrorhasattrrrnvalues_wrap_agent_operationsetattrr MethodType) rIrzrsys_pathabs_pathrmoduleeryr{ operationsrops rKrzModuleAgent.set_upos $$((5 i * % .77??84388+HHOOH- . (()9)9)=)=m)LM    V $ %%)),7  FJ/E %*! 5( # LLN224;;= EJ' E *)Li)9)9"d)CD E E  NN*4+;+;+?+? +N*OrRSQTU      $9##'' 679  s0/E, F6, F354F..F36 G/?+G**G/)rYr)rYN)rNrOrPrQrqrrrr rrrjrrprnrrGrMrKrr8s{.2     "#tCy.1  HSM*  @ CDd3i4H &ErMrceZdZUeeed<y)_RequirementsValidationActionsr)NrNrOrPrrqr|rGrMrKrrs HrMrc.eZdZUeeed<eeed<y)_RequirementsValidationWarningsrBrCNrrGrMrKrrs Xc(rMrc"eZdZUeed<eed<y)_RequirementsValidationResultrAr(N)rNrOrPrr|rrGrMrKrrs-- ++rMrc eZdZdededefdZy)GetOperationFunctionoperation_namerJrYc yNrGrIrrJs rK__call__zGetOperationFunction.__call__rorMNrNrOrPrqrAgentEngineOperationUnionrrGrMrKrr !$ 03 " rMrc eZdZdededefdZy)GetAsyncOperationFunctionrrJrYc KywrrGrs rKrz"GetAsyncOperationFunction.__call__s  r\NrrGrMrKrrrrMrr resource_namerYc|s |s td|r8tjd|}|r|jdStd|d|s tdtjd|}|r|jdStd|d) zAReturns reasoning engine ID from operation name or resource name.z0Resource name or operation name cannot be empty.z9^projects/[^/]+/locations/[^/]+/reasoningEngines/([^/]+)$rz9Failed to parse reasoning engine ID from resource name: ``zOperation name cannot be empty.zJ^projects/[^/]+/locations/[^/]+/reasoningEngines/([^/]+)/operations/[^/]+$z:Failed to parse reasoning engine ID from operation name: `)rrematchgroup)rrrs rK_get_reasoning_engine_idrs KLL H   ;;q> !!?!%  :;; HHU E {{1~   1  rMr)poll_interval_secondsget_operation_fnrcK||d{}|jsEtj|d{||jd{}|jsE|S7W707w)aFWaits for the operation for creating an agent engine to complete. Args: operation_name (str): Required. The name of the operation for creating the Agent Engine. poll_interval_seconds (float): The number of seconds to wait between each poll. get_operation_fn (Callable[[str], Awaitable[Any]]): Optional. The async function to use for getting the operation. If not provided, `self._get_agent_operation` will be used. Returns: The operation that has completed (i.e. `operation.done==True`). rN)donerdsleepr;rrrrs rK_await_async_operationrs_('nEEInnmm1222*)..II nn  F2Is7 A-A'(A-A)A-A+A-%A-)A-+A-c||}|js5tj|||j}|js5|S)a5Waits for the operation for creating an agent engine to complete. Args: operation_name (str): Required. The name of the operation for creating the Agent Engine. poll_interval_seconds (float): The number of seconds to wait between each poll. get_operation_fn (Callable[[str], Any]): Optional. The function to use for getting the operation. If not provided, `self._get_agent_operation` will be used. Returns: The operation that has completed (i.e. `operation.done==True`). r)rtimerr;rs rK_await_operationrsA(!?Inn ()$INNC nn rM)required_packages requirements constraintsrc >t}|t}ttt t t t }t |tr t|}|jD]\}}||vrH|ttj|||vr$|ttj|d|R|sU||}|s]|j||vsq|tt j|d|dt#|d|S)aCompares the requirements with the constraints. Args: requirements (Mapping[str, str]): Required. The packages (and their versions) to compare with the constraints. This is assumed to be the result of `scan_requirements`. constraints (Union[Sequence[str], Mapping[str, SpecifierSet]]): Required. The package constraints to compare against. This is assumed to be the result of `parse_constraints`. required_packages (Iterator[str]): Optional. The set of packages that are required to be in the constraints. It defaults to the set of packages that are required for deployment on Agent Engine. Returns: dict[str, dict[str, Any]]: The comparison result as a dictionary containing: * warnings: * missing: The set of packages that are not in the constraints. * incompatible: The set of packages that are in the constraints but have versions that are not in the constraint specifier. * actions: * append: The set of packages that are not in the constraints but should be appended to the constraints. )rBrC)r))rAr(rz==z (required: ))"_import_packaging_version_or_raise_DEFAULT_REQUIRED_PACKAGESrrsetrrrr_parse_constraintsitems _WARNINGS_KEY_WARNING_MISSINGadd _ACTIONS_KEY_ACTION_APPENDVersion_WARNING_INCOMPATIBLErq)rrrpackaging_versionresultpackagepackage_versionpackage_specifiers rK_compare_requirementsrs+<;< 6 *0SUS.ce<F+t$([A $0$6$6$8  + % = !"2 3 7 7 @++|$^488ir/!23   +G 4 $ ((9ARR}%&;<@@ir/!2,sCT?U>VVWX MrMr{rc `t|tr!|j} |jg}|j D]\}}|D]}t||st d|dt||} t||}t|} || t<t|dr!t|dj| t<|j!| |S#t$r}t d|d||d}~wwxYw#t$r%}tjd|d|Yd}~d}~wwxYw) a}Generates a ReasoningEngineSpec based on the registered operations. Args: agent: The AgentEngine instance. operations: A dictionary of API modes and method names. Returns: A list of ReasoningEngineSpec.ClassMethod messages. Raises: ValueError: If a method defined in `register_operations` is not found on the AgentEngine. zFailed to set up agent rNzMethod `z6` defined in `register_operations` not found on agent.) schema_namezfailed to generate schema for agent_card)rrrjrrrrrr_generate_schemarr _to_proto_MODE_KEY_IN_SCHEMAmodel_dump_json_A2A_AGENT_CARDr)) r{rrclass_methods_specmode method_names method_namemethod schema_dict class_methods rK%_generate_class_methods_spec_or_raiserNsZ$%%#[[] L LLN(..04l' 4K5+. {m,++ UK0F .v;O %[1L04L, -ul+07<1!/#_-  % %l 3) 44. 5 L6ugRsCD! K L !? }BqcRS s/C7 C? C<%C77C<? D-D((D- class_methodsc>|Dcgc] }t|c}Scc}w)zWConverts a list of class methods to a list of ReasoningEngineSpec.ClassMethod messages.)r)rrs rK$_class_methods_to_class_methods_specrs9F F Il # FF Fsparamc$|jtjjk(rydt |jvryt |jt ryt} |j|jy#t$rYywxYw)z1Checks if the parameter is pydantic serializable.T ForwardRefF) annotationinspect Parameteremptyreprrrq_import_pydantic_or_raise TypeAdapterr)rr7s rK_is_pydantic_serializablers 7,,222tE,,--%""C((*HU--. s'B BB)r descriptionsrequiredf.rrrc :t}ttj|j}|j Dcic]\}}|j tjjtjjtjjfvr|jtjk7r~t|rs||jtjjk7r#dt!|jvr |jnt"|j%|j'|df}}}|j(|j*fi|j-} | j/dd| j'dij D]\}} | j/dd||j} t1j2| t4usCt7dt1j8| vsd| j/dgD]"} | j'd} | s| d k7s| | d<nd | d <|r|| d <n|Dcgc]}||j:tjjk(r^||j tjjtjjtjjfvr|c}| d <t|j*|j<| } |r|| d<| Scc}}wcc}w)a/Generates the OpenAPI Schema for a callable object. Only positional and keyword arguments of the function `f` will be supported in the OpenAPI Schema that is generated. I.e. `*args` and `**kwargs` will not be present in the OpenAPI schema returned from this function. For those cases, you can either include it in the docstring for `f`, or modify the OpenAPI schema returned from this function to include additional arguments. Args: f (Callable): Required. The function to generate an OpenAPI Schema for. schema_name (str): Optional. The name for the OpenAPI schema. If unspecified, the name of the Callable will be used. descriptions (Mapping[str, str]): Optional. A `{name: description}` mapping for annotating input arguments of the function with user-provided descriptions. It defaults to an empty dictionary (i.e. there will not be any description for any of the inputs). required (Sequence[str]): Optional. For the user to specify the set of required arguments in function calls to `f`. If specified, it will be automatically inferred from `f`. Returns: dict[str, Any]: The OpenAPI Schema for the function `f` in JSON format. rN) descriptiontitler= propertiesanyOftypenullTnullabler)r;r parametersr;)rrpr signaturer rkindrPOSITIONAL_OR_KEYWORD KEYWORD_ONLYPOSITIONAL_ONLYrrdrerrrrFieldr create_modelrNschemapoptyping get_originrr get_argsdefaultrQ)rrrrr7defaultsr;r fields_dictr  function_argrr schema_typeks rKrrsD)*HG%%a(334H*$>>+)( D% ::    3 3    * *    - -      -2KE2R; ##w'8'8'>'>> U-=-=(>>   NN),,T48    K@'&&qzzA[AHHJJ NN7B(nn\2>DDF,l"%d^..   Z (E 1d4jFOO E 7 '**7B7 $jj0 ;+6L(   (,L $',*!) :  "  ##w'8'8'>'>>QK$$%%;;%%22%%55  " :qzzqyyZ PF$v Ma@ " sC>LBLc>|"|tvrtjd||St|trDt |t}|2t |tr"|tvrtjd||Stjd|dtdtS)aGets the agent framework to use. The agent framework is determined in the following order of priority: 1. The `agent_framework` passed to this function. 2. The `agent_framework` attribute on the `agent` object. 3. The default framework, "custom". Args: agent_framework (str): The agent framework provided by the user. agent (_AgentEngineInterface): The agent engine instance. Returns: str: The name of the agent framework to use. zUsing agent framework: zThe provided agent framework z! is not supported. Defaulting to .) _SUPPORTED_AGENT_FRAMEWORKSrinfor_AGENT_FRAMEWORK_ATTRrrrq_DEFAULT_AGENT_FRAMEWORK)r*r{agent_framework_attrs rK_get_agent_frameworkr's*":U'U -o->?@u+,&u.CD ,/5$(CC KK12F1GH I' ' KK ''89231 6 $#rM) credentialsprojectlocationstaging_bucketr(clt}|j||}|jdd} |j|}tj d||S#t j$rC|j|}|j||}tj d|d|Y|SwxYw)zGets or creates the GCS bucket.)r)r(gs://r=z Using bucket )r*zCreating bucket z in location=) _import_cloud_storage_or_raiseClientreplace get_bucketrr#rNotFoundbucket create_bucket)r)r*r+r(rstorage_client gcs_bucket new_buckets rK_get_gcs_bucketr8=s-.G#W^^GMN#++GR8NH#..~>  mN#345    H#**>: #11*x1P  &~&6n8+FG  Hs)AAB32B3cvt|tr|jSi}t|trtg|t <t|t rtg|t<t|trtg|t<t|trtg|t<t|trt g|t"<|S)z2Retrieves registered operations for a AgentEngine.)rrlrnrE_DEFAULT_METHOD_NAME_STANDARD_API_MODErU_DEFAULT_ASYNC_METHOD_NAME_ASYNC_API_MODEr^_DEFAULT_STREAM_METHOD_NAME_STREAM_API_MODErX!_DEFAULT_ASYNC_STREAM_METHOD_NAME_ASYNC_STREAM_API_MODEra _DEFAULT_BIDI_STREAM_METHOD_NAME_BIDI_STREAM_API_MODE)r{rs rK_get_registered_operationsrDRs %-.((**J%#*>)? %&%('A&B ?#%)(C'D #$%-..O-P )*%,--M,N () rMcJ ddl}|S#t$r}td|d}~wwxYw)z'Tries to import the cloudpickle module.rNz_cloudpickle is not installed. Please call 'pip install google-cloud-aiplatform[agent_engines]'.)r6 ImportError)r6rs rK_import_cloudpickle_or_raiserGhs<   D  s " "cN ddlm}|S#t$r}td|d}~wwxYw)z)Tries to import the Cloud Storage module.rrzaCloud Storage is not installed. Please call 'pip install google-cloud-aiplatform[agent_engines]'.N) google.cloudrrF)rrs rKr.r.ts;( N  D   $ $cN ddlm}|S#t$r}td|d}~wwxYw)2Tries to import the packaging.requirements module.rrzjpackaging.requirements is not installed. Please call 'pip install google-cloud-aiplatform[agent_engines]'.N) packagingrrF)rrs rK'_import_packaging_requirements_or_raiserOs<*   D  rJcN ddlm}|S#t$r}td|d}~wwxYw)rLr)versionzepackaging.version is not installed. Please call 'pip install google-cloud-aiplatform[agent_engines]'.N)rNrQrF)rQrs rKrrs;% N  D  rJc ddl}|j}|S#t$r ddlm}Y|St$r}t d|d}~wwxYw)z$Tries to import the pydantic module.rN)v1z\pydantic is not installed. Please call 'pip install google-cloud-aiplatform[agent_engines]'.)r7rrrSrF)r7_rs rKrrsX  NN O ,+ O  D  s?? :?cLt}i}|D]a} |jdrtjj |}|j |}|jxsd||j<c|S#t $r%}tjd|d|Yd}~d}~wwxYw)aAParses a list of constraints into a dict of requirements. Args: constraints (list[str]): Required. The list of package requirements to parse. This is assumed to come from the `requirements.txt` file. Returns: dict[str, SpecifierSet]: The specifiers for each package. z.whlzFailed to parse constraint: . Exception: N) rOendswithrrbasename Requirementrrr specifierr;)rrr constraint requirementrs rKrrs;K$/#8#8#@D{ A M   NN9*]STRUV W  sAA55 B#>BB#extra_packages gcs_dir_namec|yt||||}t||||t||||t|||yy)aPrepares the agent engine for creation or updates in Vertex AI. This involves packaging and uploading artifacts to Cloud Storage. Note that 1. This does not actually update the Agent Engine in Vertex AI. 2. This will only generate and upload a pickled object if specified. 3. This will only generate and upload the dependencies.tar.gz file if extra_packages is non-empty. Args: agent: The agent engine to be prepared. requirements (Sequence[str]): The set of PyPI dependencies needed. extra_packages (Sequence[str]): The set of extra user-provided packages. project (str): The project for the staging bucket. location (str): The location for the staging bucket. staging_bucket (str): The staging bucket name in the form "gs://...". gcs_dir_name (str): The GCS bucket directory under `staging_bucket` to use for staging the artifacts needed. credentials: The credentials to use for the storage client. N)r)r*r+r()r{r6r^)rr6r^)r]r6r^)r8_upload_agent_engine_upload_requirements_upload_extra_packages) r{rr]r)r*r+r^r(r6s rK_preparercsl< } % J ! %!% !)!% "rM)wrap_operation_fnrdc .|j}|sy|D]}t|vrtd|dtd|jt}|tk(rAt |vrtd|dt d|jt }t |tstdt d||jdtj|tj|ttj|t }ttt t"t$t&t(t*t,t.i}t |t0r ||vr||}nK||vr||}nAd j3d t5|j7D} td |d | d|t,k(r |jt8} ||| } n ||} || _|rt |tr|| _t?||tAjB| |y)aRegisters Agent Engine API methods based on operation schemas. This function iterates through operation schemas provided by the `agent_engine`. Each schema defines an API mode and method name. It dynamically creates and registers methods on the `agent_engine` to handle API calls based on the specified API mode. Currently, only standard API mode `` is supported. Args: agent_engine: The AgentEngine to augment with API methods. wrap_operation_fn: A dictionary of API modes and method wrapping functions. Raises: ValueError: If the API mode is not supported or if the operation schema is missing any required fields (e.g. `api_mode` or `name`). NzOperation schema z does not contain an `z` field.z does not contain a `z-Operation schema has a non-string value for `z`: r)rdefault_method_name return_typez, c3(K|] }d|d yw)rNrG).0rs rK z1_register_api_methods_or_raise..Hs, $!D6 ,szUnsupported api mode: `z`, Supported modes are: r!)rr)r)"operation_schemasrrrrC_METHOD_NAME_KEY_IN_SCHEMArrq"_DEFAULT_METHOD_DOCSTRING_TEMPLATEformat_DEFAULT_METHOD_NAME_MAPr:_DEFAULT_METHOD_RETURN_TYPE_MAP_DEFAULT_METHOD_RETURN_TYPEr;_wrap_query_operationr=_wrap_async_query_operationr?_wrap_stream_query_operationrA"_wrap_async_stream_query_operation_A2A_EXTENSION_MODE_wrap_a2a_operationrpjoinsortedkeysrrNrQrrr) r4rdrkoperation_schemar<rmethod_description_wrap_operation_map_wrap_operationsupported_api_modesrrs rK_register_api_methods_or_raisers_0%668 -CS &6 6#$4#56 34H> $''(;< , ,  %-= =#$4#569:(D '**+EF +s+/0K=B .11  . 5 5'$<$@$@2%<??/ 6     5 8 : "$F !4   ' .8?P3P/9O , ,1(;O"&)),(./B/G/G/I(J,# )(4))<(=Q@  * *)--o>J$'JF%=F% *-?"E/FN k5+;+;FL+QRGCSrM)ignore_modulespackage_distributionsinspect_getmembers_kwargsobjrrrc|t}|t}tt}|xsi}t j |fi|D]{\}}|r!t j |s t|ds*|jxsdjdd}|sO||vsT|j|gD]}|j|}|Dcic]}|tj|c}Scc}w)aScans the object for modules and returns the requirements discovered. This is not a comprehensive scan of the object, and only detects for common cases based on the members of the object returned by `dir(obj)`. Args: obj (Any): Required. The object to scan for package requirements. ignore_modules (Sequence[str]): Optional. The set of modules to ignore. It defaults to the set of built-in and stdlib modules. package_distributions (Mapping[str, Sequence[str]]): Optional. The mapping of module names to the set of packages that contain them. It defaults to the set of packages from `importlib_metadata.packages_distributions()`. inspect_getmembers_kwargs (Mapping[str, Any]): Optional. The keyword arguments to pass to `inspect.getmembers`. It defaults to an empty dictionary. Returns: Sequence[str]: The list of requirements that were discovered. rOr=r!r) _BASE_MODULESrrrr getmembers isbuiltinrrOsplitrrimportlib_metadatarQ) rrrr modules_foundrTattrrxrs rK_scan_requirementsr^s:&$ 623M 9 ?R%%cG-FG*4w((.gdL6Q ,"33C8; ;n</33KD *!!&) * *FS S6F&..v6 6 SS Ss>Cmessagec tjtj|jd}|S#t $r.tjtj|d}Y|SwxYw)aConverts the contents of the protobuf message to JSON format. Args: message (proto.Message): Required. The proto message to be converted to a JSON dictionary. Returns: dict[str, Any]: A dictionary containing the contents of the proto. T)preserving_proto_field_name)jsonloadsr MessageToJson_pbr)rrs rK_to_dictrsn !%  % % ,0 "  M  !%  % %,0 "  M s483A/.A/c|tj}t|tjtjfr|S t j ||j|S#t$rt j ||Y|SwxYw)a<Parses a JSON-like object into a message. If the object is already a message, this will return the object as-is. If the object is a JSON Dict, this will parse and merge the object into the message. Args: obj (Union[dict[str, Any], proto.Message]): Required. The object to convert to a proto message. message (proto.Message): Optional. A protocol buffer message to merge the obj into. It defaults to Struct() if unspecified. Returns: proto.Message: The same message passed as argument. ) rStructrprotor#r ParseDictrr)rrs rKrrs{(##%# z'8'89: ,c7;;/ N ,c7+ N,s A&&B B r6ct}|j|dt}|jd5} |j || ddd|jd5} |j|} dddd |jd|} tjd | dty#t $r}d}t d|d|d}~wwxYw#1swYxYw#t $r}t d|d}~wwxYw#1swYxYw) z Uploads the agent engine to GCS./wbzkhttps://cloud.google.com/vertex-ai/generative-ai/docs/agent-engine/develop/custom#deployment-considerationsz(Failed to serialize agent engine. Visit z for details.Nrbz,Agent engine serialized to an invalid formatr-z Wrote to ) rGblob_BLOB_FILENAMEopendumpr TypeErrorloadr;rr#) r{r6r^r6rrrurlrTdir_names rKr`r`s#/0K ??l^1^,<= >D 4A    UA & 4SA S  #AS z'q7H KK)H:Q~&678 C:3%}M   SJKQR R SSSsRCB2$C>&C!2 C;C  CCC! C;* C66C;;C>>Dc|j|dt}|jdj|d|jd|}t j d|dty)z%Uploads the requirements file to GCS.r r- Writing to N)r_REQUIREMENTS_FILEupload_from_stringrxr;rr#)rr6r^rrs rKrarash ??l^1-?,@A BDDIIl34z'q7H KK+hZq);(<=>rMctjdtj}t j |d5}|D]}|j | ddd|jd|j|dt}|j|jd|jd|}tjd|dty#1swYxYw) zUploads extra packages to GCS.z,Creating in-memory tarfile of extra_packagesw|gzfileobjrNrrr-r) rr#ioBytesIOtarfilerrseekr_EXTRA_PACKAGES_FILErreadr;)r]r6r^ tar_fileobjtarfilerrs rKrbrbs KK>?**,K k 73" D GGDM Q ??l^1-A,BC DDK,,./z'q7H KK+hZq)=(>?@s CC(source_packagesc ztjdtj}tj j t j}tj|d5}|D]k}tj j |}||k7r4|j|tjzstd|d|d|j|m ddd|jd|j}t!j"|j%d S#1swYNxYw) z:Creates a base64 encoded tarball from the source packages.z-Creating in-memory tarfile of source_packagesrrz File path 'z$' is outside the project directory 'z'.Nrzutf-8)rr#rrrrrealpathgetcwdrr startswithseprrrrbase64 b64encodedecode)rr project_dirrrreal_file_path tarball_bytess rK_create_base64_encoded_tarballrs   KK?@**,K''""299;/K k 7 3# DWW--d3N,^5N5Nbff$6!!$(#}B( GGDM  Q$$&M   M * 1 1' ::  s 2A1D11D:) build_optionspackagesrc|xsg}|rt|vrt|t||D]/}tjj |r#t d||S)z Tries to validates the packages.) script_pathsrz)Package specified but not found: package=)_INSTALLATION_SUBDIR'_validate_installation_scripts_or_raiserrexistsFileNotFoundError)rrrs rK_validate_packages_or_raisersi ~2H->/&';< Tww~~g&#&Pz$RS ST OrMrc |D]}|jts=tjd|dtdtdt d|dtd||vsZtjd|dt d|d |D]D}|jts||vstjd |d t d |d y )aValidates the installation scripts' path explicitly provided by the user. Args: script_paths (Sequence[str]): Required. The paths to the installation scripts. packages (Sequence[str]): Required. The user-provided packages. Raises: ValueError: If a user-defined script is not under the expected subdirectory, or not in `packages`, or if a package is in the installation scripts subdirectory, but is not specified as an installation script. z"User-defined installation script 'z' is not in the expected 'z(' subdirectory. Ensure it is placed in 'z4' within your 'extra_packages' or 'source_packages'.zRequired installation script 'z' is not under ''zo' is not in 'extra_packages' or 'source_packages'. Ensure it is added to 'extra_packages' or 'source_packages'.z:' does not exist in 'extra_packages' or 'source_packages'.z Package 'z' is in the installation scripts subdirectory, but is not specified as an installation script in `build_options`. Ensure it is added to installation_scripts for automatic execution.zq' is in the installation scripts subdirectory, but is not specified as an installation script in `build_options`.N)rrrrr)rr script_pathrs rKrr*s3&$ %%&:; NN4[MB!!5 67++?*@A9:  0 >!!5 6a9  h & NN4[MB9:  4[MBKL '0    2 3|8S NNG9%'' G9%--   rMc`|s td|jdstd|d|S)z%Tries to validate the staging bucket.zHPlease provide a `staging_bucket` in `client.agent_engines.create(...)`.r-zstaging_bucket=z must start with `gs://`)rr)r+s rK!_validate_staging_bucket_or_raiserfsA  V    $ $W -ON,,DEFF rMc$|j} t|}tjd|t |}t ||}|dj D]#\}}|s tjd|d|%|dj D]B\}}|s |tk(s|D]} |j| tjd |D |S#t$r#} tjd | Yd } ~ |Sd } ~ wwxYw) z6Compiles the requirements into a list of requirements.)rz'Identified the following requirements: r)rrrAzThe following requirements are rr(z)The following requirements are appended: z Failed to compile requirements: N) copyrrr#rrrrrr)r) rrcurrent_requirementsrmissing_requirements warning_typerA action_typer(actionrs rK_validate_requirements_or_warnrqs.  $$&L?1c: =>R=STU(\B 4-# ';:&F&L&L&N  "L(5l^2hZP  %9$C$I$I$K S K;.8%0F ''/0 GyQR  S  ?9!=>> ?s*AC#+8C#$ C#.2C## D,D  DrMc|g}nst|trc tjd|t |5}|j j }tjd|dddt||}tjd||S#1swY0xYw#t$r}td||d}~wwxYw)z#Tries to validate the requirements.Nz'Reading requirements from requirements=zRead the following lines: z.Failed to read requirements from requirements=)rrz The final list of requirements: ) rrqrr#rr splitlinesIOErrorr)r{rrerrs rK_validate_requirements_or_raisers  L# & X KKB\OD El# Iq vvx224  8GH I 2 !L KK2<.AB  I I XKl_MNTW W Xs.#B,7B 1B, B)%B,, C 5CC c ddlm}t||r-tj dddlm}|j|}t|txrt|j}t|txrt|j}t|txrt|j}t|t xrt|j"}t|t$xrt|j&}t|t(xrt|j*}|s|s|s|s|s |s t-d|r t/j0t3|d|r t/j0t3|d |r t/j0t3|d |r t/j0t3|d|r t/j0t3|d|r t/j0t3|dt|t6r|j9}|S#t$rYwxYw#t4$r} t5d| d } ~ wwxYw#t4$r} t5d | d } ~ wwxYw#t4$r} t5d | d } ~ wwxYw#t4$r} t5d| d } ~ wwxYw#t4$r} t5d| d } ~ wwxYw#t4$r} t5d| d } ~ wwxYw)aTries to validate the agent engine. The agent engine must have one of the following: * a callable method named `query` * a callable method named `stream_query` * a callable method named `async_stream_query` * a callable method named `bidi_stream_query` * a callable method named `register_operations` Args: agent: The agent to be validated. Returns: The validated agent engine. Raises: TypeError: If `agent_engine` has no callable method named `query`, `stream_query` or `register_operations`. ValueError: If `agent_engine` has an invalid `query`, `stream_query` or `register_operations` signature. rrsz4Deploying google.adk.agents.Agent as an application.) agent_engines)r{zagent_engine has none of the following callable methods: `query`, `async_query`, `stream_query`, `async_stream_query`, `bidi_stream_query`, or `register_operations`.r5zbInvalid query signature. This might be due to a missing `self` argument in the agent.query method.Nr2znInvalid async_query signature. This might be due to a missing `self` argument in the agent.async_query method.r8zpInvalid stream_query signature. This might be due to a missing `self` argument in the agent.stream_query method.r3z}Invalid async_stream_query signature. This might be due to a missing `self` argument in the agent.async_stream_query method.r9z{Invalid bidi_stream_query signature. This might be due to a missing `self` argument in the agent.bidi_stream_query method.rnz~Invalid register_operations signature. This might be due to a missing `self` argument in the agent.register_operations method.)google.adk.agentsrtrrr#vertexairAdkApprrEcallabler5rUr2r^r8rXr3rar9rlrnrrrrrrgrj) r{rtr is_queryableis_async_queryableis_stream_queryableis_async_stream_queryableis_bidi_stream_queryableis_operation_registrablers rK_validate_agent_or_raisers2  / eY ' KKN O .!((u(5EeY/IHU[[4IL#E>:x @%UO< B!+52F G!H   M *%1DE ( K *%1EF 8 !!L    # # $ =      geW5 6    ge]; <    ge^< =!    ge-AB C     ge-@A B     ge-BC D%#  Lu    H =   C   E   S   R   S  s?H0I'I I:+J J4/K0 H=<H= I II I7& I22I7: J JJ J1 J,,J14 K= K  K K+ K&&K+rcddtffd }|_t|j|_|S)zAWraps an agent operation into a method (works for all API modes).rYc|jjds|jt|jddi|S)Nr{rG)rrrr)rIrJrs rK_methodz&_wrap_agent_operation.._method(sB##G, KKM._methodBsm=> >  ?@ @??))""'' +&** rM) genai_types AgentEnginerrrs` rKrrrr2s# k-- C  NrMc dtjdtdtttttftfffd }|S)aWraps an Agent Engine method, creating an async callable for `query` API. This function creates a callable object that executes the specified Agent Engine method asynchronously using the `query` API. It handles the creation of the API request and the processing of the API response. Args: method_name: The name of the Agent Engine method to call. doc: Documentation string for the method. Returns: A callable object that executes the method on the Agent Engine via the `query` API. rIrJrYcK|js td|js td|jj|jj|ddd{}|j S7w)Nz$api_async_client is not initialized.rTrr)api_async_clientrrrr;rrs rKrz,_wrap_async_query_operation.._methodfs$$CD D  ?@ @..55""'' +&*6   sA'A=*A;+A=)rrrrrrs` rKrsrsTsB$%%14 yc3', -" NrMcNdtjdttffd }|S)aWraps an Agent Engine method, creating a callable for `stream_query` API. This function creates a callable object that executes the specified Agent Engine method using the `stream_query` API. It handles the creation of the API request and the processing of the API response. Args: method_name: The name of the Agent Engine method to call. doc: Documentation string for the method. Returns: A callable object that executes the method on the Agent Engine via the `stream_query` API. rIrYc;K|js td|js td|jj|jj|ddD]}t |D] }|| ywNrrTrr) http_response)rrr _stream_queryr;_yield_parsed_jsonrIrJrliners rKrz-_wrap_stream_query_operation.._methods=> >  ?@ @!__::""'' +&*; M+G #J  s A8B< B)rrr rrs` rKrtrtzs' k--HSM" NrMcNdtjdttffd }|S)a$Wraps an Agent Engine method, creating an async callable for `stream_query` API. This function creates a callable object that executes the specified Agent Engine method using the `stream_query` API. It handles the creation of the API request and the processing of the API response. Args: method_name: The name of the Agent Engine method to call. doc: Documentation string for the method. Returns: A callable object that executes the method on the Agent Engine via the `stream_query` API. rIrYc K|js td|js td|jj|jj|dd23d{}t |D] }|| #76ywr)rrr_async_stream_queryr;r r s rKrz3_wrap_async_stream_query_operation.._methods=> >  ?@ @#'??#F#F""'' +&*$G$  -+G #J  $ s0A$B'B +B ,B /BB B  B)rrrrrs` rKrurus)$K33-PSBT" NrMrc$dtffd }|S)aWraps an Agent Engine method, creating a callable for A2A API. Args: method_name: The name of the Agent Engine method to call. agent_card: The agent card to use for the A2A API call. Example: {'additionalInterfaces': None, 'capabilities': {'extensions': None, 'pushNotifications': None, 'stateTransitionHistory': None, 'streaming': False}, 'defaultInputModes': ['text'], 'defaultOutputModes': ['text'], 'description': ( 'A helpful assistant agent that can answer questions.' ), 'documentationUrl': None, 'iconUrl': None, 'name': 'Q&A Agent', 'preferredTransport': 'JSONRPC', 'protocolVersion': '0.3.0', 'provider': None, 'security': None, 'securitySchemes': None, 'signatures': None, 'skills': [{ 'description': ( 'A helpful assistant agent that can answer questions.' ), 'examples': ['Who is leading 2025 F1 Standings?', 'Where can i find an active volcano?'], 'id': 'question_answer', 'inputModes': None, 'name': 'Q&A Agent', 'outputModes': None, 'security': None, 'tags': ['Question-Answer']}], 'supportsAuthenticatedExtendedCard': True, 'url': 'http://localhost:8080/', 'version': '1.0.0'} Returns: A callable object that executes the method on the Agent Engine via the A2A API. rYc K|js td|js tdtdit j }|j r(|j tjk7r tdt|dstj|_|jr!|jjr tdt|jdsd|j_ |jjjjjd}|jjjj }|d|d|jj"d |_t'tjgd t)j*d d |jjj,j.i }t1|}|j3|} dk(r9|j5t7di|}g} |23d{} | j9|  dk(r$|j;t=di|d{}|S dk(r$|j?tAdi|d{}|S dk(r|jCd{}|Std 76| S7a7:7w)z>Wraps an Agent Engine method, creating a callable for A2A API.rrzBOnly HTTP+JSON is supported for preferred transport on agent card preferred_transportziStreaming is not supported in Agent Engine, please change a2a_agent_card.capabilities.streaming to False. streamingFrz/a2aT AuthorizationzBearer )headers)supported_transportsuse_client_preference httpx_clienton_message_sendN on_get_taskon_cancel_taskhandle_authenticated_agent_cardzUnknown method name: rG)"rrrr!rrrr" http_jsonr capabilitiesr _api_client _http_optionsbase_urlrstrip api_versionr;rr&httpx AsyncClient _credentialstokenr'create send_messager#r)get_taskr% cancel_taskr$get_card) rIrJr@r!r#rfactoryclientrchunkschunkrrs rKrz$_wrap_a2a_operation.._methods=> >  ?@ @"+F+F+P+PB  ~22K@49N ' ' 1??..<<EELLSQoo11??KK (z;-q9J9J9O9O8PPTU!++"#'**#!$//"="="J"J"P"P!QR   '/ + +**7+K"K#K7J99K>KK)r)rrrs`` rKrwrws\DDL NrMrc# K|jsdy|jjdD]}|s tj|}|!y#t$r%}t j d|d|Yd}~0d}~wwxYww)a Converts the body of the HTTP Response message to JSON format. Args: http_response (google.genai.types.HttpResponse): Required. The httpbody body to be converted to JSON object(s). Yields: Any: A JSON object or line of the original body or None. Nrzfailed to parse json: rV)bodyrrrrrr)rr rs rKr r ;s    ""((.  Pzz$'J  P!7v]1#NOO Ps.2BA B BA;6B;BBresource_limitsct|tstdt|d|vsd|vr t dt |d}|d}|dvrt d|t|tr|jdst d| t |d d }|d kDrt d|d|dkDrd}n|dkDrd}n|dkDrd}n |dkDrd}nd}||krt d|d|d|y #t $rt d |d wxYw)ayValidates the resource limits. Checks that the resource limits are a dict with 'cpu' and 'memory' keys. Checks that the 'cpu' value is one of 1, 2, 4, 6, 8. Checks that the 'memory' value is a string ending with 'Gi'. Checks that the memory size is smaller than 32Gi. Checks that the memory size requires at least the specified number of CPUs. Args: resource_limits: The resource limits to be validated. Raises: TypeError: If the resource limits are not a dict. KeyError: If the resource limits do not contain 'cpu' and 'memory' keys. ValueError: If the 'cpu' value is not one of 1, 2, 4, 6, 8. ValueError: If the 'memory' value is not a string ending with 'Gi'. ValueError: If the memory size is too large. ValueError: If the memory size requires more CPUs than the specified 'cpu' value. z$resource_limits must be a dict. Got cpumemoryz5resource_limits must contain 'cpu' and 'memory' keys.)rz9resource_limits['cpu'] must be one of 1, 2, 4, 6, 8. Got GizAresource_limits['memory'] must be a string ending with 'Gi'. Got NzInvalid memory value: z&. Must be an integer followed by 'Gi'. zMemory size of z) is too large. Must be smaller than 32Gi.r:r9r8r7rz requires at least z CPUs. Got ) rrprr KeyErrorintrrqrW)r3r5 memory_str memory_gbmin_cpus rK"_validate_resource_limits_or_raiserESs* ot ,>tO?T>UVWW O#x'FNOO oe$ %C *J /! J3% P   j# &j.A.A$.G < !    3B( 2~j\)R S  2~ R Q Q W}j\)rBr0_FAILED_TO_REGISTER_API_METHODS_WARNING_TEMPLATErrlrrr;r?rvrrrrrorp getLoggerrruntime_checkablerErUrXr^rargrlrrtrur}rwrrrrAgentEngineOperationAgentEngineMemoryOperation$AgentEngineGenerateMemoriesOperationrrrrfloatrrrrrprrboolrrr'r8rrrD ModuleTyperGr.rOrrrrcrrrrrr`rarbrrrrrrrrrrrsrtrurw HttpResponser rErIrOrGrMrKrpsc +  4   " &4&'"w +.+.+C+C8C=C=1113GC#$67 7+.+B+B)C.B  $ 'NI.(($+NNS 2  ((55M'#,#7#7#D#D tCyD 7I)GL!ML%O )'%)*001EF ##' +$4!$8!+?(& &".%4"& z'BC,#6 ,1 .# '%" &,/1= 368D #   8 9===LXLLG8GG8h88G(GG,,, 8   +#Hi#/(1(9%1   !Ye!]E)1]E@Yi ,I, "$$**446 8   ScSVJ$& 0!   @$& +!   @26 7#s(#7x}gc8O3L.L&MMN7  . 7 # 7t2 2S$s)^$2 %-- 2jGS#X'G %--GW%6%64."&&( tSt#t#s(# t sm t  #s(^ tn%$c]%$ !%$ 38_ %$Z"&    #   *  #tCy., e&6&6  (8(8  1A1A  E,<,<  5#3#3  # S(?+ +,J"&6 ) *6 8C=)6 Xc]+ 6  6  6 6 6 #6  6 | ^S))^S  S(C:xS'99: :;^S  ^SH/3CG=A *T *TXc]+*T$GC#,>$?@ *T (S(9: *T  S#X *TZemmS#X<(, tCH~u}}, - emm $ ]]>9 99 9  94 ?3- ? ? ?  ?ASMAA A  A&;c]; ;69=smDhsm!345c] $9 3-9 sm9  9 x s) #Y B-1 8C=)c] 0{ {{| C C HS#XVtn&8&E&E(SV-0D S#XD 4D N 0 5 0$ 0!tCsCx012! d3c3h'()!I;.+-8C=-.=:<GC#$67<= 7+4;)C.67  (($'S '(  ('#& tCy&'* ^$IGLMLOB  (s~%\:\.] ]$"]<7$^=^0\+*\+.]] ]! ]!$]98]9<^^^-,^-0 ^=<^=