o 3Ih@sddlmZmZmZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZddlmZddlmZGdddeZGd d d eZed ed ZGd ddeZGdddeZGdddeZdS))OptionalSequenceTypeVar)abstractmethod) CollectionMetadataEmbeddingRecord OperationRequestVersionContextVectorEmbeddingRecordWhere 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 Y/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/chromadb/segment/__init__.pyrsrc@sreZdZededefddZededefddZ ede fd d Z e d e dee fd d ZedddZdS)SegmentImplementationsytstemsegmentcCsdSNr )selfr#r$r r r!__init__ szSegmentImplementation.__init__request_version_contextreturncCdS)z,Get the number of embeddings in this segmentNr )r&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!delete5r,zSegmentImplementation.delete)r)N)rrrrrrr'r intr+rr. staticmethodrrr0r2r r r r!r"sr"S)boundc@sheZdZdZe      ddedeedeedee e dee d ee d e d e e fd d ZdS)MetadataReaderz$Embedding Metadata segment interfaceNTr(wherewhere_documentidslimitoffsetinclude_metadatar)cCr*)zQuery for embedding metadata.Nr )r&r(r8r9r:r;r<r=r r r! get_metadataAs zMetadataReader.get_metadata)NNNNNT)rrr__doc__rr rr r rstrr3boolrr>r r r r!r7>s4  r7c @sXeZdZdZe d dedeeedee fddZ ede deee fd d Z dS) VectorReaderz"Embedding Vector segment interfaceNr(r:r)cCr*)z]Get embeddings from the segment. If no IDs are provided, all embeddings are returned.Nr )r&r(r:r r r! get_vectorsSszVectorReader.get_vectorsquerycCr*)zYGiven a vector query, return the top-k nearest neighbors for vector in the query.Nr )r&rDr r r! query_vectors]zVectorReader.query_vectorsr%)rrrr?rr rrr@r rCr rrEr r r r!rBPs"    rBc@s^eZdZdZededeefddZede dee fddZ ede d e dd fd d Z d 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&rHr r r!#prepare_segments_for_new_collectionjrFz2SegmentManager.prepare_segments_for_new_collection 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&rJr r r!delete_segmentsrszSegmentManager.delete_segments 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&rJrLr r r!hint_use_collectionxr1z"SegmentManager.hint_use_collection) rrrr?rrrrrIrrKrrMr r r r!rGfs rGN) typingrrrabcrchromadb.typesrrrr r r r r rrrrchromadb.configrruuidrenumrrr"r5r7rBrGr r r r!s 8