o pkj6@s0dZddlmZmZddlmZddlmZmZm Z m Z ddl m Z m Z mZddlmZmZddl mZmZmZmZddlmZdd lmZdd l mZmZdd lmZd d l Z d dl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(GdddeZ)GdddZ*GdddZ+GdddeZ,GdddZ-GdddZ.d S)zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.) AsyncBaseSDKBaseSDK)SDKConfiguration)errorsmodelstypesutils)AfterParseErrorContext HookContextResponseContext) AsyncFilesFiles)OptionalNullableUNSET environments interactions)createenvironmentrequest)source)get_security_from_envresponse_helpers)unmarshal_json_responseN)AnyIterableListMappingOptionalUnioncastc@eZdZUeddZeddZeed< d#dede e d dfd d Z d d Z dddddddde e de ede e de ee e fde ee efde eeejfd ejfddZddddddddde e de eejejfde eeejeejfde ee e fde ee efde ee efde eeejfd ejfddZ dddddde de e de ee e fde ee efde eeejfd e!j"f dd Z#dddddde de e de ee e fde ee efde eeejfd ejf d!d"Z$dS)$ EnvironmentscCt|SN)EnvironmentsWithRawResponseselfr&]/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/google/genai/_gaos/environments.pywith_raw_response$zEnvironments.with_raw_responsecCr!r")!EnvironmentsWithStreamingResponser$r&r&r'with_streaming_response(r)z$Environments.with_streaming_responsefilesN sdk_config parent_refreturncC"tj|||d||_|dSN)r.)r__init__sdk_configuration _init_sdksr%r-r.r&r&r'r2. zEnvironments.__init__cCt|j|jd|_dSr1)r r3r.r,r$r&r&r'r45zEnvironments._init_sdks api_version page_size page_token extra_headers extra_querytimeoutr:r;r<r=r>r?c Csd}d}t} d} |} ||} | dur|jj} | dur| }n|||}tj|||d} t| \}} |j d3iddddd|d|d | d d d d dd ddddd| d|dtj |jj j dd|jj ddd| }| tkr|jjtur|jj} ntjdtddddd dd } d}t| tjr| gd!f}d"d#}t|j|pd$d%dt|jj tjddt|d&d'd(}|j||d)d*|d+k|d,}|d-kr#td.d/g|jr|z||Wn!ty}zt|jjd0t|||WYd}~nd}~ww|d+kr tj ntj!}t"t#|||d1|t||jj$d0d2Sz||WStyJ}zt|jjd0t|||WYd}~dSd}~ww)4Lists environments. :param api_version: Which version of the API to use. :param page_size: Optional. Maximum number of environments to return.\nIf unspecified, defaults to 50. Maximum is 1000. :param page_token: Optional. Pagination token. :param extra_headers: Additional headers to set or replace on requests. :param extra_query: Additional query parameters to append to requests. :param timeout: Override the default request timeout configuration for this method in seconds Nr:r;r<methodGETpath/{api_version}/environmentsbase_url url_variablesrequestrequest_body_requiredFrequest_has_path_paramsTrequest_has_query_paramsuser_agent_header user-agentaccept_header_valueapplication/json http_headersextra_query_params_globalsr:securityallow_empty_value timeout_msattempt-count-backoff@0u max_retries4084094295XXcSvt|ddrt|}td||t|ddr&t|}td||t|ddr5ttj|ddStd | N4XX*zAPI error occurredrcdefaultrOF)validatezUnexpected response received)rmatch_responsestream_to_textrGenAiDefaultErrorrrListEnvironmentsResponsehttp_res http_res_textr&r&r'_speakeasy_parse_response   zAEnvironments.list_environments.._speakeasy_parse_responseListEnvironmentssyncmode executionconfigrF operation_id oauth2_scopessecurity_sourcetags extensionsresponsecStddg|SNrfrcrmatch_status_codescr&r&r'z0Environments.list_environments.. streaminghook_ctxrHis_error_status_codestream retry_configparsedrfrc_hooksbufferedrawparserrw client_refrhooksr&)%r_coerce_timeout_msr3rV_get_urlrListEnvironmentsRequestrconsume_response_mode_build_requestListEnvironmentsGlobalsglobalsr:rTrr RetryConfigBackoffStrategy isinstancer rrSecurityr do_requestr status_coderead Exceptionraise_parse_error__dict__r StreamedAPIResponse APIResponserrgetr%r:r;r<r=r>r?rFrGretries server_urlrPrVrH_speakeasy_response_modereqrrq_speakeasy_hook_ctxro parse_exc__speakeasy_response_clsr&r&r'list_environments8s                      zEnvironments.list_environmentsr:networksourcesr=r> extra_bodyr?rrrc sd}d} t} d} |} ||} | dur|jj} | dur| }n||| }tj|tjt |t tj t |t t t jddt| \}} |jd5iddddd|d | d d d d d dd ddddd| d|dtj|jjjdd|jjdfddddd| }| tkr|jjtur|jj} ntjdtddd d!d d"d#} d}t| tjr| gd$f}d%d&}t|j|pd'd(dt|jjtjddt|d)d*d+}|j||d,d|d-k|d.}|d/kr?t d0d1g|j!r |"z||Wn!t#y}zt$|jj%d2t&|||WYd}~nd}~ww|d-kr(tj'ntj(}t)t*|||d3|t&||jj%+d2d4Sz||WSt#yf}zt$|jj%d2t&|||WYd}~dSd}~ww)6ACreates an environment. :param api_version: Which version of the API to use. :param network: Network configuration for the environment. :param sources: Sources to be mounted into the environment. :param extra_headers: Additional headers to set or replace on requests. :param extra_query: Additional query parameters to append to requests. :param extra_body: Additional JSON object fields to merge into request bodies. :param timeout: Override the default request timeout configuration for this method in seconds Nrrr:bodyrBPOSTrDrErFrGrHrITrJrKrLrMrNrOrPrQrRrSrTget_serialized_bodyctjjdddtjdSNFjson)rrserialize_request_bodyrrCreateEnvironmentRequestr&rrHr&r'rz1Environments.create_environment..rUrVrWrXrYrZr[r\r]r_cSrdrerrjrkrrlrr Environmentrnr&r&r'rq9rrzBEnvironments.create_environment.._speakeasy_parse_responsersCreateEnvironmentrurvrycSrrrrr&r&r'rZrrrrrfrcrrrr&),rrr3rVrrrrrget_pydantic_modelr$CreateEnvironmentRequestNetworkUnionrrSourcerrrCreateEnvironmentGlobalsrr:rTrrrrr rrrr rrrrrrrr rrrrrr%r:rrr=r>rr?rFrGrrrPrVrrrrqrrorrr&rr'create_environments                         zEnvironments.create_environmentr:r=r>r?idc Cd}d}t}d} |} ||} | dur|jj} | dur| }n|||}tj||d} t| \} } |j d3iddddd|d|d | d d d d dd ddddd| d|dtj |jj j dd|jj ddd| }|tkr|jjtur|jj}ntjdtddddd dd }d}t|tjr|gd!f}d"d#}t|j|pd$d%dt|jj tjddt| d&d'd(}|j||d)d*| d+k|d,}| d-kr"td.d/g|jr|z||Wn!ty}zt|jjd0t|||WYd}~nd}~ww| d+kr tj ntj!}t"t#|||d1|t||jj$d0d2Sz||WStyI}zt|jjd0t|||WYd}~dSd}~ww)4 Deletes an environment. :param id: Resource ID segment making up resource `name`. It identifies the resource within its parent collection as described in https://google.aip.dev/122. :param api_version: Which version of the API to use. :param extra_headers: Additional headers to set or replace on requests. :param extra_query: Additional query parameters to append to requests. :param timeout: Override the default request timeout configuration for this method in seconds Nr:rrBDELETErD /{api_version}/environments/{id}rFrGrHrIFrJTrKrLrMrNrOrPrQrRrSrTrUrVrWrXrYrZr[r\r]r_cSrdre)rrjrkrrlrrEmptyrnr&r&r'rqrrzBEnvironments.delete_environment.._speakeasy_parse_responsersDeleteEnvironmentrurvrycSrrrrr&r&r'rrz1Environments.delete_environment..rrrrfrcrrrr&)%rrr3rVrrDeleteEnvironmentRequestrrrDeleteEnvironmentGlobalsrr:rTrrrrrr rrrr rrrrrrrr rrrrrr%rr:r=r>r?rFrGrrrPrVrHrrrrqrrorrr&r&r'delete_environment                      zEnvironments.delete_environmentc Cr)4Gets an environment. :param id: Resource ID segment making up resource `name`. It identifies the resource within its parent collection as described in https://google.aip.dev/122. :param api_version: Which version of the API to use. :param extra_headers: Additional headers to set or replace on requests. :param extra_query: Additional query parameters to append to requests. :param timeout: Override the default request timeout configuration for this method in seconds NrrBrCrDrrFrGrHrIFrJTrKrLrMrNrOrPrQrRrSrTrUrVrWrXrYrZr[r\r]r_cSrdrerrnr&r&r'rqerrz?Environments.get_environment.._speakeasy_parse_responsersGetEnvironmentrurvrycSrrrrr&r&r'rrz.Environments.get_environment..rrrrfrcrrrr&)%rrr3rVrrGetEnvironmentRequestrrrGetEnvironmentGlobalsrr:rTrrrrrr rrrr rrrrrrrr rrrrrrr&r&r'get_environmentrzEnvironments.get_environmentr")%__name__ __module__ __qualname__propertyr(r+r __annotations__rrobjectr2r4strintrrrfloathttpxTimeoutrrmr%environments_createenvironmentrequestr)CreateEnvironmentRequestNetworkUnionParamrinteractions_sourcer SourceParamrrrrrrr&r&r&r'r #s       8 r c@*eZdZdeddfddZeddZdS)r#sdkr/NcCJ||_t|jd|_t|jd|_t|jd|_t|jd|_dSNr=)_sdkrto_raw_response_wrapperrrrrr%rr&r&r'r2 z$EnvironmentsWithRawResponse.__init__cC |jjjSr"rr,r(r$r&r&r'r, z!EnvironmentsWithRawResponse.filesrrrr r2rr,r&r&r&r'r#r#c@r)r*rr/NcCrr)rrto_streamed_response_wrapperrrrrrr&r&r'r2rz*EnvironmentsWithStreamingResponse.__init__cCrr"rr,r+r$r&r&r'r,rz'EnvironmentsWithStreamingResponse.filesrr&r&r&r'r*rr*c@r)$AsyncEnvironmentscCr!r") AsyncEnvironmentsWithRawResponser$r&r&r'r(r)z#AsyncEnvironments.with_raw_responsecCr!r")&AsyncEnvironmentsWithStreamingResponser$r&r&r'r+r)z)AsyncEnvironments.with_streaming_responser,Nr-r.r/cCr0r1)rr2r3r4r5r&r&r'r2r6zAsyncEnvironments.__init__cCr7r1)r r3r.r,r$r&r&r'r4r8zAsyncEnvironments._init_sdksr9r:r;r<r=r>r?c sd}d}t} d} |} ||} | dur|jj} | dur| }n|||}tj|||d} t| \}} |j d4iddddd|d|d | d d d d dd ddddd| d|dtj |jj j dd|jj ddd| }| tkr|jjtur|jj} ntjdtddddd dd } d}t| tjr| gd!f}d"d#}t|j|pd$d%dt|jj tjddt|d&d'd(}|j||d)d*|d+k|d,IdH}|d-kr=td.d/g|jr|IdHz ||IdHWn+ty}zt|jjd0|jjd1t |||IdHWYd}~nd}~ww|d+kr tj!ntj"}t#t$|||d2|t ||jjd1|jjd0d3Sz||IdHWStyq}zt|jjd0|jjd1t |||IdHWYd}~dSd}~ww)5r@NrArBrCrDrErFrGrHrIFrJTrKrLrMrNrOrPrQrRrSrTrUrVrWrXrYrZr[r\r]r_ct|ddrt|IdH}td||t|ddr-t|IdH}td||t|ddr  zFAsyncEnvironments.list_environments.._speakeasy_parse_responsersrtasyncrvrycSrrrrr&r&r'r_rz5AsyncEnvironments.list_environments..rrrrfrc _async_hooksrrrrrwrrr async_hooksr&)%rrr3rVrrrrr_build_request_asyncrrr:rTrrrrrr rrrr do_request_asyncrrareadrraise_parse_error_asyncrrr AsyncStreamedAPIResponseAsyncAPIResponserrrr&r&r'rs                        z#AsyncEnvironments.list_environmentsrrrrc sd}d} t} d} |} ||} | dur|jj} | dur| }n||| }tj|tjt |t tj t |t t t jddt| \}} |jd6iddddd|d | d d d d d dd ddddd| d|dtj|jjjdd|jjdfddddd| }| tkr|jjtur|jj} ntjdtddd d!d d"d#} d}t| tjr| gd$f}d%d&}t|j|pd'd(dt|jjtjddt|d)d*d+}|j||d,d|d-k|d.IdH}|d/krYt d0d1g|j!r4|"IdHz ||IdHWn+t#y3}zt$|jj%&d2|jj%&d3t'|||IdHWYd}~nd}~ww|d-kr.rUrVrWrXrYrZr[r\r]r_crrerrjrrrlrrrrnr&r&r'rqrzGAsyncEnvironments.create_environment.._speakeasy_parse_responsersrrrvrycSrrrrr&r&r'rrrrrrfrcrrrrr&),rrr3rVrrrrrrrrrrrrrr rrr:rTrrrrr rrrr r rrr rr rrr rrrrrr&rr'rs                          z$AsyncEnvironments.create_environmentrrc d}d}t}d} |} ||} | dur|jj} | dur| }n|||}tj||d} t| \} } |j d4iddddd|d|d | d d d d dd ddddd| d|dtj |jj j dd|jj ddd| }|tkr|jjtur|jj}ntjdtddddd dd }d}t|tjr|gd!f}d"d#}t|j|pd$d%dt|jj tjddt| d&d'd(}|j||d)d*| d+k|d,IdH}| d-kr._speakeasy_parse_responsersrrrvrycSrrrrr&r&r'rrz6AsyncEnvironments.delete_environment..rrrrfrcrrrrr&)%rrr3rVrrrrrr rrr:rTrrrrrr rrrr r rrr rr rrr rrrrrr&r&r'rB                        z$AsyncEnvironments.delete_environmentc r)5rNrrBrCrDrrFrGrHrIFrJTrKrLrMrNrOrPrQrRrSrTrUrVrWrXrYrZr[r\r]r_crrerrnr&r&r'rq&rzDAsyncEnvironments.get_environment.._speakeasy_parse_responsersrrrvrycSrrrrr&r&r'rGrz3AsyncEnvironments.get_environment..rrrrfrcrrrrr&)%rrr3rVrrrrrr rrr:rTrrrrrr rrrr r rrr rr rrr rrrrrr&r&r'rrz!AsyncEnvironments.get_environmentr")%rrrrr(r+r rrrrr2r4rrrrrrrrrrmrrrrrrrrrrrrrrr&r&r&r'rs        ; rc@r)rrr/NcCrr)rrasync_to_raw_response_wrapperrrrrrr&r&r'r2vrz)AsyncEnvironmentsWithRawResponse.__init__cCrr"rr$r&r&r'r,rz&AsyncEnvironmentsWithRawResponse.filesrrrrr2rr,r&r&r&r'rurrc@r)rrr/NcCrr)rr"async_to_streamed_response_wrapperrrrrrr&r&r'r2rz/AsyncEnvironmentsWithStreamingResponse.__init__cCrr"rr$r&r&r'r,rz,AsyncEnvironmentsWithStreamingResponse.filesrr&r&r&r'rrr)/__doc__basesdkrrsdkconfigurationrrsrrrrrr r r r,r r rrrrtypes.environmentsrrtypes.interactionsrrrrutils.unmarshal_json_responserrtypingrrrrrrrr r#r*rrrr&r&r&r's:    $