o ɭRh;@sddlmZmZmZddlmZmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZddlmZddlmZer@ddlmZGdddedZd S) ) TYPE_CHECKINGOptionalUnion)URICollectionMetadata Embedding PyEmbeddingIncludeMetadataDocumentImageWhereIDs GetResult QueryResultID OneOrMany WhereDocument)CollectionCommon)UpdateCollectionConfiguration)AsyncServerAPIc@seZdZ     d-deedeeeeeefdeee deee deee deee ddfd d Z defd d Zdddddddgfdeeed eedeedeedeededefddZd.dedefddZddddddddgdf deeeeeefdeee deee deee deeeded eedeededefddZ   d/deed eed!eeddfd"d#Zd$eddfd%d&Z     d-deedeeeeeefdeee deee deee deee ddfd'd(Z     d-deedeeeeeefdeee deee deee deee ddfd)d*Z   d/deed eedeeddfd+d,Z dS)0AsyncCollectionNids embeddings metadatas documentsimagesurisreturnc X|j||||||d}|jj|j|d|d|d|d|d|j|jdIdHdS) a]Add embeddings to the data store. Args: ids: The ids of the embeddings you wish to add embeddings: The embeddings to add. If None, embeddings will be computed based on the documents or images using the embedding_function set for the Collection. Optional. metadatas: The metadata to associate with the embeddings. When querying, you can filter on this metadata. Optional. documents: The documents to associate with the embeddings. Optional. images: The images to associate with the embeddings. Optional. uris: The uris of the images to associate with the embeddings. Optional. Returns: None Raises: ValueError: If you don't provide either embeddings or documents ValueError: If the length of ids, embeddings, metadatas, or documents don't match ValueError: If you don't provide an embedding function and don't provide embeddings ValueError: If you provide both embeddings and documents ValueError: If you provide an id that already exists rrrrrrrrrrr collection_idrrrrrtenantdatabaseN)!_validate_and_prepare_add_request_client_addidr#r$)selfrrrrrr add_requestr+e/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/chromadb/api/models/AsyncCollection.pyadds&" zAsyncCollection.addcs |jj|j|j|jdIdHS)zThe total number of embeddings added to the database Returns: int: The total number of embeddings added to the database )r"r#r$N)r&_countr(r#r$)r)r+r+r,countSs  zAsyncCollection.countwherelimitoffsetwhere_documentincludec s`|j||||d}|jj|j|d|d|d|d|||j|jd IdH}|j||ddS) aGet embeddings and their associate data from the data store. If no ids or where filter is provided returns all embeddings up to limit starting at offset. Args: ids: The ids of the embeddings to get. Optional. where: A Where type dict used to filter results by. E.g. `{"$and": [{"color" : "red"}, {"price": {"$gte": 4.20}}]}`. Optional. limit: The number of documents to return. Optional. offset: The offset to start returning results from. Useful for paging results with limit. Optional. where_document: A WhereDocument type dict used to filter by the documents. E.g. `{"$contains": "hello"}`. Optional. include: A list of what to include in the results. Can contain `"embeddings"`, `"metadatas"`, `"documents"`. Ids are always included. Defaults to `["metadatas", "documents"]`. Optional. Returns: GetResult: A GetResult object containing the results. )rr0r3r4rr0r3r4) r"rr0r3r4r1r2r#r$Nresponser4)!_validate_and_prepare_get_requestr&_getr(r#r$_transform_get_response) r)rr0r1r2r3r4 get_request get_resultsr+r+r,get`s*  zAsyncCollection.get cs(||jj|j||j|jdIdHS)zGet the first few results in the database up to limit Args: limit: The number of results to return. Returns: GetResult: A GetResult object containing the results. )r"nr#r$N)_transform_peek_responser&_peekr(r#r$)r)r1r+r+r,peeks  zAsyncCollection.peek)rr distancesquery_embeddings query_texts query_images query_uris n_resultsc sr|j||||||||| d } |jj|j| d| d| d| d| d| d|j|jd Id H} |j| | dd S) aGet the n_results nearest neighbor embeddings for provided query_embeddings or query_texts. Args: query_embeddings: The embeddings to get the closes neighbors of. Optional. query_texts: The document texts to get the closes neighbors of. Optional. query_images: The images to get the closes neighbors of. Optional. ids: A subset of ids to search within. Optional. n_results: The number of neighbors to return for each query_embedding or query_texts. Optional. where: A Where type dict used to filter results by. E.g. `{"$and": [{"color" : "red"}, {"price": {"$gte": 4.20}}]}`. Optional. where_document: A WhereDocument type dict used to filter by the documents. E.g. `{"$contains": "hello"}`. Optional. include: A list of what to include in the results. Can contain `"embeddings"`, `"metadatas"`, `"documents"`, `"distances"`. Ids are always included. Defaults to `["metadatas", "documents", "distances"]`. Optional. Returns: QueryResult: A QueryResult object containing the results. Raises: ValueError: If you don't provide either query_embeddings, query_texts, or query_images ValueError: If you provide both query_embeddings and query_texts ValueError: If you provide both query_embeddings and query_images ValueError: If you provide both query_texts and query_images ) rCrDrErFrrGr0r3r4rrrGr0r3r4) r"rrCrGr0r3r4r#r$Nr5)#_validate_and_prepare_query_requestr&_queryr(r#r$_transform_query_response) r)rCrDrErFrrGr0r3r4 query_request query_resultsr+r+r,querys4,   zAsyncCollection.querynamemetadata configurationcsB|||jj|j||||j|jdIdH||||dS)zModify the collection name or metadata Args: name: The updated name for the collection. Optional. metadata: The updated metadata for the collection. Optional. Returns: None )r(new_name new_metadatanew_configurationr#r$N)_validate_modify_requestr&_modifyr(r#r$"_update_model_after_modify_success)r)rNrOrPr+r+r,modifys   zAsyncCollection.modifyrQcs8|jj|j||j|jdIdH}t|j||j|jdS)aFork the current collection under a new name. The returning collection should contain identical data to the current collection. This is an experimental API that only works for Hosted Chroma for now. Args: new_name: The name of the new collection. Returns: Collection: A new collection with the specified name and containing identical data to the current collection. )r"rQr#r$N)clientmodelembedding_function data_loader)r&_forkr(r#r$r_embedding_function _data_loader)r)rQrYr+r+r,fork s  zAsyncCollection.forkc r) aUpdate the embeddings, metadatas or documents for provided ids. Args: ids: The ids of the embeddings to update embeddings: The embeddings to update. If None, embeddings will be computed based on the documents or images using the embedding_function set for the Collection. Optional. metadatas: The metadata to associate with the embeddings. When querying, you can filter on this metadata. Optional. documents: The documents to associate with the embeddings. Optional. images: The images to associate with the embeddings. Optional. Returns: None r rrrrrr!N)$_validate_and_prepare_update_requestr&_updater(r#r$)r)rrrrrrupdate_requestr+r+r,update#& zAsyncCollection.updatec r) aOUpdate the embeddings, metadatas or documents for provided ids, or create them if they don't exist. Args: ids: The ids of the embeddings to update embeddings: The embeddings to add. If None, embeddings will be computed based on the documents using the embedding_function set for the Collection. Optional. metadatas: The metadata to associate with the embeddings. When querying, you can filter on this metadata. Optional. documents: The documents to associate with the embeddings. Optional. Returns: None r rrrrrr!N)$_validate_and_prepare_upsert_requestr&_upsertr(r#r$)r)rrrrrrupsert_requestr+r+r,upsertPrdzAsyncCollection.upsertcsD||||}|jj|j|d|d|d|j|jdIdHdS)a0Delete the embeddings based on ids and/or a where filter Args: ids: The ids of the embeddings to delete where: A Where type dict used to filter the delection by. E.g. `{"$and": [{"color" : "red"}, {"price": {"$gte": 4.20}}]}`. Optional. where_document: A WhereDocument type dict used to filter the deletion by the document content. E.g. `{"$contains": "hello"}`. Optional. Returns: None Raises: ValueError: If you don't provide either ids, where, or where_document rr0r3)r"rr0r3r#r$N)$_validate_and_prepare_delete_requestr&_deleter(r#r$)r)rr0r3delete_requestr+r+r,delete}szAsyncCollection.delete)NNNNN)r=)NNN)!__name__ __module__ __qualname__rrrrrrr r r rr-intr/r rr rr<rArrMstrrrrWr_rcrhrrlr+r+r+r,rsN      6  /       J   "     5     /rrN)typingrrrchromadb.api.typesrrrrr r r r r rrrrrr$chromadb.api.models.CollectionCommonr%chromadb.api.collection_configurationr chromadb.apirrr+r+r+r,sD