o 1Xxi7@sddlmZmZmZmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZddlmZddlmZGdddeZGd d d eZed ed ZGd ddeZGdddeZGdddeZdS))OptionalSequenceTypeVarType)abstractmethod) CollectionMetadataEmbeddingRecord OperationVectorEmbeddingRecordWhere WhereDocument VectorQueryVectorQueryResultSegmentSeqIdMetadata) ComponentSystem)UUID)Enumc@s$eZdZdZdZdZdZdZdZdS) SegmentTypez"urn:chroma:segment/metadata/sqlitez+urn:chroma:segment/vector/hnsw-local-memoryz.urn:chroma:segment/vector/hnsw-local-persistedz*urn:chroma:segment/vector/hnsw-distributedz#urn:chroma:segment/record/blockfilez%urn:chroma:segment/metadata/blockfileN) __name__ __module__ __qualname__SQLITEHNSW_LOCAL_MEMORYHNSW_LOCAL_PERSISTEDHNSW_DISTRIBUTEDBLOCKFILE_RECORDBLOCKFILE_METADATAr r X/home/air/biblejyuku/back/venv/lib/python3.10/site-packages/chromadb/segment/__init__.pyrsrc@sneZdZededefddZedefddZede fdd Z e d e de e fd d ZedddZd S)SegmentImplementationsytstemsegmentcCsdSNr )selfr#r$r r r!__init__szSegmentImplementation.__init__returncCdS)z,Get the number of embeddings in this segmentNr r&r r r!count#zSegmentImplementation.countcCr))z7Get the maximum SeqID currently indexed by this segmentNr r*r r r! max_seqid(r,zSegmentImplementation.max_seqidmetadatacCr))zGiven an arbitrary metadata map (e.g, from a collection), validate it and return metadata (if any) that is applicable and should be applied to the segment. Validation errors will be reported to the user.Nr )r.r r r!propagate_collection_metadata-z3SegmentImplementation.propagate_collection_metadataNcCr))z#Delete the segment and all its dataNr r*r r r!delete4r,zSegmentImplementation.delete)r(N)rrrrrrr'intr+rr- staticmethodrrr/r1r r r r!r"sr"S)boundc@s^eZdZdZe     d deedeedeee dee dee dee f d d Z dS) MetadataReaderz$Embedding Metadata segment interfaceNwherewhere_documentidslimitoffsetr(cCr))zQuery for embedding metadata.Nr )r&r7r8r9r:r;r r r! get_metadata@s zMetadataReader.get_metadata)NNNNN) rrr__doc__rrr r rstrr2rr<r r r r!r6=s* r6c@sTeZdZdZe d deeedeefddZ ede deee fdd Z dS) VectorReaderz"Embedding Vector segment interfaceNr9r(cCr))z]Get embeddings from the segment. If no IDs are provided, all embeddings are returned.Nr )r&r9r r r! get_vectorsPzVectorReader.get_vectorsquerycCr))zYGiven a vector query, return the top-k nearest neighbors for vector in the query.Nr )r&rBr r r! query_vectorsXrAzVectorReader.query_vectorsr%) rrrr=rrrr>r r@r rrCr r r r!r?Ms   r?c@s|eZdZdZededeefddZede dee fddZ ede d e e de fd d Z ede d edd fddZd S)SegmentManagerzfInterface for a pluggable strategy for creating, retrieving and instantiating segments as required collectionr(cCr))zsReturn the segments required for a new collection. Returns only segment data, does not persist to the SysDBNr )r&rEr r r!create_segmentsezSegmentManager.create_segments collection_idcCr))zDelete any local state for all the segments associated with a collection, and returns a sequence of their IDs. Does not update the SysDB.Nr )r&rHr r r!delete_segmentskrGzSegmentManager.delete_segmentstypecCr))a`Return the segment that should be used for servicing queries to a collection. Implementations should cache appropriately; clients are intended to call this method repeatedly rather than storing the result (thereby giving this implementation full control over which segment impls are in or out of memory at a given time.)Nr )r&rHrJr r r! get_segmenttszSegmentManager.get_segment hint_typeNcCr))zSignal to the segment manager that a collection is about to be used, so that it can preload segments as needed. This is only a hint, and implementations are free to ignore it.Nr )r&rHrLr r r!hint_use_collection}r0z"SegmentManager.hint_use_collection)rrrr=rrrrrFrrIrr4rKr rMr r r r!rDasrDN) typingrrrrabcrchromadb.typesrrr r r r r rrrrchromadb.configrruuidrenumrrr"r4r6r?rDr r r r!s 4