ui7,ddlmZmZmZmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZddlmZddlmZGddeZGd d eZed e ZGd deZGddeZGddeZdS))OptionalSequenceTypeVarType)abstractmethod) CollectionMetadataEmbeddingRecord OperationVectorEmbeddingRecordWhere WhereDocument VectorQueryVectorQueryResultSegmentSeqIdMetadata) ComponentSystem)UUID)Enumc&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_METADATAL/var/www/icac/venv/lib/python3.11/site-packages/chromadb/segment/__init__.pyrrs3 1FEKC<@r#rceZdZededefdZedefdZede fdZ e de de e fdZed d Zd S) SegmentImplementationsytstemsegmentcdSNr")selfr'r(s r$__init__zSegmentImplementation.__init__s r#returncdS)z,Get the number of embeddings in this segmentNr"r+s r$countzSegmentImplementation.count#  r#cdS)z7Get the maximum SeqID currently indexed by this segmentNr"r/s r$ max_seqidzSegmentImplementation.max_seqid(r1r#metadatacdS)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")r4s r$propagate_collection_metadataz3SegmentImplementation.propagate_collection_metadata-s tr#NcdS)z#Delete the segment and all its dataNr"r/s r$deletezSegmentImplementation.delete4s  r#)r-N)rrrrrrr,intr0rr3 staticmethodrrr6r8r"r#r$r&r&s     ^  s   ^  5   ^ Xh=O\    ^   r#r&S)boundceZdZdZe d deedeedeee dee dee dee f d Z dS) MetadataReaderz$Embedding Metadata segment interfaceNwherewhere_documentidslimitoffsetr-cdS)zQuery for embedding metadata.Nr")r+r?r@rArBrCs r$ get_metadatazMetadataReader.get_metadata@s  r#)NNNNN) rrr__doc__rrr r rstrr9r rEr"r#r$r>r>=s.."&26'+# $     !/  hsm $  }     ) *    ^    r#r>ceZdZdZe ddeeedeefdZ ede deee fdZ dS) VectorReaderz"Embedding Vector segment interfaceNrAr-cdS)z]Get embeddings from the segment. If no IDs are provided, all embeddings are returned.Nr")r+rAs r$ get_vectorszVectorReader.get_vectorsP r#querycdS)zYGiven a vector query, return the top-k nearest neighbors for vector in the query.Nr")r+rMs r$ query_vectorszVectorReader.query_vectorsXrLr#r*) rrrrFrrrrGr rKrrrOr"r#r$rIrIMs,,-1  HSM* ' (   ^    (,- .   ^   r#rIceZdZdZededeefdZede dee fdZ ede de e de fdZ ede d edd fd Zd S) SegmentManagerzfInterface for a pluggable strategy for creating, retrieving and instantiating segments as required collectionr-cdS)zsReturn the segments required for a new collection. Returns only segment data, does not persist to the SysDBNr")r+rRs r$create_segmentszSegmentManager.create_segmentse  r# collection_idcdS)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+rVs r$delete_segmentszSegmentManager.delete_segmentskrUr#typecdS)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+rVrYs r$ get_segmentzSegmentManager.get_segmentts  r# hint_typeNcdS)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+rVr\s r$hint_use_collectionz"SegmentManager.hint_use_collection}s r#)rrrrFrrrrrTrrXrr;r[r r^r"r#r$rQrQas * '9J   ^  T htn   ^   T!W    ^   ) PT   ^   r#rQN) typingrrrrabcrchromadb.typesrr r r r r rrrrrchromadb.configrruuidrenumrrr&r;r>rIrQr"r#r$res444444444444                          .-------AAAAA$AAA     I   8 GC,---      *         (   (! ! ! ! ! Y! ! ! ! ! r#