o YiZ? @s.ddlmZddlZddlZddlmZmZmZddlm Z ddl m Z ddl m Z ddl m Z dd l mZdd l mZdd lmZmZed Z d d eeeefefdeeeefdeeeffddZ d d eeeefefdeeeefdeeeffddZ d d eeeefefdeeeefdeeeffddZ d d eeeefefdeeeefdeeeffddZ d d eeeefefdeeeefdeeeffddZ d d eeeefefdeeeefdeeeffddZGddde j Z!Gddde j Z"dS)!)partialN)AnyOptionalUnion) urlencode) _api_module)_common)types)get_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.documents from_object parent_objectreturncCs0i}t|dgdurt|ddgt|dg|S)Nforce_querygetvsetvrrZ to_objectrP/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/documents.py_DeleteDocumentConfig_to_mldev#srcCRi}t|dgdurt|ddgt|dgt|dgdur'tt|dg||S)Nname_urlconfig)rrrrrrr"_DeleteDocumentParameters_to_mldev/ rcCs0i}t|dgdurt|ddgt|dg|S)Nrrrrrrr_GetDocumentParameters_to_mldev=sr!cCsXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|S)NZ page_sizerZpageSizeZ page_tokenZ pageTokenrrrrr_ListDocumentsConfig_to_mldevHs r"cCr)Nparentrr)rrr"rrrr!_ListDocumentsParameters_to_mldev]r r$cCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgD|S)NZsdkHttpResponseZsdk_http_responseZ nextPageTokenZnext_page_token documentscSsg|]}|qSrr).0itemrrr |sz5_ListDocumentsResponse_from_mldev..rrrrr!_ListDocumentsResponse_from_mldevksr)c@eZdZdddedeejdejfddZdddedeej ddfdd Z ddd edeej dej fd d Z ddd edeej deejfd dZdS) DocumentsNrrrrc Cstj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||} | jsaint| j} tjj| |d } |j| | S aGets metadata about a Document. Args: name (str): The resource name of the Document. Example: ragStores/rag-store-foo/documents/documents-bar config (GetDocumentConfig | None): Optional parameters for the request. Returns: The Document. rr=This method is only supported in the Gemini Developer client.r{name}r?rNgetresponsekwargs)r _GetDocumentParameters _api_clientvertexai ValueErrorr!r2 format_maprpopr http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsDocument_from_response model_dump_verify_response selfrrparameter_model request_dictrequest_url_dictpath query_paramsr<r4 response_dict return_valuerrrr2s<         z Documents.getc Cstj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||} dS aDeletes a Document. Args: name (str): The resource name of the Document. Example: ragStores/rag-store-foo/documents/documents-bar config (DeleteDocumentConfig | None): Optional parameters for the request. Returns: None r.r/rr0rr1rNdelete)r _DeleteDocumentParametersr7r8r9rr2r:rr;rr<r r=r>r? rHrrrIrJrKrLrMr<r4rrrrQs4        zDocuments.deleter#c Cstj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd|||} | jsaint| j} |jjsot| } tjj| |d } |j| | S Nr#rr/rz{parent}/documentsrr1rr2r3)r _ListDocumentsParametersr7r8r9r$r2r:rr;rr<r r=r>r?r@rArBr)ListDocumentsResponserDrErF rHr#rrIrJrKrLrMr<r4rNrOrrr_lists@         zDocuments._listcCs&t|j|d}td||j||d|S)aLists documents. Args: parent (str): The name of the RagStore containing the Documents. config (ListDocumentsConfig): Optional configuration for the list request. Returns: A Pager object that contains one page of documents. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python for document in client.documents.list(parent='rag_store_name'): print(f"document: {document.name} - {document.display_name}") r#r%rU)rrYrrHr#rZ list_requestrrrlist,s zDocuments.list)__name__ __module__ __qualname__strrr GetDocumentConfigOrDictrCr2DeleteDocumentConfigOrDictrQListDocumentsConfigOrDictrWrYrr\rrrrr+sB ? ; :r+c@r*)AsyncDocumentsNr,rrrc stj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdH} | jseint| j} tjj| |d } |j| | Sr-)r r6r7r8r9r!r2r:rr;rr<r r=r> async_requestr@rArBrCrDrErFrGrrrr2LsB          zAsyncDocuments.getc stj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdH} dSrP)r rRr7r8r9rr2r:rr;rr<r r=r>rerSrrrrQs6        zAsyncDocuments.deleter#c s tj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd|||IdH} | jseint| j} |jjsst| } tjj| |d } |j| | SrT)r rVr7r8r9r$r2r:rr;rr<r r=r>rer@rArBr)rWrDrErFrXrrrrYsF          zAsyncDocuments._listcs.t|j|d}td||j||dIdH|S)a6Lists documents asynchronously. Args: parent (str): The name of the RagStore containing the Documents. config (ListDocumentsConfig): Optional configuration for the list request. Returns: A Pager object that contains one page of documents. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python async for document in await client.aio.documents.list(parent='rag_store_name'): print(f"document: {document.name} - {document.display_name}") rZr%rUN)rrYr r[rrrr\szAsyncDocuments.list)r]r^r_r`rr rarCr2rbrQrcrWrYr r\rrrrrdJsB A ; <rd)N)# functoolsrrAloggingtypingrrr urllib.parserrr r r rr rZpagersr r getLoggerloggerdictr`objectrrr!r"r$r)Z BaseModuler+rdrrrrs~                     I