Zi,ddlmZmZmZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZddlmZddlmZGddeZGd d eZed e ZGd deZGddeZGddeZdS))OptionalSequenceTypeVar)abstractmethod) CollectionMetadataEmbeddingRecord OperationRequestVersionContextVectorEmbeddingRecordWhere 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_METADATAV/root/projects/butler/venv/lib64/python3.11/site-packages/chromadb/segment/__init__.pyrrs3 1FEKC<@r#rceZdZededefdZededefdZ ede fdZ e de dee fd Zed d Zd S) SegmentImplementationsytstemsegmentcdSNr")selfr'r(s r$__init__zSegmentImplementation.__init__ s r#request_version_contextreturncdS)z,Get the number of embeddings in this segmentNr")r+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")r5s r$propagate_collection_metadataz3SegmentImplementation.propagate_collection_metadata.s tr#NcdS)z#Delete the segment and all its dataNr"r3s r$deletezSegmentImplementation.delete5s  r#)r.N)rrrrrrr,r intr0rr4 staticmethodrrr7r9r"r#r$r&r&s     ^  -B s   ^  5   ^ Xh=O\    ^   r#r&S)boundceZdZdZe d dedeedeedee e dee d ee d e d e e fd ZdS)MetadataReaderz$Embedding Metadata segment interfaceNTr-wherewhere_documentidslimitoffsetinclude_metadatar.cdS)zQuery for embedding metadata.Nr")r+r-r@rArBrCrDrEs r$ get_metadatazMetadataReader.get_metadataAs  r#)NNNNNT)rrr__doc__rr rr r rstrr:boolrrGr"r#r$r?r?>s.."&26'+# $!%   !6    !/  hsm $  }        ) *    ^    r#r?c eZdZdZe d dedeeedee fdZ ede deee fdZ dS) VectorReaderz"Embedding Vector segment interfaceNr-rBr.cdS)z]Get embeddings from the segment. If no IDs are provided, all embeddings are returned.Nr")r+r-rBs r$ get_vectorszVectorReader.get_vectorsSs  r#querycdS)zYGiven a vector query, return the top-k nearest neighbors for vector in the query.Nr")r+rOs r$ query_vectorszVectorReader.query_vectors] r#r*)rrrrHrr rrrIr rNrrrQr"r#r$rLrLPs,,(,  !6 hsm $  ' (    ^    (,- .   ^   r#rLceZdZdZededeefdZede dee fdZ ede de ddfd Z dS) 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+rUs r$#prepare_segments_for_new_collectionz2SegmentManager.prepare_segments_for_new_collectionjrRr# 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+rXs r$delete_segmentszSegmentManager.delete_segmentsrs  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+rXr[s r$hint_use_collectionz"SegmentManager.hint_use_collectionxs r#) rrrrHrrrrrWrrZr r]r"r#r$rTrTfs $ '    ^  T htn   ^   ) PT   ^   r#rTN) typingrrrabcrchromadb.typesrrr r r r r rrrrrchromadb.configrruuidrenumrrr&r<r?rLrTr"r#r$rds..........                            .-------AAAAA$AAA     I   8 GC,---     *   $     (   ,     Y     r#