lQj]ddlmZmZmZmZmZmZmZmZddl m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZddl m!Z!ddl"m#Z#ddl$Z$ddl%m&Z&erddl'm(Z(e$jRe*Z+erdd l,m-Z-Gd d e d Z.y) ) TYPE_CHECKINGOptionalUnionListcastDictAnyTuple)CollectionCommon)URICollectionMetadata Embedding PyEmbeddingIncludeIndexingStatusMetadataDocumentImageWhereIDs GetResult QueryResultID OneOrMany ReadLevel WhereDocument SearchResult DeleteResultmaybe_cast_one_to_many)UpdateCollectionConfiguration)SearchN)Function)AttachedFunction) ServerAPIceZdZejfdedefdZdefdZ d0de e de e e e e efde e ed e e ed e e ed e e eddfd Zddddddd gfde e e d e ede ede ede ededefdZd1dedefdZddddddddgdf de e e e e efde e ede e ede e ede e e ded e ede ededefdZ d2de ede ede eddfdZd eddfd!Z defd"Z!ejfd#e e"dede#fd$Z$ d0de e de e e e e efde e ed e e ed e e ed e e eddfd%Z% d0de e de e e e e efde e ed e e ed e e ed e e eddfd&Z& d3de e'd e ede ede ede(f d'Z) d4d(e*ded)ed*e e+ee,fde-d+e.ff d,Z/dedd+fd-Z0 d5ded.e.de.fd/Z1y)6 Collection read_levelreturnc||jj|j|j|j|S)aReturn the number of records in the collection. Args: read_level: Controls whether to read from the write-ahead log (WAL): - ReadLevel.INDEX_AND_WAL: Read from both the compacted index and WAL (default). All committed writes will be visible. - ReadLevel.INDEX_ONLY: Read only from the compacted index, skipping the WAL. Faster, but recent writes that haven't been compacted may not be visible. - ReadLevel.INDEX_AND_BOUNDED_WAL: Read from the index and up to a server-configured number of WAL entries for bounded query latency. ) collection_idtenantdatabaser')_client_countidr+r,)selfr's c/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/chromadb/api/models/Collection.pycountzCollection.count+s8||""'';;]]! #  cz|jj|j|j|jS)aGet the indexing status of this collection. Returns: IndexingStatus: An object containing: - num_indexed_ops: Number of user operations that have been indexed - num_unindexed_ops: Number of user operations pending indexing - total_ops: Total number of user operations in collection - op_indexing_progress: Proportion of user operations that have been indexed as a float between 0 and 1 r*r+r,)r-_get_indexing_statusr/r+r,r0s r1get_indexing_statuszCollection.get_indexing_status>s5||00'';;]]1  r3Nids embeddings metadatas documentsimagesurisc |j||||||}|jj|j|d|d|d|d|d|j|j y) aAdd records to the collection. Args: ids: Record IDs to add. embeddings: Embeddings to add. If None, embeddings are computed. metadatas: Optional metadata for each record. documents: Optional documents for each record. images: Optional images for each record. uris: Optional URIs for loading images. Raises: ValueError: If embeddings and documents are both missing. ValueError: If embeddings and documents are both provided. ValueError: If lengths of provided fields do not match. ValueError: If an ID already exists. r9r:r;r<r=r>r9r:r;r<r>r*r9r:r;r<r>r+r,N)!_validate_and_prepare_add_requestr-_addr/r+r,)r0r9r:r;r<r=r> add_requests r1addzCollection.addNs><<! =  ''E""<0!+.!+.V$;;]]  r3wherelimitoffsetwhere_documentincludec |j||||}|jj|j|d|d|d|d|||j|j  }|j ||dS)aRetrieve records from the collection. If no filters are provided, returns records up to ``limit`` starting at ``offset``. Args: ids: If provided, only return records with these IDs. where: A Where filter used to filter based on metadata values. limit: Maximum number of results to return. offset: Number of results to skip before returning. where_document: A WhereDocument filter used to filter based on K.DOCUMENT. include: Fields to include in results. Can contain "embeddings", "metadatas", "documents", "uris". Defaults to "metadatas" and "documents". Returns: GetResult: Retrieved records and requested fields as a GetResult object. )r9rFrIrJr9rFrIrJ) r*r9rFrIrJrGrHr+r,responserJ)!_validate_and_prepare_get_requestr-_getr/r+r,_transform_get_response) r0r9rFrGrHrIrJ get_request get_resultss r1getzCollection.gets2<<) = ll''''E"g&&'78 *;;]](  ++ +i*@,  r3 c|j|jj|j||j|j S)zReturn the first ``limit`` records from the collection. Args: limit: Maximum number of records to return. Returns: GetResult: Retrieved records and requested fields. )r*nr+r,)_transform_peek_responser-_peekr/r+r,)r0rGs r1peekzCollection.peeksF,, LL  "gg{{    r3)r;r< distancesquery_embeddings query_texts query_images query_uris n_resultsc  |j|||||||||  } |jj|j| d| d| d| d| d| d|j|j  } |j | | d S) aQuery for the K nearest neighbor records in the collection. This is a batch query API. Multiple queries can be performed at once by providing multiple embeddings, texts, or images. >>> query_1 = [0.1, 0.2, 0.3] >>> query_2 = [0.4, 0.5, 0.6] >>> results = collection.query( >>> query_embeddings=[query_1, query_2], >>> n_results=10, >>> ) If query_texts, query_images, or query_uris are provided, the collection's embedding function will be used to create embeddings before querying the API. The `ids`, `where`, `where_document`, and `include` parameters are applied to all queries. Args: query_embeddings: Raw embeddings to query for. query_texts: Documents to embed and query against. query_images: Images to embed and query against. query_uris: URIs to be loaded and embedded. ids: Optional subset of IDs to search within. n_results: Number of neighbors to return per query. where: Metadata filter. where_document: Document content filter. include: Fields to include in results. Can contain "embeddings", "metadatas", "documents", "uris", "distances". Defaults to "metadatas", "documents", "distances". Returns: QueryResult: Nearest neighbor results. Raises: ValueError: If no query input is provided. ValueError: If multiple query input types are provided. ) r[r\r]r^r9r_rFrIrJr9r:r_rFrIrJ) r*r9r[r_rFrIrJr+r,rL)#_validate_and_prepare_query_requestr-_queryr/r+r,_transform_query_response) r0r[r\r]r^r9r_rFrIrJ query_request query_resultss r1queryzCollection.querysv@@-#%!)A   ++''e$*<8#K0(()9:!),;;]],  --"M),D.  r3namemetadata configurationc|j||jj|j||||j|j |j |||y)zUpdate collection name, metadata, or configuration. Args: name: New collection name. metadata: New metadata for the collection. configuration: New configuration for the collection. )r/new_name new_metadatanew_configurationr+r,N)_validate_modify_requestr-_modifyr/r+r,"_update_model_after_modify_success)r0rgrhris r1modifyzCollection.modifys] %%h/ ww!+;;]]   //h Nr3rkc|jj|j||j|j}t |j||j |jS)aFork the current collection under a new name. The returning collection should contain identical data to the current collection. This 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*rkr+r,)clientmodelembedding_function data_loader)r-_forkr/r+r,r&_embedding_function _data_loader)r0rkrts r1forkzCollection.fork7s` ""'';;]] #  <<#77))   r3cz|jj|j|j|jS)zGet the number of forks that exist for this collection. This only works for Hosted Chroma for now. Returns: int: The number of forks for this collection. r5)r- _fork_countr/r+r,r7s r1 fork_countzCollection.fork_countQs5||'''';;]](  r3searchesc t|}|g}|Dcgc]}|j|}}|jj|jt t t||j|j|Scc}w)a Perform hybrid search on the collection. This is an experimental API that only works for distributed and hosted Chroma for now. Args: searches: A single Search object or a list of Search objects, each containing: - where: Where expression for filtering - rank: Ranking expression for hybrid search (defaults to Val(0.0)) - limit: Limit configuration for pagination (defaults to no limit) - select: Select configuration for keys to return (defaults to empty) read_level: Controls whether to read from the write-ahead log (WAL): - ReadLevel.INDEX_AND_WAL: Read from both the compacted index and WAL (default). All committed writes will be visible. - ReadLevel.INDEX_ONLY: Read only from the compacted index, skipping the WAL. Faster, but recent writes that haven't been compacted may not be visible. - ReadLevel.INDEX_AND_BOUNDED_WAL: Read from the index and up to a server-configured number of WAL entries for bounded query latency. Returns: SearchResult: Column-major format response with: - ids: List of result IDs for each search payload - documents: Optional documents for each payload - embeddings: Optional embeddings for each payload - metadatas: Optional metadata for each payload - scores: Optional scores for each payload - select: List of selected keys for each payload Raises: NotImplementedError: For local/segment API implementations Examples: # Using builder pattern with Key constants from chromadb.execution.expression import ( Search, Key, K, Knn, Val ) # Note: K is an alias for Key, so K.DOCUMENT == Key.DOCUMENT search = (Search() .where((K("category") == "science") & (K("score") > 0.5)) .rank(Knn(query=[0.1, 0.2, 0.3]) * 0.8 + Val(0.5) * 0.2) .limit(10, offset=0) .select(K.DOCUMENT, K.SCORE, "title")) # Direct construction from chromadb.execution.expression import ( Search, Eq, And, Gt, Knn, Limit, Select, Key ) search = Search( where=And([Eq("category", "science"), Gt("score", 0.5)]), rank=Knn(query=[0.1, 0.2, 0.3]), limit=Limit(offset=0, limit=10), select=Select(keys={Key.DOCUMENT, Key.SCORE, "title"}) ) # Single search result = collection.search(search) # Multiple searches at once searches = [ Search().where(K("type") == "article").rank(Knn(query=[0.1, 0.2])), Search().where(K("type") == "paper").rank(Knn(query=[0.3, 0.4])) ] results = collection.search(searches) # Skip WAL for faster queries (may miss recent uncommitted writes) from chromadb.api.types import ReadLevel result = collection.search(search, read_level=ReadLevel.INDEX_ONLY) )r*r~r+r,r') r_embed_search_string_queriesr-_searchr/rrr!r+r,)r0r~r' searches_listsearchembedded_searchess r1rzCollection.search^sT/x8  MER :@D - -f 5  ||##''$v,(9:;;]]! $   sBc |j||||||}|jj|j|d|d|d|d|d|j|j y) a-Update existing records by ID. Records are provided in columnar format. If provided, the `embeddings`, `metadatas`, `documents`, and `uris` lists must be the same length. Entries in each list correspond to the same record. >>> ids = ["id1", "id2", "id3"] >>> embeddings = [[0.1, 0.2, 0.3], [0.4, 0.5, 0.6], [0.7, 0.8, 0.9]] >>> metadatas = [{"key": "value"}, {"key": "value"}, {"key": "value"}] >>> documents = ["document1", "document2", "document3"] >>> uris = ["uri1", "uri2", "uri3"] >>> collection.update(ids, embeddings, metadatas, documents, uris) If `embeddings` are not provided, the embeddings will be computed based on `documents` using the collection's embedding function. Args: ids: Record IDs to update. embeddings: Updated embeddings. If None, embeddings are computed. metadatas: Updated metadata. documents: Updated documents. images: Updated images. uris: Updated URIs for loading images. r@r9r:r;r<r>rAN)$_validate_and_prepare_update_requestr-_updater/r+r,)r0r9r:r;r<r=r>update_requests r1updatezCollection.updatesHBB! C  ''u%%l3$[1$[1';;]]  r3c |j||||||}|jj|j|d|d|d|d|d|j|j y) arCreate or update records by ID. Args: ids: Record IDs to upsert. embeddings: Embeddings to add or update. If None, embeddings are computed. metadatas: Metadata to add or update. documents: Documents to add or update. images: Images to add or update. uris: URIs for loading images. r@r9r:r;r<r>rAN)$_validate_and_prepare_upsert_requestr-_upsertr/r+r,)r0r9r:r;r<r=r>upsert_requests r1upsertzCollection.upserts0BB! C  ''u%%l3$[1$[1';;]]  r3c |j||||}|jj|j|d|d|d|d|j|j S)aDelete records by ID or filters. All documents that match the `ids` or `where` and `where_document` filters will be deleted. Args: ids: Record IDs to delete. where: Metadata filter. where_document: Document content filter. limit: Maximum number of records to delete. Can only be used with where or where_document filters. Returns: DeleteResult: A dict containing the number of records deleted. Raises: ValueError: If no IDs or filters are provided. ValueError: If limit is specified without a where or where_document clause. )rGr9rFrIrG)r*r9rFrIrGr+r,)$_validate_and_prepare_delete_requestr-_deleter/r+r,)r0r9rFrIrGdelete_requests r1deletezCollection.deletesw0BB eC ||##''u% ))*:; );;]]$  r3functionoutput_collectionparamsr#c t|tr |jn|}|jj |||j |||j |jS)aAttach a function to this collection. Args: function: A Function enum value (e.g., STATISTICS_FUNCTION, RECORD_COUNTER_FUNCTION) name: Unique name for this attached function output_collection: Name of the collection where function output will be stored params: Optional dictionary with function-specific parameters Returns: Tuple of (AttachedFunction, created) where created is True if newly created, False if already existed (idempotent request) Example: >>> from chromadb.api.functions import STATISTICS_FUNCTION >>> attached_fn = collection.attach_function( ... function=STATISTICS_FUNCTION, ... name="mycoll_stats_fn", ... output_collection="mycoll_stats", ... ) >>> if created: ... print("New function attached") ... else: ... print("Function already existed") ) function_idrginput_collection_idrrr+r,) isinstancer"valuer-attach_functionr/r+r,)r0rrgrrrs r1rzCollection.attach_functionCsU>)38X(FhnnH ||++# $/;;]],  r3c||jj||j|j|jS)aGet an attached function by name for this collection. Args: name: Name of the attached function Returns: AttachedFunction: The attached function object Raises: NotFoundError: If the attached function doesn't exist )rgrr+r,)r-get_attached_functionr/r+r,)r0rgs r1rz Collection.get_attached_functionms8||11 $;;]] 2  r3delete_output_collectionc~|jj||j||j|jS)aDetach a function from this collection. Args: name: The name of the attached function delete_output_collection: Whether to also delete the output collection. Defaults to False. Returns: bool: True if successful Example: >>> success = collection.detach_function("my_function", delete_output_collection=True) )rgr delete_outputr+r,)r-detach_functionr/r+r,)r0rgrs r1rzCollection.detach_functions;"||++ $2;;]] ,  r3)NNNNN)rT)NNN)NNNN)N)F)2__name__ __module__ __qualname__r INDEX_AND_WALintr2rr8rrrrrrrrrr rErrrrrSrYrrfstrr r rqrzr}r!rrrrrrrr"rr r boolrrrr3r1r&r&*s,5,C,C   & ^ 0 3737-1)-1 r]1  )$+&(  1 Ih/01 Ih/01 5)*1 y~&1  1 j(,!%# $26'5- im $- - } -  - !/ - -  - ^ # y 2 5937/3'+!%26 U " )$+&(  U i12U y/0U Ys^,U im $U U U !/U U ( )U r#15AE OsmO-.O => O  O<    4  C  !* 7 7Y F#Y Y   Y F 3737-1)-6 r]6  )$+&(  6 Ih/06 Ih/06 5)*6 y~&6  6 @ 3737-1)-* r]*  )$+&(  * Ih/0* Ih/0* 5)** y~&*  * \"!%26# $ c]$ $ !/ $ } $  $ V,0 ( ( (  ( c3h( ( !4' ( ( T # 2D ,*/  #'    r3r&r$)/typingrrrrrrr r $chromadb.api.models.CollectionCommonr chromadb.api.typesr r rrrrrrrrrrrrrrrrrr%chromadb.api.collection_configurationr "chromadb.execution.expression.planr!loggingchromadb.api.functionsr"$chromadb.api.models.AttachedFunctionr# getLoggerrlogger chromadb.apir$r&rr3r1rspOOOA,P5+E   8 $&m  !+.m  r3