o lWi8@sddlmZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZd d lmZmZd d lmZejrWd dlmZmZd dlmZmZeejdZ GdddZ!GdddZ"dS)) annotationsN)AsyncClientWrapperSyncClientWrapper)RequestOptions)!GetKnowledgeBaseListResponseModel)GetOrCreateRagIndexRequestModel)KnowledgeBaseDocumentType)KnowledgeBaseSortBy) SortDirection)AsyncRawKnowledgeBaseClientRawKnowledgeBaseClient)/KnowledgeBaseGetOrCreateRagIndexesResponseValue)AsyncDocumentClientDocumentClient)AsyncDocumentsClientDocumentsClient.c @reZdZd+ddZed,ddZd d d d d d d d d d d d d d-dd Zd d!d.d%d&Zed'd(Zed)d*Z d S)/KnowledgeBaseClientclient_wrapperrcC"t|d|_||_d|_d|_dSNr)r _raw_client_client_wrapper _documents _documentselfrr v/var/www/html/asistente-voz-ia/venv/lib/python3.10/site-packages/elevenlabs/conversational_ai/knowledge_base/client.py__init__  zKnowledgeBaseClient.__init__returnrcC|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawKnowledgeBaseClient rrr r r!with_raw_response! z%KnowledgeBaseClient.with_raw_responseN page_sizesearchshow_only_owned_documentstypesparent_folder_idancestor_folder_id folders_firstsort_directionsort_by use_typesensecursorrequest_optionsr+typing.Optional[int]r,typing.Optional[str]r-typing.Optional[bool]r.dtyping.Optional[typing.Union[KnowledgeBaseDocumentType, typing.Sequence[KnowledgeBaseDocumentType]]]r/r0r1r2typing.Optional[SortDirection]r3$typing.Optional[KnowledgeBaseSortBy]r4r5r6typing.Optional[RequestOptions]rc Cs*|jj||||||||| | | | d } | jS)a Get a list of available knowledge base documents Parameters ---------- page_size : typing.Optional[int] How many documents to return at maximum. Can not exceed 100, defaults to 30. search : typing.Optional[str] If specified, the endpoint returns only such knowledge base documents whose names start with this string. show_only_owned_documents : typing.Optional[bool] If set to true, the endpoint will return only documents owned by you (and not shared from somebody else). types : typing.Optional[typing.Union[KnowledgeBaseDocumentType, typing.Sequence[KnowledgeBaseDocumentType]]] If present, the endpoint will return only documents of the given types. parent_folder_id : typing.Optional[str] If set, the endpoint will return only documents that are direct children of the given folder. ancestor_folder_id : typing.Optional[str] If set, the endpoint will return only documents that are descendants of the given folder. folders_first : typing.Optional[bool] Whether folders should be returned first in the list of documents. sort_direction : typing.Optional[SortDirection] The direction to sort the results sort_by : typing.Optional[KnowledgeBaseSortBy] The field to sort the results by use_typesense : typing.Optional[bool] If set to true, the endpoint will use typesense DB to search for the documents). cursor : typing.Optional[str] Used for fetching next page. Cursor is returned in the response. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- GetKnowledgeBaseListResponseModel Successful Response Examples -------- from elevenlabs import ElevenLabs client = ElevenLabs( api_key="YOUR_API_KEY", ) client.conversational_ai.knowledge_base.list( page_size=1, search="search", show_only_owned_documents=True, parent_folder_id="parent_folder_id", ancestor_folder_id="ancestor_folder_id", folders_first=True, sort_direction="asc", sort_by="name", use_typesense=True, cursor="cursor", ) r*rlistdatarr+r,r-r.r/r0r1r2r3r4r5r6 _responser r r!r?,sTzKnowledgeBaseClient.listr6items0typing.Sequence[GetOrCreateRagIndexRequestModel]Atyping.Dict[str, KnowledgeBaseGetOrCreateRagIndexesResponseValue]cCs|jj||d}|jS)a Retrieves and/or creates RAG indexes for multiple knowledge base documents in a single request. Parameters ---------- items : typing.Sequence[GetOrCreateRagIndexRequestModel] List of requested RAG indexes. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- typing.Dict[str, KnowledgeBaseGetOrCreateRagIndexesResponseValue] Successful Response Examples -------- from elevenlabs import ElevenLabs, GetOrCreateRagIndexRequestModel client = ElevenLabs( api_key="YOUR_API_KEY", ) client.conversational_ai.knowledge_base.get_or_create_rag_indexes( items=[ GetOrCreateRagIndexRequestModel( document_id="document_id", create_if_missing=True, model="e5_mistral_7b_instruct", ) ], ) rDr6rget_or_create_rag_indexesr@rrDr6rBr r r!rIs'z-KnowledgeBaseClient.get_or_create_rag_indexescC*|jdurddlm}||jd|_|jS)Nr )rr)rdocuments.clientrr)rrr r r! documents  zKnowledgeBaseClient.documentscCrK)Nr )rr)rdocument.clientrr)rrr r r!documentrNzKnowledgeBaseClient.document)rr)r$rr+r7r,r8r-r9r.r:r/r8r0r8r1r9r2r;r3r<r4r9r5r8r6r=r$rrDrEr6r=r$rF __name__ __module__ __qualname__r"propertyr(r?rIrMrPr r r r!rs.   h* rc @r)/AsyncKnowledgeBaseClientrrcCrr)r rrrrrr r r!r"r#z!AsyncKnowledgeBaseClient.__init__r$r cCr%)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawKnowledgeBaseClient r&r'r r r!r(r)z*AsyncKnowledgeBaseClient.with_raw_responseNr*r+r7r,r8r-r9r.r:r/r0r1r2r;r3r<r4r5r6r=rc s2|jj||||||||| | | | d IdH} | jS)a) Get a list of available knowledge base documents Parameters ---------- page_size : typing.Optional[int] How many documents to return at maximum. Can not exceed 100, defaults to 30. search : typing.Optional[str] If specified, the endpoint returns only such knowledge base documents whose names start with this string. show_only_owned_documents : typing.Optional[bool] If set to true, the endpoint will return only documents owned by you (and not shared from somebody else). types : typing.Optional[typing.Union[KnowledgeBaseDocumentType, typing.Sequence[KnowledgeBaseDocumentType]]] If present, the endpoint will return only documents of the given types. parent_folder_id : typing.Optional[str] If set, the endpoint will return only documents that are direct children of the given folder. ancestor_folder_id : typing.Optional[str] If set, the endpoint will return only documents that are descendants of the given folder. folders_first : typing.Optional[bool] Whether folders should be returned first in the list of documents. sort_direction : typing.Optional[SortDirection] The direction to sort the results sort_by : typing.Optional[KnowledgeBaseSortBy] The field to sort the results by use_typesense : typing.Optional[bool] If set to true, the endpoint will use typesense DB to search for the documents). cursor : typing.Optional[str] Used for fetching next page. Cursor is returned in the response. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- GetKnowledgeBaseListResponseModel Successful Response Examples -------- import asyncio from elevenlabs import AsyncElevenLabs client = AsyncElevenLabs( api_key="YOUR_API_KEY", ) async def main() -> None: await client.conversational_ai.knowledge_base.list( page_size=1, search="search", show_only_owned_documents=True, parent_folder_id="parent_folder_id", ancestor_folder_id="ancestor_folder_id", folders_first=True, sort_direction="asc", sort_by="name", use_typesense=True, cursor="cursor", ) asyncio.run(main()) r*Nr>rAr r r!r?s \ zAsyncKnowledgeBaseClient.listrCrDrErFcs|jj||dIdH}|jS)a Retrieves and/or creates RAG indexes for multiple knowledge base documents in a single request. Parameters ---------- items : typing.Sequence[GetOrCreateRagIndexRequestModel] List of requested RAG indexes. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- typing.Dict[str, KnowledgeBaseGetOrCreateRagIndexesResponseValue] Successful Response Examples -------- import asyncio from elevenlabs import AsyncElevenLabs, GetOrCreateRagIndexRequestModel client = AsyncElevenLabs( api_key="YOUR_API_KEY", ) async def main() -> None: await client.conversational_ai.knowledge_base.get_or_create_rag_indexes( items=[ GetOrCreateRagIndexRequestModel( document_id="document_id", create_if_missing=True, model="e5_mistral_7b_instruct", ) ], ) asyncio.run(main()) rGNrHrJr r r!rIIs/z2AsyncKnowledgeBaseClient.get_or_create_rag_indexescCrK)Nr )rr)rrLrr)rrr r r!rM{rNz"AsyncKnowledgeBaseClient.documentscCrK)Nr )rr)rrOrr)rrr r r!rPrNz!AsyncKnowledgeBaseClient.document)rr)r$r rQrRrSr r r r!rXs.   p2 rX)# __future__rtypingcore.client_wrapperrrcore.request_optionsr,types.get_knowledge_base_list_response_modelr+types.get_or_create_rag_index_request_modelr"types.knowledge_base_document_typer types.knowledge_base_sort_byr types.sort_directionr raw_clientr r=types.knowledge_base_get_or_create_rag_indexes_response_valuer TYPE_CHECKINGrOrrrLrrcastAnyOMITrrXr r r r!s$        2