o iA @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_objectr\/var/www/html/karishye-ai-python/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_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)NsdkHttpResponsesdk_http_response nextPageTokennext_page_token documentscSsg|]}|qSrr).0itemrrr |sz5_ListDocumentsResponse_from_mldev..rrrrr!_ListDocumentsResponse_from_mldevksr2c@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"r; format_maprpopr http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsDocument_from_response model_dump_verify_response selfrrparameter_model request_dictrequest_url_dictpath query_paramsrEr= response_dict return_valuerrrr;s<         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 _DeleteDocumentParametersr@rArBr r;rCrrDrrEr rFrGrH rQrrrRrSrTrUrVrEr=rrrrZs4        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 a<Lists all Documents in a FileSearchStore. Args: parent (str): The name of the FileSearchStore containing the Documents. config (ListDocumentsConfig | None): Optional parameters for the request, such as page_size. Returns: ListDocumentsResponse: A paginated list of Documents. r(rr8rz{parent}/documentsrr:rNr;r<)r _ListDocumentsParametersr@rArBr)r;rCrrDrrEr rFrGrHrIrJrKr2ListDocumentsResponserMrNrO rQr(rrRrSrTrUrVrEr=rWrXrrr_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.r^)rrbrrQr(r list_requestrrrlist7s zDocuments.list)__name__ __module__ __qualname__strrr GetDocumentConfigOrDictrLr;DeleteDocumentConfigOrDictrZListDocumentsConfigOrDictr`rbrrfrrrrr4sB ? ; Er4c@r3)AsyncDocumentsNr5rrrc 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| | Sr6)r r?r@rArBr"r;rCrrDrrEr rFrG async_requestrIrJrKrLrMrNrOrPrrrr;VsB          zAsyncDocuments.getc stj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdH} dSrY)r r[r@rArBr r;rCrrDrrEr rFrGror\rrrrZs6        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| | Sr])r r_r@rArBr)r;rCrrDrrEr rFrGrorIrJrKr2r`rMrNrOrarrrrbsF          zAsyncDocuments._listcs.t|j|d}td||j||dIdH|S)a2Lists 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.documents.list(parent='rag_store_name'): print(f"document: {document.name} - {document.display_name}") rcr.r^N)rrbr rdrrrrf szAsyncDocuments.list)rgrhrirjrr rkrLr;rlrZrmr`rbr rfrrrrrnTsB A ; Grn)N)# functoolsrrJloggingtypingrrr urllib.parserrr r r rr rpagersr r getLoggerloggerdictrjobjectrr r"r'r)r2 BaseModuler4rnrrrrs~                     S