o j6h"@sddlmZddlmZmZmZddlmmZ ddl m Z ddl m Z ddl mZddlmZddlmZdd lmZ    d-d.ddZ    d-d/ddZ  d0d1ddZ  d0d2ddZ   d3d4d d!Z   d3d5d"d#Zdddd$d6d(d)Zdddd$d7d+d,ZdS)8) annotations) AsyncIterableIterableOptionalN)protos)get_default_retriever_client)"get_default_retriever_async_client) helper_types) idecode_time)retriever_typesname str | None display_nameclient!glm.RetrieverServiceClient | Nonerequest_options&helper_types.RequestOptionsType | Nonereturnretriever_types.CorpuscCs|duri}|dur t}|durtj|d}nt|r(tjd||d}n ttjjt||dtj |d}|j |fi|}t | |}t |dt |dtjd i|}|S) aCalls the API to create a new `Corpus` by specifying either a corpus resource name as an ID or a display name, and returns the created `Corpus`. Args: name: The corpus resource name (ID). The name must be alphanumeric and fewer than 40 characters. display_name: The human readable display name. The display name must be fewer than 128 characters. All characters, including alphanumeric, spaces, and dashes are supported. request_options: Options for the request. Return: `retriever_types.Corpus` object with specified name or display name. Raises: ValueError: When the name is not specified or formatted incorrectly. Nrcorpora/r rlengthr corpus create_time update_time)rrCorpusr valid_name ValueErrorNAME_ERROR_MSGformatlenCreateCorpusRequest create_corpustypeto_dictr r rrrrrequestresponserrl/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/google/generativeai/retriever.pyr&s     r&&glm.RetrieverServiceAsyncClient | Nonecs|duri}|durt}|durtj|d}nt|r)tjd||d}n ttjjt||dtj |d}|j |fi|IdH}t | |}t |dt |dtjd i|}|S) z7This is the async version of `retriever.create_corpus`.Nrrrrrrrr)rrrr r r!r"r#r$r%r&r'r(r r)rrr,create_corpus_asyncJs"    r.strcCs~|duri}|dur t}d|vrd|}tj|d}|j|fi|}t||}t|dt|dtjdi|}|S)zCalls the API to fetch a `Corpus` by name and returns the `Corpus`. Args: name: The `Corpus` name. request_options: Options for the request. Return: a `retriever_types.Corpus` of interest. N/rr rrr) rrGetCorpusRequest get_corpusr'r(r r rr rrr*r+rrr,r3gs   r3cs|duri}|durt}d|vrd|}tj|d}|j|fi|IdH}t||}t|dt|dtjdi|}|S)z4This is the async version of `retriever.get_corpus`.Nr0rr1rrr) rrr2r3r'r(r r rr4rrr,get_corpus_asyncs   r5FforceboolcCsN|duri}|dur t}d|vrd|}tj||d}|j|fi|dS)abCalls the API to remove a `Corpus` from the service, optionally deleting associated `Document`s and objects if the `force` parameter is set to true. Args: name: The `Corpus` name. force: If set to true, any `Document`s and objects related to this `Corpus` will also be deleted. request_options: Options for the request. Nr0rr r6)rrDeleteCorpusRequest delete_corpusr r6rrr*rrr,r:sr:csV|duri}|durt}d|vrd|}tj||d}|j|fi|IdHdS)z7This is the async version of `retriever.delete_corpus`.Nr0rr8)rrr9r:r;rrr,delete_corpus_asyncsr<) page_sizerrr= Optional[int] Iterable[retriever_types.Corpus]ccsx|duri}|durt}tj|d}|j|fi|D]}t||}t|dt|dtjdi|VqdS)aYCalls the API to list all `Corpora` in the service and returns a list of paginated `Corpora`. Args: page_size: Maximum number of `Corpora` to request. page_token: A page token, received from a previous ListCorpora call. request_options: Options for the request. Return: Paginated list of `Corpora`. Nr=rrr) rrListCorporaRequest list_corporar'r(r r rr=rrr*rrrr,rBs   rB%AsyncIterable[retriever_types.Corpus]cCs|duri}|durt}tj|d}|j|fi|IdH2z 3dHW}t||}t|dt|dtjdi|Vq 6dS)z6This is the async version of `retriever.list_corpora`.Nr@rrr) rrrArBr'r(r r rrCrrr,list_corpora_asyncs $  rE)NNNN) r r rr rrrrrr) r r rr rr-rrrr)NN)r r/rrrrrr)r r/rr-rrrr)FNN)r r/r6r7rrrr)r r/r6r7rr-rr)r=r>rrrrrr?)r=r>rrrrrrD) __future__rtypingrrrgoogle.ai.generativelanguageaigenerativelanguageglmgoogle.generativeairgoogle.generativeai.clientrrgoogle.generativeai.typesr %google.generativeai.types.model_typesr r r&r.r3r5r:r<rBrErrrr,sR        -  "