쎒jJ RddlmZddlZddlZddlmZmZmZddlm Z ddl m Z ddl m Z ddl m Z dd l mZdd l mZdd lmZmZejd Z dd eeeefefdeeeefdeeeffdZ dd eeeefefdeeeefdeeeffdZ dd eeeefefdeeeefdeeeffdZ dd eeeefefdeeeefdeeeffdZ dd eeeefefdeeeefdeeeffdZ dd eeeefefdeeeefdeeeffdZGdde j Z!Gdde j Z"dS))partialN)AnyOptionalUnion) urlencode) _api_module)_common)types)get_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.documents from_object parent_objectreturncpi}t|dg"t|ddgt|dg|S)Nforce_querygetvsetvrr to_objects DC:\Projects\DSCons\.venv\Lib\site-packages\google\genai\documents.py_DeleteDocumentConfig_to_mldevr"sD!) +y!!-7+T+y-I-IJJJ ci}t|dg"t|ddgt|dgt|dgtt|dg||S)Nname_urlconfig)rrrrs r"_DeleteDocumentParameters_to_mldevr".sr!) +x  ,VV$d;&A&ABBB +z""."4 hZ#@#@)LLL rcpi}t|dg"t|ddgt|dg|S)Nrr rrs r_GetDocumentParameters_to_mldevr$<sD!) +x  ,VV$d;&A&ABBB rci}t|dg"t|ddgt|dgt|dg"t|ddgt|dg|S)N page_sizerpageSize page_token pageTokenrrs r_ListDocumentsConfig_to_mldevr*Gs!) + }%%1*-tK+/O/O + ~&&2 ; [<.)) rci}t|dg"t|ddgt|dgt|dgtt|dg||S)Nparentr r!)rrr*rs r!_ListDocumentsParameters_to_mldevr-\sr!) +z"".VX&[8*(E(EFFF +z"".!${XJ"?"?KKK rc Ji}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg+t|dgdt|dgD|S)NsdkHttpResponsesdk_http_response nextPageTokennext_page_token documentscg|]}|Sr5).0items r z5_ListDocumentsResponse_from_mldev..{s;;;$;;;rrrs r!_ListDocumentsResponse_from_mldevr9js!) +)*++7'(${=N00 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !1L& %%l333 rc2tj||}|jjrt dt |}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|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@r rArrBr!Ndelete)r _DeleteDocumentParametersrLrMrNr"rCrOrrPr!rQr rRrSrT r]rr!r^r_r`rarbrQs rrgzDocuments.deletesB"5 O     4   8HHl%))&11 ""#344##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDLXt\<HHHHHrr,c tj||}|jjrt dt |}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd|||} | jsint#j| j} |jjst'| } tj| t-|ddrDdt-|j d dt-|j d dt-|j d dd ini } |j| | SNr,r!r@r z{parent}/documentsrrBr!rCrDrErFrGrH)r _ListDocumentsParametersrLrMrNr-rCrOrrPr!rQr rRrSrTrUrVrWr9ListDocumentsResponserYrZr[ r]r,r!r^r_r`rarbrQrIrcrds r_listzDocuments._lists0 4O   $  4   7GGl%))&11 $#../?@@###H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''t\<PPH&mJBBHM1J1JM   $G7 FFm.== ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !>L& %%l333 rc|t|j|}td|||||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,r3rl)rrprr]r,r! list_requests rlistzDocuments.listJsF*4:f555L  & 00   r)__name__ __module__ __qualname__strrr GetDocumentConfigOrDictrXrCDeleteDocumentConfigOrDictrgListDocumentsConfigOrDictrnrprrur5rrr;r;s_MQIIII"*5+H"II ~IIII^<@ 4I4I4I 4Iu78 4I  4I4I4I4It;? DDDDu67 D  " DDDDT;? u67   U^ rr;ceZdZdddedeejdejfdZdddedeej ddfdZ dddedeej dej fd Z dddedeej deejfd ZdS) AsyncDocumentsNr<rr!rc Ktj||}|jjrt dt |}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd |||d{V} | jsint#j| j} tj| t+|ddrDdt+|j d dt+|j d dt+|j d dd ini} |j| | Sr>)r rKrLrMrNr$rCrOrrPr!rQr rRrS async_requestrUrVrWrXrYrZr[r\s rrCzAsyncDocuments.getjs<2 O     4   5_EEl%))&11 ""#344##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33 t\<H'mJBBHM1J1JM>00 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !1L& %%l333 rcBKtj||}|jjrt dt |}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd |||d{VdSrf)r rhrLrMrNr"rCrOrrPr!rQr rRrSrris rrgzAsyncDocuments.deleteso"5 O     4   8HHl%))&11 ""#344##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL   ( ($ l  rr,c Ktj||}|jjrt dt |}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd|||d{V} | jsint#j| j} |jjst'| } tj| t-|ddrDdt-|j d dt-|j d dt-|j d dd ini } |j| | Srk)r rmrLrMrNr-rCrOrrPr!rQr rRrSrrUrVrWr9rnrYrZr[ros rrpzAsyncDocuments._listsY 4O   $  4   7GGl%))&11 $#../?@@###H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33 t\<H'mJBBHM1J1JM   $G7 FFm.== ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !>L& %%l333 rcKt|j|}td||||d{V|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}") rrr3rlN)rrprrss rruzAsyncDocuments.list7s\,4:f555L jjvj66666666   r)rvrwrxryrr rzrXrCr{rgr|rnrprrur5rrr~r~hsVMQKKKK"*5+H"IK ~KKKKb<@ 666 6u78 6  6666x;? FFFFu67 F  " FFFFX;? u67  %.! rr~)N)# functoolsrrVloggingtypingrrr urllib.parserr r r r rr rpagersrr getLoggerloggerdictryobjectrr"r$r*r-r9 BaseModuler;r~r5rrrs$ ''''''''''""""""............%%%%%%%%  3 4 4 /3  tCH~v-. DcN+  #s(^    /3  tCH~v-. DcN+  #s(^     /3tCH~v-.DcN+ #s(^/3tCH~v-.DcN+ #s(^./3  tCH~v-. DcN+  #s(^     /3tCH~v-.DcN+ #s(^.ddddd &dddNkkkkk[+kkkkkr