o pkjwH @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_querygetvsetvrr to_objectrT/home/djax/ivt_ai_plugin/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. r cCs0i}t|dgdurt|ddgt|dg|S)Nrrrrrrr_GetDocumentParameters_to_mldev<sr"cCsXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|S)N page_sizerpageSize page_token pageTokenrrrrr_ListDocumentsConfig_to_mldevGs r'cCr)Nparentrr)rrr'rrrr!_ListDocumentsParameters_to_mldev\r!r)cCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgD|S)NsdkHttpResponsesdk_http_response nextPageTokennext_page_token documentscSsg|]}|qSrr).0itemrrr {sz5_ListDocumentsResponse_from_mldev..rrrrr!_ListDocumentsResponse_from_mldevjsr2c@eZdZdddedeejdejfddZdddedeej ddfdd Z ddd edeej dej fd d Z ddd edeej deejfd dZdS) DocumentsNrrrrc Cs(tj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||} | jsaint| j} tjj| t|ddrdt|j d dt|j d dt|j d dd inid} |j| | SaGets 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. rriThis method is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.r{name}r?rNgetresponse_schemaresponse_json_schemainclude_all_fieldsr<r=r>responsekwargs)r _GetDocumentParameters _api_clientvertexai ValueErrorr"r; format_maprpopr http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsDocument_from_responsegetattr_verify_response selfrrparameter_model request_dictrequest_url_dictpath query_paramsrIrA response_dict return_valuerrrr;sZ          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 r7r8rr9rr:rNdelete)r _DeleteDocumentParametersrDrErFr r;rGrrHrrIr rJrKrL rUrrrVrWrXrYrZrIrrrr^s0        zDocuments.deleter(c Cs8tj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd|||} | jsaint| j} |jjsot| } tjj| t|ddrdt|j d dt|j d dt|j d dd inid } |j| | SNr(rr8rz{parent}/documentsrr:rr;r<r=r>r?r@)r _ListDocumentsParametersrDrErFr)r;rGrrHrrIr rJrKrLrMrNrOr2ListDocumentsResponserQrRrS rUr(rrVrWrXrYrZrIrAr[r\rrr_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.rb)rrfrrUr(r list_requestrrrlistJs zDocuments.list)__name__ __module__ __qualname__strrr GetDocumentConfigOrDictrPr;DeleteDocumentConfigOrDictr^ListDocumentsConfigOrDictrdrfrrjrrrrr4sB O : Jr4c@r3)AsyncDocumentsNr5rrrc s0tj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdH} | jseint| j} tjj| t|ddrdt|j d dt|j d dt|j d dd inid} |j| | Sr6)r rCrDrErFr"r;rGrrHrrIr rJrK async_requestrMrNrOrPrQrRrSrTrrrr;js`           zAsyncDocuments.getc stj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdHdSr])r r_rDrErFr r;rGrrHrrIr rJrKrsr`rrrr^s6        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| t|ddrdt|j d dt|j d dt|j d dd inid } |j| | Sra)r rcrDrErFr)r;rGrrHrrIr rJrKrsrMrNrOr2rdrQrRrSrerrrrfsd           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}") rgr.rbN)rrfr rhrrrrj7szAsyncDocuments.list)rkrlrmrnrr rorPr;rpr^rqrdrfr rjrrrrrrhsB Q < Lrr)N)# functoolsrrNloggingtypingrrr urllib.parserrr r r rr rpagersr r getLoggerloggerdictrnobjectrr r"r'r)r2 BaseModuler4rrrrrrs~                     h