o YiE @sddlZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddl m Z ddl m Zdd l mZed Z dd eeeefefd eeeefd eeeffddZ dd eeeefefd eeeefd eeeffddZ dd eeeefefd eeeefd eeeffddZ dd eeeefefd eeeefd eeeffddZGddde jZGddde jZdS)N)AnyOptionalTypeVarUnion) urlencode) _api_module)_common)types)get_value_by_path)set_value_by_pathzgoogle_genai.operations from_object parent_objectreturncCsVi}t|dgdurt|dgt|dgt|dgdur)t|ddgt|dg|S)Noperation_name operationName resource_name_urlZ resourceNamegetvsetvr rZ to_objectrQ/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/operations.py*_FetchPredictOperationParameters_to_vertex!s rcC0i}t|dgdurt|ddgt|dg|SNrrrrrrrr _GetOperationParameters_to_mldev3 rcCrrrrrrr!_GetOperationParameters_to_vertexBrrcCs0i}t|dgdurt|ddgt|dg|S)N operation_idrrrrrr(_GetProjectOperationParameters_to_vertexQs r!c @eZdZdddedeejdefddZdddededeej defd d Z ddd edeejdej fd d Z e dejdZdddedeejdefddZdS) OperationsNconfigrr%rc Cstj||d}|jjrt|}|d}|rd|}nd}nt|}|d}|r0d|}nd}|d}|rB|dt|}| ddd}|j durY|j j durY|j j }t |}t |}|jd|||} | jssi} | St| j} | SNrr%rz{operationName}_query?r%get)r _GetOperationParameters _api_clientvertexairr* format_maprrpopr% http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloads selfrr%parameter_model request_dictrequest_url_dictpath query_paramsr0response response_dictrrr_get_videos_operation`s<            z Operations._get_videos_operationrc Cstj|||d}|jjstdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd} |j durJ|j j durJ|j j } t |}t |}|jd||| } | jsdi} | St| j} | S Nrrr%6This method is only supported in the Vertex AI client.rz${resourceName}:fetchPredictOperationr(r)r%post)r _FetchPredictOperationParametersr,r- ValueErrorrr*r.rr/r%r0r r1r2r3r4r5r6 r8rrr%r9r:r;r<r=r0r>r?rrr_fetch_predict_videos_operations>         z*Operations._fetch_predict_videos_operationr c Cstj||d}|jjstdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd|||} | jsaint| j} tjj| |d } |j| | S N)r r%rCrzoperations/{operation_id}r(r)r%r*)r>kwargs)r _GetProjectOperationParametersr,r-rFr!r*r.rr/r%r0r r1r2r3r4r5r6ProjectOperation_from_response model_dump_verify_response r8r r%r9r:r;r<r=r0r>r? return_valuerrr_gets8         zOperations._getTbound operationc Cs|j}|s td|jjre|dd}t}t|tr2| dd}|dur1tjd it|}nt|tj rH|durH|j durD|j nt}tj |d}|j |||d}|j|dd } |j| | S|j||d }|j|d d } |j| | S) aGets the status of a long-running operation. Args: operation (Operation): The operation instance to get the status for. config (GetOperationConfig): Configuration for getting the operation. Returns: Operation: The updated operation instance with the latest status or result. Usage: .. code-block:: python import time operation = client.models.generate_videos( model="veo-2.0-generate-001", source=types.GenerateVideosSource( prompt="A neon hologram of a cat driving at top speed", ), ) while not operation.done: time.sleep(10) operation = client.operations.get(operation) print(operation.result) Operation name is empty. /operations/rr0Nr0rBTZ is_vertex_air'Fr)namerFr,r- rpartitionr HttpOptions isinstancedictr*GetOperationConfigr0FetchPredictOperationConfigrHfrom_api_responserOr@ r8rVr%rrr0Z dict_optionsZfetch_operation_configr?Zresponse_operationrrrr*sL$     zOperations.get__name__ __module__ __qualname__strrr ZGetOperationConfigOrDictrr@Z!FetchPredictOperationConfigOrDictrHrLrRrZ OperationrSr*rrrrr#^sH 6 6 1r#c @r")AsyncOperationsNr$rr%rc stj||d}|jjr t|}|d}|rd|}nd}nt|}|d}|r1d|}nd}|d}|rC|dt|}| ddd}|j durZ|j j durZ|j j }t |}t |}|jd|||IdH} | jswi} | St| j} | Sr&)r r+r,r-rr*r.rrr/r%r0r r1r2 async_requestr4r5r6r7rrrr@JsB            z%AsyncOperations._get_videos_operationrc stj|||d}|jjstdt|}|d}|r"d|}nd}|d}|r4|dt|}| ddd} |j durK|j j durK|j j } t |}t |}|jd||| IdH} | jshi} | St| j} | SrA)r rEr,r-rFrr*r.rr/r%r0r r1r2rjr4r5r6rGrrrrH}s@         z/AsyncOperations._fetch_predict_videos_operationr c stj||d}|jjstdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd|||IdH} | jseint| j} tjj| |d } |j| | SrI)r rKr,r-rFr!r*r.rr/r%r0r r1r2rjr4r5r6rLrMrNrOrPrrrrRs>          zAsyncOperations._getrSrTrVc s|j}|s td|jjrc|dd}t}t|tr3| dd}|dur2tjd it|}nt|tj rI|durI|j durE|j nt}tj |d}|j |||dIdH}|j|dd } | S|j||d IdH}|j|d d } | S) aGets the status of a long-running operation. Args: operation (Operation): The operation instance to get the status for. config (GetOperationConfig): Configuration for getting the operation. Returns: Operation: The updated operation instance with the latest status or result. Usage: .. code-block:: python import asyncio operation = await client.aio.models.generate_videos( model="veo-2.0-generate-001", source=types.GenerateVideosSource( prompt="A neon hologram of a cat driving at top speed", ), ) while not operation.done: await asyncio.sleep(10) operation = await client.aio.operations.get(operation) print(operation.result) rWrXrr0NrYrBTrZr'Fr)r[rFr,r-r\r r]r^r_r*r`r0rarHrbr@rcrrrr*sJ#     zAsyncOperations.getrdrrrrriHsH 8 6 3ri)N)r5loggingtypingrrrr urllib.parserrr r r rr r getLoggerloggerr_rhobjectrrrr!Z BaseModuler#rirrrrsZ                k