ugddlmZmZmZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZddlmZddlmZGddeZGd d eZed e ZGd deZGddeZGddeZy))OptionalSequenceTypeVar)abstractmethod) CollectionMetadataEmbeddingRecord OperationRequestVersionContextVectorEmbeddingRecordWhere WhereDocument VectorQueryVectorQueryResultSegmentSeqIdMetadata) ComponentSystem)UUID)Enumc$eZdZdZdZdZdZdZdZy) 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_METADATAN/var/www/openai/venv/lib/python3.12/site-packages/chromadb/segment/__init__.pyrrs% 1FEKC<@r#rceZdZededefdZededefdZ ede fdZ e de dee fd Zed d Zy ) SegmentImplementationsytstemsegmentcyNr")selfr'r(s r$__init__zSegmentImplementation.__init__ s r#request_version_contextreturncy)z,Get the number of embeddings in this segmentNr")r+r-s r$countzSegmentImplementation.count$ r#cy)z7Get the maximum SeqID currently indexed by this segmentNr"r+s r$ max_seqidzSegmentImplementation.max_seqid)r1r#metadatacy)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 r#Ncy)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)boundczeZdZdZe d dedeedeedee e dee dee d e d e e fd Zy) MetadataReaderz$Embedding Metadata segment interfaceNr-wherewhere_documentidslimitoffsetinclude_metadatar.cy)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 jeZdZdZe d dedeeedee fdZ ede deee fdZ y) VectorReaderz"Embedding Vector segment interfaceNr-rBr.cy)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#querycy)zYGiven a vector query, return the top-k nearest neighbors for vector in the query.Nr")r+rOs r$ query_vectorszVectorReader.query_vectors]s r#r*)rrrrHrr rrrIr rNrrrQr"r#r$rLrLPsv,(, !6 hsm $  ' (      (,- .  r#rLcneZdZdZededeefdZede dee fdZ ede de ddfd Z y) SegmentManagerzfInterface for a pluggable strategy for creating, retrieving and instantiating segments as required collectionr.cy)zsReturn the segments required for a new collection. Returns only segment data, does not persist to the SysDBNr")r+rTs r$#prepare_segments_for_new_collectionz2SegmentManager.prepare_segments_for_new_collectionj r# collection_idcy)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_segmentsprWr# hint_typeNcy)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_collectionvs r#) rrrrHrrrrrVrrZr r]r"r#r$rSrSfs j XV]M^   T htn    ) PT  r#rSN) typingrrrabcrchromadb.typesrrr r r r r rrrrrchromadb.configrruuidrenumrrr&r<r?rLrSr"r#r$rds|..    .A$A I 8 C,- * $ ( , Y r#