o ɭRh;@sddlmZmZmZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZddlmZddlZeeZerIddlmZGddded ZdS) ) TYPE_CHECKINGOptionalUnion)CollectionCommon)URICollectionMetadata Embedding PyEmbeddingIncludeMetadataDocumentImageWhereIDs GetResult QueryResultID OneOrMany WhereDocument)UpdateCollectionConfigurationN) ServerAPIc@seZdZdefddZ     d-deedeeee ee fdeee deee d eee d eeeddfd d Zdddddddgfdeeed eedeedeedeededefddZd.dedefddZddddddddgdf deeee ee fdeee deee deeedeeeded eedeededefddZ   d/deed eed!eeddfd"d#Zd$eddfd%d&Z     d-deedeeee ee fdeee deee d eee d eeeddfd'd(Z     d-deedeeee ee fdeee deee d eee d eeeddfd)d*Z   d/deed eedeeddfd+d,Z dS)0 CollectionreturncCs|jj|j|j|jdS)zThe total number of embeddings added to the database Returns: int: The total number of embeddings added to the database ) collection_idtenantdatabase)_client_countidrr)selfr `/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/chromadb/api/models/Collection.pycount s zCollection.countNids embeddings metadatas documentsimagesurisc CP|j||||||d}|jj|j|d|d|d|d|d|j|jddS) 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 r#r$r%r&r'r(r#r$r%r&r(rr#r$r%r&r(rrN)!_validate_and_prepare_add_requestr_addrrr)rr#r$r%r&r'r( add_requestr r r!add-s$#  zCollection.addwherelimitoffsetwhere_documentincludec CsX|j||||d}|jj|j|d|d|d|d|||j|jd }|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. )r#r0r3r4r#r0r3r4) rr#r0r3r4r1r2rrresponser4)!_validate_and_prepare_get_requestr_getrrr_transform_get_response) rr#r0r1r2r3r4 get_request get_resultsr r r!getds( zCollection.get cCs ||jj|j||j|jdS)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. )rnrr)_transform_peek_responser_peekrrr)rr1r r r!peeks zCollection.peek)r%r& distancesquery_embeddings query_texts query_images query_uris n_resultsc Csj|j||||||||| d } |jj|j| d| d| d| d| d| d|j|jd } |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. query_uris: The URIs to be used with data loader. 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 ) rCrDrErFr#rGr0r3r4r#r$rGr0r3r4) rr#rCrGr0r3r4rrr5)#_validate_and_prepare_query_requestr_queryrrr_transform_query_response) rrCrDrErFr#rGr0r3r4 query_request query_resultsr r r!querys2-  zCollection.querynamemetadata configurationcCs:|||jj|j||||j|jd||||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 )rnew_name new_metadatanew_configurationrrN)_validate_modify_requestr_modifyrrr"_update_model_after_modify_success)rrNrOrPr r r!modifys  zCollection.modifyrQcCs0|jj|j||j|jd}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. )rrQrr)clientmodelembedding_function data_loader)r_forkrrrr_embedding_function _data_loader)rrQrYr r r!fork s zCollection.forkc Cr)) 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*r#r$r%r&r(r+N)$_validate_and_prepare_update_requestr_updaterrr)rr#r$r%r&r'r(update_requestr r r!update'$  zCollection.updatec Cr)) 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*r#r$r%r&r(r+N)$_validate_and_prepare_upsert_requestr_upsertrrr)rr#r$r%r&r'r(upsert_requestr r r!upsertTrdzCollection.upsertcCs<||||}|jj|j|d|d|d|j|jddS)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 r#r0r3)rr#r0r3rrN)$_validate_and_prepare_delete_requestr_deleterrr)rr#r0r3delete_requestr r r!deletes zCollection.delete)NNNNN)r=)NNN)!__name__ __module__ __qualname__intr"rrrrrr r r r rr/rrr rr<rArrMstrrrrWr_rcrhrrlr r r r!rsN     9  .       K   "     5     /rr)typingrrr$chromadb.api.models.CollectionCommonrchromadb.api.typesrrrr r r r r rrrrrrr%chromadb.api.collection_configurationrlogging getLoggerrmlogger chromadb.apirrr r r r!s D