R-jPPddlmZddlmZddlZddlmZmZddlm Z m Z m Z m Z m Z mZmZddlmZmZddlmZdd lmZmZdd lmZmZmZmZdd lmZdd lm Z dd l!m"Z"ddl#m$Z$ddgZ%GddeZ&GddeZ'GddZ(GddZ)GddZ*GddZ+y)) annotations)IterableN)agent_list_paramsagent_create_params)BodyOmitQueryHeadersNotGivenomit not_given)maybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_raw_response_wrapperto_streamed_response_wrapperasync_to_raw_response_wrapper"async_to_streamed_response_wrapper)Agent)make_request_options)AgentListResponse)AgentDeleteResponseAgentsResourceAsyncAgentsResourcec eZdZed dZed dZdeeeeeeddded d dZdeeeddded ddZ dddded dd Z dddded dd Z y)rct|Sa$ This property can be used as a prefix for any HTTP method call to return the raw response object instead of the parsed content. For more information, see https://www.github.com/stainless-sdks/gemini-next-gen-api-python#accessing-raw-response-data-eg-headers )AgentsResourceWithRawResponseselfs y/Users/jameslopez/projects/TradingBot25/.venv/lib/python3.12/site-packages/google/genai/_interactions/resources/agents.pywith_raw_responsez AgentsResource.with_raw_response,s-T22ct|Sz An alternative to `.with_raw_response` that doesn't eagerly read the response body. For more information, see https://www.github.com/stainless-sdks/gemini-next-gen-api-python#with_streaming_response )#AgentsResourceWithStreamingResponser"s r$with_streaming_responsez&AgentsResource.with_streaming_response6s3488r&N api_versionid base_agentbase_environment descriptionsystem_instructiontools extra_headers extra_query extra_bodytimeoutc ||jj}|std||j|jj |dt ||||||dt jt|| | | tS) Creates a new Agent (Typed version for SDK). Args: id: The unique identifier for the agent. base_agent: The base agent to extend. base_environment: The environment configuration for the agent. description: Agent description for developers to quickly read and understand. system_instruction: System instruction for the agent. tools: The tools available to the agent. extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds :Expected a non-empty value for `api_version` but received agentsr,pathr-r.r/r0r1r2r3r4r5r6bodyoptionscast_to) _client_get_api_version_path_param ValueError_post_build_maybe_vertex_pathrrAgentCreateParamsrr r#r,r-r.r/r0r1r2r3r4r5r6s r$createzAgentsResource.create?sR  ,,BBDKYZeYhij jzz LL 1 1kPX 1 Y ",(8#.*<" $55 )+Q[el!  r&r, page_size page_tokenparentr3r4r5r6c||jj}|std||j|jj |dt ||||t |||dtjtS)E Lists all Agents. Args: extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds r9r:r;rLrMrNr3r4r5r6queryrArB) rCrDrE_getrGrrrAgentListParamsr r#r,rLrMrNr3r4r5r6s r$listzAgentsResource.lists4  ,,BBDKYZeYhij jyy LL 1 1kPX 1 Y(+'%%%.&0"( &55  &!  r&r,r3r4r5r6c ||jj}|std||std||j|jj |d|t ||||t S)E Deletes an Agent. Args: extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds r91Expected a non-empty value for `id` but received agents/r;r>rTrCrDrE_deleterGrrr#r-r,r3r4r5r6s r$deletezAgentsResource.deletes0  ,,BBDKYZeYhij jPQSPVWX X|| LL 1 1kRYZ\Y]P^ 1 _(+Q[el(   r&c ||jj}|std||std||j|jj |d|t ||||t S)J Gets a specific Agent. Args: extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds r9r\r]r;r>rTrCrDrErUrGrrr`s r$getzAgentsResource.gets0  ,,BBDKYZeYhij jPQSPVWX Xyy LL 1 1kRYZ\Y]P^ 1 _(+Q[el   r&)returnr!)rfr)r, str | Noner- str | Omitr.rir/z*agent_create_params.BaseEnvironment | Omitr0rir1rir2z)Iterable[agent_create_params.Tool] | Omitr3Headers | Noner4 Query | Noner5 Body | Noner6'float | httpx.Timeout | None | NotGivenrfrr,rhrLz int | OmitrMrirNrir3rjr4rkr5rlr6rmrfrr-strr,rhr3rjr4rkr5rlr6rmrfrr-rpr,rhr3rjr4rkr5rlr6rmrfr __name__ __module__ __qualname__rr%r*r rrJrXrarer&r$rr+s3399#'!%GK"&)-;?)-$("&;D>  >   >  > E >  > '> 9> &> ">  > 9> !> F#' $!%!)-$("&;D/  /  /  /  / &/ "/  / 9/  / j#')-$("&;D$ $  $ &$ "$  $ 9$  $ T#')-$("&;D$ $  $ &$ "$  $ 9$  $ r&c eZdZed dZed dZdeeeeeeddded d dZdeeeddded ddZ dddded dd Z dddded dd Z y)rct|Sr )"AsyncAgentsResourceWithRawResponser"s r$r%z%AsyncAgentsResource.with_raw_responses2$77r&ct|Sr()(AsyncAgentsResourceWithStreamingResponser"s r$r*z+AsyncAgentsResource.with_streaming_responses8==r&Nr+c JK||jj}|std||j|jj |dt ||||||dt jd{t|| | | td{S7$7w)r8Nr9r:r;r=r>r?) rCrDrErFrGrrrHrrrIs r$rJzAsyncAgentsResource.createsR  ,,BBDKYZeYhij jZZ LL 1 1kPX 1 Y,",(8#.*<" $55  )+Q[el!      s$A8B#:B ;B#B!B#!B#rKcDK||jj}|std||j|jj |dt ||||t |||dtjd{td{S77w)rPNr9r:r;rQrRrT) rCrDrErUrGrrrrVrrWs r$rXzAsyncAgentsResource.listQs4  ,,BBDKYZeYhij jYY LL 1 1kPX 1 Y(+'%1%.&0"( &55  &!    s$A>B BB BB B rYc K||jj}|std||std||j|jj |d|t ||||t d{S7w)r[Nr9r\r]r;r>rTr^r`s r$razAsyncAgentsResource.deletes0  ,,BBDKYZeYhij jPQSPVWX X\\ LL 1 1kRYZ\Y]P^ 1 _(+Q[el( "    BB B B c K||jj}|std||std||j|jj |d|t ||||t d{S7w)rcNr9r\r]r;r>rTrdr`s r$rezAsyncAgentsResource.gets0  ,,BBDKYZeYhij jPQSPVWX XYY LL 1 1kRYZ\Y]P^ 1 _(+Q[el     r)rfry)rfr{rgrnrorqrrrvr&r$rrs88>>#'!%GK"&)-;?)-$("&;D>  >   >  > E >  > '> 9> &> ">  > 9> !> F#' $!%!)-$("&;D/  /  /  /  / &/ "/  / 9/  / j#')-$("&;D$ $  $ &$ "$  $ 9$  $ T#')-$("&;D$ $  $ &$ "$  $ 9$  $ r&ceZdZddZy)r!c||_t|j|_t|j|_t|j|_t|j |_yN)_agentsrrJrXrarer#r:s r$__init__z&AgentsResourceWithRawResponse.__init__s[ - MM  , KK  . MM  + JJ r&Nr:rrfNonersrtrurrvr&r$r!r! r&r!ceZdZddZy)ryc||_t|j|_t|j|_t|j|_t|j |_yr)rrrJrXrarers r$rz+AsyncAgentsResourceWithRawResponse.__init__s[ 3 MM  2 KK  4 MM  1 JJ r&Nr:rrfrrrvr&r$ryryrr&ryceZdZddZy)r)c||_t|j|_t|j|_t|j|_t|j |_yr)rrrJrXrarers r$rz,AgentsResourceWithStreamingResponse.__init__s[ 2 MM  1 KK  3 MM  0 JJ r&Nrrrvr&r$r)r)rr&r)ceZdZddZy)r{c||_t|j|_t|j|_t|j|_t|j |_yr)rrrJrXrarers r$rz1AsyncAgentsResourceWithStreamingResponse.__init__s[ 8 MM  7 KK  9 MM  6 JJ r&Nrrrvr&r$r{r{rr&r{), __future__rtypingrhttpxtypesrr_typesrr r r r r r_utilsrr_compatr _resourcerr _responserrrr types.agentr _base_clientrtypes.agent_list_responsertypes.agent_delete_responser__all__rrr!ryr)r{rvr&r$rs$# :JJJ;%9  /9= 2 3O _O dO *O d  $  $  $  r&