From 0b2f5b7712441febe3e924d0edbeca0fa0c03639 Mon Sep 17 00:00:00 2001 From: Trayan Azarov Date: Wed, 1 May 2024 18:13:58 +0300 Subject: [PATCH] fix: Rebased --- chromadb/proto/chroma_pb2.py | 59 ++++++++-------- chromadb/proto/chroma_pb2.pyi | 106 +++++++++++++++++------------ chromadb/proto/coordinator_pb2.pyi | 68 +++++++++--------- chromadb/proto/logservice_pb2.py | 5 +- chromadb/proto/logservice_pb2.pyi | 20 +++--- 5 files changed, 143 insertions(+), 115 deletions(-) diff --git a/chromadb/proto/chroma_pb2.py b/chromadb/proto/chroma_pb2.py index 9fe46b13890..79972bef5f9 100644 --- a/chromadb/proto/chroma_pb2.py +++ b/chromadb/proto/chroma_pb2.py @@ -1,6 +1,7 @@ # -*- coding: utf-8 -*- # Generated by the protocol buffer compiler. DO NOT EDIT! # source: chromadb/proto/chroma.proto +# Protobuf Python Version: 4.25.1 """Generated protocol buffer code.""" from google.protobuf import descriptor as _descriptor from google.protobuf import descriptor_pool as _descriptor_pool @@ -13,34 +14,34 @@ -DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1b\x63hromadb/proto/chroma.proto\x12\x06\x63hroma\"&\n\x06Status\x12\x0e\n\x06reason\x18\x01 \x01(\t\x12\x0c\n\x04\x63ode\x18\x02 \x01(\x05\"U\n\x06Vector\x12\x11\n\tdimension\x18\x01 \x01(\x05\x12\x0e\n\x06vector\x18\x02 \x01(\x0c\x12(\n\x08\x65ncoding\x18\x03 \x01(\x0e\x32\x16.chroma.ScalarEncoding\"\x1a\n\tFilePaths\x12\r\n\x05paths\x18\x01 \x03(\t\"\xa5\x02\n\x07Segment\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04type\x18\x02 \x01(\t\x12#\n\x05scope\x18\x03 \x01(\x0e\x32\x14.chroma.SegmentScope\x12\x17\n\ncollection\x18\x05 \x01(\tH\x00\x88\x01\x01\x12-\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.chroma.UpdateMetadataH\x01\x88\x01\x01\x12\x32\n\nfile_paths\x18\x07 \x03(\x0b\x32\x1e.chroma.Segment.FilePathsEntry\x1a\x43\n\x0e\x46ilePathsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12 \n\x05value\x18\x02 \x01(\x0b\x32\x11.chroma.FilePaths:\x02\x38\x01\x42\r\n\x0b_collectionB\x0b\n\t_metadata\"\xd1\x01\n\nCollection\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12-\n\x08metadata\x18\x04 \x01(\x0b\x32\x16.chroma.UpdateMetadataH\x00\x88\x01\x01\x12\x16\n\tdimension\x18\x05 \x01(\x05H\x01\x88\x01\x01\x12\x0e\n\x06tenant\x18\x06 \x01(\t\x12\x10\n\x08\x64\x61tabase\x18\x07 \x01(\t\x12\x14\n\x0clog_position\x18\x08 \x01(\x03\x12\x0f\n\x07version\x18\t \x01(\x05\x42\x0b\n\t_metadataB\x0c\n\n_dimension\"4\n\x08\x44\x61tabase\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x0e\n\x06tenant\x18\x03 \x01(\t\"\x16\n\x06Tenant\x12\x0c\n\x04name\x18\x01 \x01(\t\"b\n\x13UpdateMetadataValue\x12\x16\n\x0cstring_value\x18\x01 \x01(\tH\x00\x12\x13\n\tint_value\x18\x02 \x01(\x03H\x00\x12\x15\n\x0b\x66loat_value\x18\x03 \x01(\x01H\x00\x42\x07\n\x05value\"\x96\x01\n\x0eUpdateMetadata\x12\x36\n\x08metadata\x18\x01 \x03(\x0b\x32$.chroma.UpdateMetadata.MetadataEntry\x1aL\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12*\n\x05value\x18\x02 \x01(\x0b\x32\x1b.chroma.UpdateMetadataValue:\x02\x38\x01\"\xaf\x01\n\x0fOperationRecord\x12\n\n\x02id\x18\x01 \x01(\t\x12#\n\x06vector\x18\x02 \x01(\x0b\x32\x0e.chroma.VectorH\x00\x88\x01\x01\x12-\n\x08metadata\x18\x03 \x01(\x0b\x32\x16.chroma.UpdateMetadataH\x01\x88\x01\x01\x12$\n\toperation\x18\x04 \x01(\x0e\x32\x11.chroma.OperationB\t\n\x07_vectorB\x0b\n\t_metadata\"\xc2\x01\n\x14QueryMetadataRequest\x12\x12\n\nsegment_id\x18\x01 \x01(\t\x12\x1c\n\x05where\x18\x02 \x01(\x0b\x32\r.chroma.Where\x12-\n\x0ewhere_document\x18\x03 \x01(\x0b\x32\x15.chroma.WhereDocument\x12\x0b\n\x03ids\x18\x04 \x03(\t\x12\x12\n\x05limit\x18\x05 \x01(\x05H\x00\x88\x01\x01\x12\x13\n\x06offset\x18\x06 \x01(\x05H\x01\x88\x01\x01\x42\x08\n\x06_limitB\t\n\x07_offset\"I\n\x15QueryMetadataResponse\x12\x30\n\x07records\x18\x01 \x03(\x0b\x32\x1f.chroma.MetadataEmbeddingRecord\"O\n\x17MetadataEmbeddingRecord\x12\n\n\x02id\x18\x01 \x01(\t\x12(\n\x08metadata\x18\x02 \x01(\x0b\x32\x16.chroma.UpdateMetadata\"\x83\x01\n\rWhereDocument\x12-\n\x06\x64irect\x18\x01 \x01(\x0b\x32\x1b.chroma.DirectWhereDocumentH\x00\x12\x31\n\x08\x63hildren\x18\x02 \x01(\x0b\x32\x1d.chroma.WhereDocumentChildrenH\x00\x42\x10\n\x0ewhere_document\"X\n\x13\x44irectWhereDocument\x12\x10\n\x08\x64ocument\x18\x01 \x01(\t\x12/\n\x08operator\x18\x02 \x01(\x0e\x32\x1d.chroma.WhereDocumentOperator\"k\n\x15WhereDocumentChildren\x12\'\n\x08\x63hildren\x18\x01 \x03(\x0b\x32\x15.chroma.WhereDocument\x12)\n\x08operator\x18\x02 \x01(\x0e\x32\x17.chroma.BooleanOperator\"r\n\x05Where\x12\x35\n\x11\x64irect_comparison\x18\x01 \x01(\x0b\x32\x18.chroma.DirectComparisonH\x00\x12)\n\x08\x63hildren\x18\x02 \x01(\x0b\x32\x15.chroma.WhereChildrenH\x00\x42\x07\n\x05where\"\x9b\x03\n\x10\x44irectComparison\x12\x0b\n\x03key\x18\x01 \x01(\t\x12?\n\x15single_string_operand\x18\x02 \x01(\x0b\x32\x1e.chroma.SingleStringComparisonH\x00\x12;\n\x13string_list_operand\x18\x03 \x01(\x0b\x32\x1c.chroma.StringListComparisonH\x00\x12\x39\n\x12single_int_operand\x18\x04 \x01(\x0b\x32\x1b.chroma.SingleIntComparisonH\x00\x12\x35\n\x10int_list_operand\x18\x05 \x01(\x0b\x32\x19.chroma.IntListComparisonH\x00\x12?\n\x15single_double_operand\x18\x06 \x01(\x0b\x32\x1e.chroma.SingleDoubleComparisonH\x00\x12;\n\x13\x64ouble_list_operand\x18\x07 \x01(\x0b\x32\x1c.chroma.DoubleListComparisonH\x00\x42\x0c\n\ncomparison\"[\n\rWhereChildren\x12\x1f\n\x08\x63hildren\x18\x01 \x03(\x0b\x32\r.chroma.Where\x12)\n\x08operator\x18\x02 \x01(\x0e\x32\x17.chroma.BooleanOperator\"S\n\x14StringListComparison\x12\x0e\n\x06values\x18\x01 \x03(\t\x12+\n\rlist_operator\x18\x02 \x01(\x0e\x32\x14.chroma.ListOperator\"V\n\x16SingleStringComparison\x12\r\n\x05value\x18\x01 \x01(\t\x12-\n\ncomparator\x18\x02 \x01(\x0e\x32\x19.chroma.GenericComparator\"P\n\x11IntListComparison\x12\x0e\n\x06values\x18\x01 \x03(\x03\x12+\n\rlist_operator\x18\x02 \x01(\x0e\x32\x14.chroma.ListOperator\"\xa2\x01\n\x13SingleIntComparison\x12\r\n\x05value\x18\x01 \x01(\x03\x12\x37\n\x12generic_comparator\x18\x02 \x01(\x0e\x32\x19.chroma.GenericComparatorH\x00\x12\x35\n\x11number_comparator\x18\x03 \x01(\x0e\x32\x18.chroma.NumberComparatorH\x00\x42\x0c\n\ncomparator\"S\n\x14\x44oubleListComparison\x12\x0e\n\x06values\x18\x01 \x03(\x01\x12+\n\rlist_operator\x18\x02 \x01(\x0e\x32\x14.chroma.ListOperator\"\xa5\x01\n\x16SingleDoubleComparison\x12\r\n\x05value\x18\x01 \x01(\x01\x12\x37\n\x12generic_comparator\x18\x02 \x01(\x0e\x32\x19.chroma.GenericComparatorH\x00\x12\x35\n\x11number_comparator\x18\x03 \x01(\x0e\x32\x18.chroma.NumberComparatorH\x00\x42\x0c\n\ncomparator\"4\n\x11GetVectorsRequest\x12\x0b\n\x03ids\x18\x01 \x03(\t\x12\x12\n\nsegment_id\x18\x02 \x01(\t\"D\n\x12GetVectorsResponse\x12.\n\x07records\x18\x01 \x03(\x0b\x32\x1d.chroma.VectorEmbeddingRecord\"C\n\x15VectorEmbeddingRecord\x12\n\n\x02id\x18\x01 \x01(\t\x12\x1e\n\x06vector\x18\x03 \x01(\x0b\x32\x0e.chroma.Vector\"\x86\x01\n\x13QueryVectorsRequest\x12\x1f\n\x07vectors\x18\x01 \x03(\x0b\x32\x0e.chroma.Vector\x12\t\n\x01k\x18\x02 \x01(\x05\x12\x13\n\x0b\x61llowed_ids\x18\x03 \x03(\t\x12\x1a\n\x12include_embeddings\x18\x04 \x01(\x08\x12\x12\n\nsegment_id\x18\x05 \x01(\t\"C\n\x14QueryVectorsResponse\x12+\n\x07results\x18\x01 \x03(\x0b\x32\x1a.chroma.VectorQueryResults\"@\n\x12VectorQueryResults\x12*\n\x07results\x18\x01 \x03(\x0b\x32\x19.chroma.VectorQueryResult\"a\n\x11VectorQueryResult\x12\n\n\x02id\x18\x01 \x01(\t\x12\x10\n\x08\x64istance\x18\x03 \x01(\x02\x12#\n\x06vector\x18\x04 \x01(\x0b\x32\x0e.chroma.VectorH\x00\x88\x01\x01\x42\t\n\x07_vector*8\n\tOperation\x12\x07\n\x03\x41\x44\x44\x10\x00\x12\n\n\x06UPDATE\x10\x01\x12\n\n\x06UPSERT\x10\x02\x12\n\n\x06\x44\x45LETE\x10\x03*(\n\x0eScalarEncoding\x12\x0b\n\x07\x46LOAT32\x10\x00\x12\t\n\x05INT32\x10\x01*@\n\x0cSegmentScope\x12\n\n\x06VECTOR\x10\x00\x12\x0c\n\x08METADATA\x10\x01\x12\n\n\x06RECORD\x10\x02\x12\n\n\x06SQLITE\x10\x03*7\n\x15WhereDocumentOperator\x12\x0c\n\x08\x43ONTAINS\x10\x00\x12\x10\n\x0cNOT_CONTAINS\x10\x01*\"\n\x0f\x42ooleanOperator\x12\x07\n\x03\x41ND\x10\x00\x12\x06\n\x02OR\x10\x01*\x1f\n\x0cListOperator\x12\x06\n\x02IN\x10\x00\x12\x07\n\x03NIN\x10\x01*#\n\x11GenericComparator\x12\x06\n\x02\x45Q\x10\x00\x12\x06\n\x02NE\x10\x01*4\n\x10NumberComparator\x12\x06\n\x02GT\x10\x00\x12\x07\n\x03GTE\x10\x01\x12\x06\n\x02LT\x10\x02\x12\x07\n\x03LTE\x10\x03\x32`\n\x0eMetadataReader\x12N\n\rQueryMetadata\x12\x1c.chroma.QueryMetadataRequest\x1a\x1d.chroma.QueryMetadataResponse\"\x00\x32\xa2\x01\n\x0cVectorReader\x12\x45\n\nGetVectors\x12\x19.chroma.GetVectorsRequest\x1a\x1a.chroma.GetVectorsResponse\"\x00\x12K\n\x0cQueryVectors\x12\x1b.chroma.QueryVectorsRequest\x1a\x1c.chroma.QueryVectorsResponse\"\x00\x42:Z8github.com/chroma-core/chroma/go/pkg/proto/coordinatorpbb\x06proto3') +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1b\x63hromadb/proto/chroma.proto\x12\x06\x63hroma\"&\n\x06Status\x12\x0e\n\x06reason\x18\x01 \x01(\t\x12\x0c\n\x04\x63ode\x18\x02 \x01(\x05\"U\n\x06Vector\x12\x11\n\tdimension\x18\x01 \x01(\x05\x12\x0e\n\x06vector\x18\x02 \x01(\x0c\x12(\n\x08\x65ncoding\x18\x03 \x01(\x0e\x32\x16.chroma.ScalarEncoding\"\x1a\n\tFilePaths\x12\r\n\x05paths\x18\x01 \x03(\t\"\xa5\x02\n\x07Segment\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04type\x18\x02 \x01(\t\x12#\n\x05scope\x18\x03 \x01(\x0e\x32\x14.chroma.SegmentScope\x12\x17\n\ncollection\x18\x05 \x01(\tH\x00\x88\x01\x01\x12-\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.chroma.UpdateMetadataH\x01\x88\x01\x01\x12\x32\n\nfile_paths\x18\x07 \x03(\x0b\x32\x1e.chroma.Segment.FilePathsEntry\x1a\x43\n\x0e\x46ilePathsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12 \n\x05value\x18\x02 \x01(\x0b\x32\x11.chroma.FilePaths:\x02\x38\x01\x42\r\n\x0b_collectionB\x0b\n\t_metadata\"\xd1\x01\n\nCollection\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12-\n\x08metadata\x18\x04 \x01(\x0b\x32\x16.chroma.UpdateMetadataH\x00\x88\x01\x01\x12\x16\n\tdimension\x18\x05 \x01(\x05H\x01\x88\x01\x01\x12\x0e\n\x06tenant\x18\x06 \x01(\t\x12\x10\n\x08\x64\x61tabase\x18\x07 \x01(\t\x12\x14\n\x0clog_position\x18\x08 \x01(\x03\x12\x0f\n\x07version\x18\t \x01(\x05\x42\x0b\n\t_metadataB\x0c\n\n_dimension\"4\n\x08\x44\x61tabase\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x0e\n\x06tenant\x18\x03 \x01(\t\"\x16\n\x06Tenant\x12\x0c\n\x04name\x18\x01 \x01(\t\"b\n\x13UpdateMetadataValue\x12\x16\n\x0cstring_value\x18\x01 \x01(\tH\x00\x12\x13\n\tint_value\x18\x02 \x01(\x03H\x00\x12\x15\n\x0b\x66loat_value\x18\x03 \x01(\x01H\x00\x42\x07\n\x05value\"\x96\x01\n\x0eUpdateMetadata\x12\x36\n\x08metadata\x18\x01 \x03(\x0b\x32$.chroma.UpdateMetadata.MetadataEntry\x1aL\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12*\n\x05value\x18\x02 \x01(\x0b\x32\x1b.chroma.UpdateMetadataValue:\x02\x38\x01\"\xaf\x01\n\x0fOperationRecord\x12\n\n\x02id\x18\x01 \x01(\t\x12#\n\x06vector\x18\x02 \x01(\x0b\x32\x0e.chroma.VectorH\x00\x88\x01\x01\x12-\n\x08metadata\x18\x03 \x01(\x0b\x32\x16.chroma.UpdateMetadataH\x01\x88\x01\x01\x12$\n\toperation\x18\x04 \x01(\x0e\x32\x11.chroma.OperationB\t\n\x07_vectorB\x0b\n\t_metadata\"\xc2\x01\n\x14QueryMetadataRequest\x12\x12\n\nsegment_id\x18\x01 \x01(\t\x12\x1c\n\x05where\x18\x02 \x01(\x0b\x32\r.chroma.Where\x12-\n\x0ewhere_document\x18\x03 \x01(\x0b\x32\x15.chroma.WhereDocument\x12\x0b\n\x03ids\x18\x04 \x03(\t\x12\x12\n\x05limit\x18\x05 \x01(\x05H\x00\x88\x01\x01\x12\x13\n\x06offset\x18\x06 \x01(\x05H\x01\x88\x01\x01\x42\x08\n\x06_limitB\t\n\x07_offset\"I\n\x15QueryMetadataResponse\x12\x30\n\x07records\x18\x01 \x03(\x0b\x32\x1f.chroma.MetadataEmbeddingRecord\"O\n\x17MetadataEmbeddingRecord\x12\n\n\x02id\x18\x01 \x01(\t\x12(\n\x08metadata\x18\x02 \x01(\x0b\x32\x16.chroma.UpdateMetadata\"\x83\x01\n\rWhereDocument\x12-\n\x06\x64irect\x18\x01 \x01(\x0b\x32\x1b.chroma.DirectWhereDocumentH\x00\x12\x31\n\x08\x63hildren\x18\x02 \x01(\x0b\x32\x1d.chroma.WhereDocumentChildrenH\x00\x42\x10\n\x0ewhere_document\"X\n\x13\x44irectWhereDocument\x12\x10\n\x08\x64ocument\x18\x01 \x01(\t\x12/\n\x08operator\x18\x02 \x01(\x0e\x32\x1d.chroma.WhereDocumentOperator\"k\n\x15WhereDocumentChildren\x12\'\n\x08\x63hildren\x18\x01 \x03(\x0b\x32\x15.chroma.WhereDocument\x12)\n\x08operator\x18\x02 \x01(\x0e\x32\x17.chroma.BooleanOperator\"r\n\x05Where\x12\x35\n\x11\x64irect_comparison\x18\x01 \x01(\x0b\x32\x18.chroma.DirectComparisonH\x00\x12)\n\x08\x63hildren\x18\x02 \x01(\x0b\x32\x15.chroma.WhereChildrenH\x00\x42\x07\n\x05where\"\x9b\x03\n\x10\x44irectComparison\x12\x0b\n\x03key\x18\x01 \x01(\t\x12?\n\x15single_string_operand\x18\x02 \x01(\x0b\x32\x1e.chroma.SingleStringComparisonH\x00\x12;\n\x13string_list_operand\x18\x03 \x01(\x0b\x32\x1c.chroma.StringListComparisonH\x00\x12\x39\n\x12single_int_operand\x18\x04 \x01(\x0b\x32\x1b.chroma.SingleIntComparisonH\x00\x12\x35\n\x10int_list_operand\x18\x05 \x01(\x0b\x32\x19.chroma.IntListComparisonH\x00\x12?\n\x15single_double_operand\x18\x06 \x01(\x0b\x32\x1e.chroma.SingleDoubleComparisonH\x00\x12;\n\x13\x64ouble_list_operand\x18\x07 \x01(\x0b\x32\x1c.chroma.DoubleListComparisonH\x00\x42\x0c\n\ncomparison\"[\n\rWhereChildren\x12\x1f\n\x08\x63hildren\x18\x01 \x03(\x0b\x32\r.chroma.Where\x12)\n\x08operator\x18\x02 \x01(\x0e\x32\x17.chroma.BooleanOperator\"S\n\x14StringListComparison\x12\x0e\n\x06values\x18\x01 \x03(\t\x12+\n\rlist_operator\x18\x02 \x01(\x0e\x32\x14.chroma.ListOperator\"V\n\x16SingleStringComparison\x12\r\n\x05value\x18\x01 \x01(\t\x12-\n\ncomparator\x18\x02 \x01(\x0e\x32\x19.chroma.GenericComparator\"P\n\x11IntListComparison\x12\x0e\n\x06values\x18\x01 \x03(\x03\x12+\n\rlist_operator\x18\x02 \x01(\x0e\x32\x14.chroma.ListOperator\"\xa2\x01\n\x13SingleIntComparison\x12\r\n\x05value\x18\x01 \x01(\x03\x12\x37\n\x12generic_comparator\x18\x02 \x01(\x0e\x32\x19.chroma.GenericComparatorH\x00\x12\x35\n\x11number_comparator\x18\x03 \x01(\x0e\x32\x18.chroma.NumberComparatorH\x00\x42\x0c\n\ncomparator\"S\n\x14\x44oubleListComparison\x12\x0e\n\x06values\x18\x01 \x03(\x01\x12+\n\rlist_operator\x18\x02 \x01(\x0e\x32\x14.chroma.ListOperator\"\xa5\x01\n\x16SingleDoubleComparison\x12\r\n\x05value\x18\x01 \x01(\x01\x12\x37\n\x12generic_comparator\x18\x02 \x01(\x0e\x32\x19.chroma.GenericComparatorH\x00\x12\x35\n\x11number_comparator\x18\x03 \x01(\x0e\x32\x18.chroma.NumberComparatorH\x00\x42\x0c\n\ncomparator\"4\n\x11GetVectorsRequest\x12\x0b\n\x03ids\x18\x01 \x03(\t\x12\x12\n\nsegment_id\x18\x02 \x01(\t\"D\n\x12GetVectorsResponse\x12.\n\x07records\x18\x01 \x03(\x0b\x32\x1d.chroma.VectorEmbeddingRecord\"C\n\x15VectorEmbeddingRecord\x12\n\n\x02id\x18\x01 \x01(\t\x12\x1e\n\x06vector\x18\x03 \x01(\x0b\x32\x0e.chroma.Vector\"\x86\x01\n\x13QueryVectorsRequest\x12\x1f\n\x07vectors\x18\x01 \x03(\x0b\x32\x0e.chroma.Vector\x12\t\n\x01k\x18\x02 \x01(\x05\x12\x13\n\x0b\x61llowed_ids\x18\x03 \x03(\t\x12\x1a\n\x12include_embeddings\x18\x04 \x01(\x08\x12\x12\n\nsegment_id\x18\x05 \x01(\t\"C\n\x14QueryVectorsResponse\x12+\n\x07results\x18\x01 \x03(\x0b\x32\x1a.chroma.VectorQueryResults\"@\n\x12VectorQueryResults\x12*\n\x07results\x18\x01 \x03(\x0b\x32\x19.chroma.VectorQueryResult\"a\n\x11VectorQueryResult\x12\n\n\x02id\x18\x01 \x01(\t\x12\x10\n\x08\x64istance\x18\x03 \x01(\x02\x12#\n\x06vector\x18\x04 \x01(\x0b\x32\x0e.chroma.VectorH\x00\x88\x01\x01\x42\t\n\x07_vector\"U\n\x19LocalSegmentMetadataTuple\x12\x14\n\x0c\x65mbedding_id\x18\x01 \x01(\t\x12\x12\n\nhnsw_label\x18\x02 \x01(\x05\x12\x0e\n\x06seq_id\x18\x03 \x01(\x05\"\x93\x01\n\x14LocalSegmentMetadata\x12\x31\n\x06tuples\x18\x01 \x03(\x0b\x32!.chroma.LocalSegmentMetadataTuple\x12\x16\n\x0e\x64imensionality\x18\x02 \x01(\x05\x12\x1c\n\x14total_elements_added\x18\x03 \x01(\x03\x12\x12\n\nmax_seq_id\x18\x04 \x01(\x03*8\n\tOperation\x12\x07\n\x03\x41\x44\x44\x10\x00\x12\n\n\x06UPDATE\x10\x01\x12\n\n\x06UPSERT\x10\x02\x12\n\n\x06\x44\x45LETE\x10\x03*(\n\x0eScalarEncoding\x12\x0b\n\x07\x46LOAT32\x10\x00\x12\t\n\x05INT32\x10\x01*@\n\x0cSegmentScope\x12\n\n\x06VECTOR\x10\x00\x12\x0c\n\x08METADATA\x10\x01\x12\n\n\x06RECORD\x10\x02\x12\n\n\x06SQLITE\x10\x03*7\n\x15WhereDocumentOperator\x12\x0c\n\x08\x43ONTAINS\x10\x00\x12\x10\n\x0cNOT_CONTAINS\x10\x01*\"\n\x0f\x42ooleanOperator\x12\x07\n\x03\x41ND\x10\x00\x12\x06\n\x02OR\x10\x01*\x1f\n\x0cListOperator\x12\x06\n\x02IN\x10\x00\x12\x07\n\x03NIN\x10\x01*#\n\x11GenericComparator\x12\x06\n\x02\x45Q\x10\x00\x12\x06\n\x02NE\x10\x01*4\n\x10NumberComparator\x12\x06\n\x02GT\x10\x00\x12\x07\n\x03GTE\x10\x01\x12\x06\n\x02LT\x10\x02\x12\x07\n\x03LTE\x10\x03\x32`\n\x0eMetadataReader\x12N\n\rQueryMetadata\x12\x1c.chroma.QueryMetadataRequest\x1a\x1d.chroma.QueryMetadataResponse\"\x00\x32\xa2\x01\n\x0cVectorReader\x12\x45\n\nGetVectors\x12\x19.chroma.GetVectorsRequest\x1a\x1a.chroma.GetVectorsResponse\"\x00\x12K\n\x0cQueryVectors\x12\x1b.chroma.QueryVectorsRequest\x1a\x1c.chroma.QueryVectorsResponse\"\x00\x42:Z8github.com/chroma-core/chroma/go/pkg/proto/coordinatorpbb\x06proto3') _globals = globals() _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals) _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'chromadb.proto.chroma_pb2', _globals) if _descriptor._USE_C_DESCRIPTORS == False: - DESCRIPTOR._options = None - DESCRIPTOR._serialized_options = b'Z8github.com/chroma-core/chroma/go/pkg/proto/coordinatorpb' - _SEGMENT_FILEPATHSENTRY._options = None - _SEGMENT_FILEPATHSENTRY._serialized_options = b'8\001' - _UPDATEMETADATA_METADATAENTRY._options = None - _UPDATEMETADATA_METADATAENTRY._serialized_options = b'8\001' - _globals['_OPERATION']._serialized_start=3757 - _globals['_OPERATION']._serialized_end=3813 - _globals['_SCALARENCODING']._serialized_start=3815 - _globals['_SCALARENCODING']._serialized_end=3855 - _globals['_SEGMENTSCOPE']._serialized_start=3857 - _globals['_SEGMENTSCOPE']._serialized_end=3921 - _globals['_WHEREDOCUMENTOPERATOR']._serialized_start=3923 - _globals['_WHEREDOCUMENTOPERATOR']._serialized_end=3978 - _globals['_BOOLEANOPERATOR']._serialized_start=3980 - _globals['_BOOLEANOPERATOR']._serialized_end=4014 - _globals['_LISTOPERATOR']._serialized_start=4016 - _globals['_LISTOPERATOR']._serialized_end=4047 - _globals['_GENERICCOMPARATOR']._serialized_start=4049 - _globals['_GENERICCOMPARATOR']._serialized_end=4084 - _globals['_NUMBERCOMPARATOR']._serialized_start=4086 - _globals['_NUMBERCOMPARATOR']._serialized_end=4138 + _globals['DESCRIPTOR']._options = None + _globals['DESCRIPTOR']._serialized_options = b'Z8github.com/chroma-core/chroma/go/pkg/proto/coordinatorpb' + _globals['_SEGMENT_FILEPATHSENTRY']._options = None + _globals['_SEGMENT_FILEPATHSENTRY']._serialized_options = b'8\001' + _globals['_UPDATEMETADATA_METADATAENTRY']._options = None + _globals['_UPDATEMETADATA_METADATAENTRY']._serialized_options = b'8\001' + _globals['_OPERATION']._serialized_start=3994 + _globals['_OPERATION']._serialized_end=4050 + _globals['_SCALARENCODING']._serialized_start=4052 + _globals['_SCALARENCODING']._serialized_end=4092 + _globals['_SEGMENTSCOPE']._serialized_start=4094 + _globals['_SEGMENTSCOPE']._serialized_end=4158 + _globals['_WHEREDOCUMENTOPERATOR']._serialized_start=4160 + _globals['_WHEREDOCUMENTOPERATOR']._serialized_end=4215 + _globals['_BOOLEANOPERATOR']._serialized_start=4217 + _globals['_BOOLEANOPERATOR']._serialized_end=4251 + _globals['_LISTOPERATOR']._serialized_start=4253 + _globals['_LISTOPERATOR']._serialized_end=4284 + _globals['_GENERICCOMPARATOR']._serialized_start=4286 + _globals['_GENERICCOMPARATOR']._serialized_end=4321 + _globals['_NUMBERCOMPARATOR']._serialized_start=4323 + _globals['_NUMBERCOMPARATOR']._serialized_end=4375 _globals['_STATUS']._serialized_start=39 _globals['_STATUS']._serialized_end=77 _globals['_VECTOR']._serialized_start=79 @@ -109,8 +110,12 @@ _globals['_VECTORQUERYRESULTS']._serialized_end=3656 _globals['_VECTORQUERYRESULT']._serialized_start=3658 _globals['_VECTORQUERYRESULT']._serialized_end=3755 - _globals['_METADATAREADER']._serialized_start=4140 - _globals['_METADATAREADER']._serialized_end=4236 - _globals['_VECTORREADER']._serialized_start=4239 - _globals['_VECTORREADER']._serialized_end=4401 + _globals['_LOCALSEGMENTMETADATATUPLE']._serialized_start=3757 + _globals['_LOCALSEGMENTMETADATATUPLE']._serialized_end=3842 + _globals['_LOCALSEGMENTMETADATA']._serialized_start=3845 + _globals['_LOCALSEGMENTMETADATA']._serialized_end=3992 + _globals['_METADATAREADER']._serialized_start=4377 + _globals['_METADATAREADER']._serialized_end=4473 + _globals['_VECTORREADER']._serialized_start=4476 + _globals['_VECTORREADER']._serialized_end=4638 # @@protoc_insertion_point(module_scope) diff --git a/chromadb/proto/chroma_pb2.pyi b/chromadb/proto/chroma_pb2.pyi index b2222094ebc..d0cd5bbe479 100644 --- a/chromadb/proto/chroma_pb2.pyi +++ b/chromadb/proto/chroma_pb2.pyi @@ -7,46 +7,46 @@ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Map DESCRIPTOR: _descriptor.FileDescriptor class Operation(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): - __slots__ = [] + __slots__ = () ADD: _ClassVar[Operation] UPDATE: _ClassVar[Operation] UPSERT: _ClassVar[Operation] DELETE: _ClassVar[Operation] class ScalarEncoding(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): - __slots__ = [] + __slots__ = () FLOAT32: _ClassVar[ScalarEncoding] INT32: _ClassVar[ScalarEncoding] class SegmentScope(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): - __slots__ = [] + __slots__ = () VECTOR: _ClassVar[SegmentScope] METADATA: _ClassVar[SegmentScope] RECORD: _ClassVar[SegmentScope] SQLITE: _ClassVar[SegmentScope] class WhereDocumentOperator(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): - __slots__ = [] + __slots__ = () CONTAINS: _ClassVar[WhereDocumentOperator] NOT_CONTAINS: _ClassVar[WhereDocumentOperator] class BooleanOperator(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): - __slots__ = [] + __slots__ = () AND: _ClassVar[BooleanOperator] OR: _ClassVar[BooleanOperator] class ListOperator(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): - __slots__ = [] + __slots__ = () IN: _ClassVar[ListOperator] NIN: _ClassVar[ListOperator] class GenericComparator(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): - __slots__ = [] + __slots__ = () EQ: _ClassVar[GenericComparator] NE: _ClassVar[GenericComparator] class NumberComparator(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): - __slots__ = [] + __slots__ = () GT: _ClassVar[NumberComparator] GTE: _ClassVar[NumberComparator] LT: _ClassVar[NumberComparator] @@ -75,7 +75,7 @@ LT: NumberComparator LTE: NumberComparator class Status(_message.Message): - __slots__ = ["reason", "code"] + __slots__ = ("reason", "code") REASON_FIELD_NUMBER: _ClassVar[int] CODE_FIELD_NUMBER: _ClassVar[int] reason: str @@ -83,7 +83,7 @@ class Status(_message.Message): def __init__(self, reason: _Optional[str] = ..., code: _Optional[int] = ...) -> None: ... class Vector(_message.Message): - __slots__ = ["dimension", "vector", "encoding"] + __slots__ = ("dimension", "vector", "encoding") DIMENSION_FIELD_NUMBER: _ClassVar[int] VECTOR_FIELD_NUMBER: _ClassVar[int] ENCODING_FIELD_NUMBER: _ClassVar[int] @@ -93,15 +93,15 @@ class Vector(_message.Message): def __init__(self, dimension: _Optional[int] = ..., vector: _Optional[bytes] = ..., encoding: _Optional[_Union[ScalarEncoding, str]] = ...) -> None: ... class FilePaths(_message.Message): - __slots__ = ["paths"] + __slots__ = ("paths",) PATHS_FIELD_NUMBER: _ClassVar[int] paths: _containers.RepeatedScalarFieldContainer[str] def __init__(self, paths: _Optional[_Iterable[str]] = ...) -> None: ... class Segment(_message.Message): - __slots__ = ["id", "type", "scope", "collection", "metadata", "file_paths"] + __slots__ = ("id", "type", "scope", "collection", "metadata", "file_paths") class FilePathsEntry(_message.Message): - __slots__ = ["key", "value"] + __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str @@ -122,7 +122,7 @@ class Segment(_message.Message): def __init__(self, id: _Optional[str] = ..., type: _Optional[str] = ..., scope: _Optional[_Union[SegmentScope, str]] = ..., collection: _Optional[str] = ..., metadata: _Optional[_Union[UpdateMetadata, _Mapping]] = ..., file_paths: _Optional[_Mapping[str, FilePaths]] = ...) -> None: ... class Collection(_message.Message): - __slots__ = ["id", "name", "metadata", "dimension", "tenant", "database", "log_position", "version"] + __slots__ = ("id", "name", "metadata", "dimension", "tenant", "database", "log_position", "version") ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] METADATA_FIELD_NUMBER: _ClassVar[int] @@ -142,7 +142,7 @@ class Collection(_message.Message): def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., metadata: _Optional[_Union[UpdateMetadata, _Mapping]] = ..., dimension: _Optional[int] = ..., tenant: _Optional[str] = ..., database: _Optional[str] = ..., log_position: _Optional[int] = ..., version: _Optional[int] = ...) -> None: ... class Database(_message.Message): - __slots__ = ["id", "name", "tenant"] + __slots__ = ("id", "name", "tenant") ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] TENANT_FIELD_NUMBER: _ClassVar[int] @@ -152,13 +152,13 @@ class Database(_message.Message): def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., tenant: _Optional[str] = ...) -> None: ... class Tenant(_message.Message): - __slots__ = ["name"] + __slots__ = ("name",) NAME_FIELD_NUMBER: _ClassVar[int] name: str def __init__(self, name: _Optional[str] = ...) -> None: ... class UpdateMetadataValue(_message.Message): - __slots__ = ["string_value", "int_value", "float_value"] + __slots__ = ("string_value", "int_value", "float_value") STRING_VALUE_FIELD_NUMBER: _ClassVar[int] INT_VALUE_FIELD_NUMBER: _ClassVar[int] FLOAT_VALUE_FIELD_NUMBER: _ClassVar[int] @@ -168,9 +168,9 @@ class UpdateMetadataValue(_message.Message): def __init__(self, string_value: _Optional[str] = ..., int_value: _Optional[int] = ..., float_value: _Optional[float] = ...) -> None: ... class UpdateMetadata(_message.Message): - __slots__ = ["metadata"] + __slots__ = ("metadata",) class MetadataEntry(_message.Message): - __slots__ = ["key", "value"] + __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str @@ -181,7 +181,7 @@ class UpdateMetadata(_message.Message): def __init__(self, metadata: _Optional[_Mapping[str, UpdateMetadataValue]] = ...) -> None: ... class OperationRecord(_message.Message): - __slots__ = ["id", "vector", "metadata", "operation"] + __slots__ = ("id", "vector", "metadata", "operation") ID_FIELD_NUMBER: _ClassVar[int] VECTOR_FIELD_NUMBER: _ClassVar[int] METADATA_FIELD_NUMBER: _ClassVar[int] @@ -193,7 +193,7 @@ class OperationRecord(_message.Message): def __init__(self, id: _Optional[str] = ..., vector: _Optional[_Union[Vector, _Mapping]] = ..., metadata: _Optional[_Union[UpdateMetadata, _Mapping]] = ..., operation: _Optional[_Union[Operation, str]] = ...) -> None: ... class QueryMetadataRequest(_message.Message): - __slots__ = ["segment_id", "where", "where_document", "ids", "limit", "offset"] + __slots__ = ("segment_id", "where", "where_document", "ids", "limit", "offset") SEGMENT_ID_FIELD_NUMBER: _ClassVar[int] WHERE_FIELD_NUMBER: _ClassVar[int] WHERE_DOCUMENT_FIELD_NUMBER: _ClassVar[int] @@ -209,13 +209,13 @@ class QueryMetadataRequest(_message.Message): def __init__(self, segment_id: _Optional[str] = ..., where: _Optional[_Union[Where, _Mapping]] = ..., where_document: _Optional[_Union[WhereDocument, _Mapping]] = ..., ids: _Optional[_Iterable[str]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ...) -> None: ... class QueryMetadataResponse(_message.Message): - __slots__ = ["records"] + __slots__ = ("records",) RECORDS_FIELD_NUMBER: _ClassVar[int] records: _containers.RepeatedCompositeFieldContainer[MetadataEmbeddingRecord] def __init__(self, records: _Optional[_Iterable[_Union[MetadataEmbeddingRecord, _Mapping]]] = ...) -> None: ... class MetadataEmbeddingRecord(_message.Message): - __slots__ = ["id", "metadata"] + __slots__ = ("id", "metadata") ID_FIELD_NUMBER: _ClassVar[int] METADATA_FIELD_NUMBER: _ClassVar[int] id: str @@ -223,7 +223,7 @@ class MetadataEmbeddingRecord(_message.Message): def __init__(self, id: _Optional[str] = ..., metadata: _Optional[_Union[UpdateMetadata, _Mapping]] = ...) -> None: ... class WhereDocument(_message.Message): - __slots__ = ["direct", "children"] + __slots__ = ("direct", "children") DIRECT_FIELD_NUMBER: _ClassVar[int] CHILDREN_FIELD_NUMBER: _ClassVar[int] direct: DirectWhereDocument @@ -231,7 +231,7 @@ class WhereDocument(_message.Message): def __init__(self, direct: _Optional[_Union[DirectWhereDocument, _Mapping]] = ..., children: _Optional[_Union[WhereDocumentChildren, _Mapping]] = ...) -> None: ... class DirectWhereDocument(_message.Message): - __slots__ = ["document", "operator"] + __slots__ = ("document", "operator") DOCUMENT_FIELD_NUMBER: _ClassVar[int] OPERATOR_FIELD_NUMBER: _ClassVar[int] document: str @@ -239,7 +239,7 @@ class DirectWhereDocument(_message.Message): def __init__(self, document: _Optional[str] = ..., operator: _Optional[_Union[WhereDocumentOperator, str]] = ...) -> None: ... class WhereDocumentChildren(_message.Message): - __slots__ = ["children", "operator"] + __slots__ = ("children", "operator") CHILDREN_FIELD_NUMBER: _ClassVar[int] OPERATOR_FIELD_NUMBER: _ClassVar[int] children: _containers.RepeatedCompositeFieldContainer[WhereDocument] @@ -247,7 +247,7 @@ class WhereDocumentChildren(_message.Message): def __init__(self, children: _Optional[_Iterable[_Union[WhereDocument, _Mapping]]] = ..., operator: _Optional[_Union[BooleanOperator, str]] = ...) -> None: ... class Where(_message.Message): - __slots__ = ["direct_comparison", "children"] + __slots__ = ("direct_comparison", "children") DIRECT_COMPARISON_FIELD_NUMBER: _ClassVar[int] CHILDREN_FIELD_NUMBER: _ClassVar[int] direct_comparison: DirectComparison @@ -255,7 +255,7 @@ class Where(_message.Message): def __init__(self, direct_comparison: _Optional[_Union[DirectComparison, _Mapping]] = ..., children: _Optional[_Union[WhereChildren, _Mapping]] = ...) -> None: ... class DirectComparison(_message.Message): - __slots__ = ["key", "single_string_operand", "string_list_operand", "single_int_operand", "int_list_operand", "single_double_operand", "double_list_operand"] + __slots__ = ("key", "single_string_operand", "string_list_operand", "single_int_operand", "int_list_operand", "single_double_operand", "double_list_operand") KEY_FIELD_NUMBER: _ClassVar[int] SINGLE_STRING_OPERAND_FIELD_NUMBER: _ClassVar[int] STRING_LIST_OPERAND_FIELD_NUMBER: _ClassVar[int] @@ -273,7 +273,7 @@ class DirectComparison(_message.Message): def __init__(self, key: _Optional[str] = ..., single_string_operand: _Optional[_Union[SingleStringComparison, _Mapping]] = ..., string_list_operand: _Optional[_Union[StringListComparison, _Mapping]] = ..., single_int_operand: _Optional[_Union[SingleIntComparison, _Mapping]] = ..., int_list_operand: _Optional[_Union[IntListComparison, _Mapping]] = ..., single_double_operand: _Optional[_Union[SingleDoubleComparison, _Mapping]] = ..., double_list_operand: _Optional[_Union[DoubleListComparison, _Mapping]] = ...) -> None: ... class WhereChildren(_message.Message): - __slots__ = ["children", "operator"] + __slots__ = ("children", "operator") CHILDREN_FIELD_NUMBER: _ClassVar[int] OPERATOR_FIELD_NUMBER: _ClassVar[int] children: _containers.RepeatedCompositeFieldContainer[Where] @@ -281,7 +281,7 @@ class WhereChildren(_message.Message): def __init__(self, children: _Optional[_Iterable[_Union[Where, _Mapping]]] = ..., operator: _Optional[_Union[BooleanOperator, str]] = ...) -> None: ... class StringListComparison(_message.Message): - __slots__ = ["values", "list_operator"] + __slots__ = ("values", "list_operator") VALUES_FIELD_NUMBER: _ClassVar[int] LIST_OPERATOR_FIELD_NUMBER: _ClassVar[int] values: _containers.RepeatedScalarFieldContainer[str] @@ -289,7 +289,7 @@ class StringListComparison(_message.Message): def __init__(self, values: _Optional[_Iterable[str]] = ..., list_operator: _Optional[_Union[ListOperator, str]] = ...) -> None: ... class SingleStringComparison(_message.Message): - __slots__ = ["value", "comparator"] + __slots__ = ("value", "comparator") VALUE_FIELD_NUMBER: _ClassVar[int] COMPARATOR_FIELD_NUMBER: _ClassVar[int] value: str @@ -297,7 +297,7 @@ class SingleStringComparison(_message.Message): def __init__(self, value: _Optional[str] = ..., comparator: _Optional[_Union[GenericComparator, str]] = ...) -> None: ... class IntListComparison(_message.Message): - __slots__ = ["values", "list_operator"] + __slots__ = ("values", "list_operator") VALUES_FIELD_NUMBER: _ClassVar[int] LIST_OPERATOR_FIELD_NUMBER: _ClassVar[int] values: _containers.RepeatedScalarFieldContainer[int] @@ -305,7 +305,7 @@ class IntListComparison(_message.Message): def __init__(self, values: _Optional[_Iterable[int]] = ..., list_operator: _Optional[_Union[ListOperator, str]] = ...) -> None: ... class SingleIntComparison(_message.Message): - __slots__ = ["value", "generic_comparator", "number_comparator"] + __slots__ = ("value", "generic_comparator", "number_comparator") VALUE_FIELD_NUMBER: _ClassVar[int] GENERIC_COMPARATOR_FIELD_NUMBER: _ClassVar[int] NUMBER_COMPARATOR_FIELD_NUMBER: _ClassVar[int] @@ -315,7 +315,7 @@ class SingleIntComparison(_message.Message): def __init__(self, value: _Optional[int] = ..., generic_comparator: _Optional[_Union[GenericComparator, str]] = ..., number_comparator: _Optional[_Union[NumberComparator, str]] = ...) -> None: ... class DoubleListComparison(_message.Message): - __slots__ = ["values", "list_operator"] + __slots__ = ("values", "list_operator") VALUES_FIELD_NUMBER: _ClassVar[int] LIST_OPERATOR_FIELD_NUMBER: _ClassVar[int] values: _containers.RepeatedScalarFieldContainer[float] @@ -323,7 +323,7 @@ class DoubleListComparison(_message.Message): def __init__(self, values: _Optional[_Iterable[float]] = ..., list_operator: _Optional[_Union[ListOperator, str]] = ...) -> None: ... class SingleDoubleComparison(_message.Message): - __slots__ = ["value", "generic_comparator", "number_comparator"] + __slots__ = ("value", "generic_comparator", "number_comparator") VALUE_FIELD_NUMBER: _ClassVar[int] GENERIC_COMPARATOR_FIELD_NUMBER: _ClassVar[int] NUMBER_COMPARATOR_FIELD_NUMBER: _ClassVar[int] @@ -333,7 +333,7 @@ class SingleDoubleComparison(_message.Message): def __init__(self, value: _Optional[float] = ..., generic_comparator: _Optional[_Union[GenericComparator, str]] = ..., number_comparator: _Optional[_Union[NumberComparator, str]] = ...) -> None: ... class GetVectorsRequest(_message.Message): - __slots__ = ["ids", "segment_id"] + __slots__ = ("ids", "segment_id") IDS_FIELD_NUMBER: _ClassVar[int] SEGMENT_ID_FIELD_NUMBER: _ClassVar[int] ids: _containers.RepeatedScalarFieldContainer[str] @@ -341,13 +341,13 @@ class GetVectorsRequest(_message.Message): def __init__(self, ids: _Optional[_Iterable[str]] = ..., segment_id: _Optional[str] = ...) -> None: ... class GetVectorsResponse(_message.Message): - __slots__ = ["records"] + __slots__ = ("records",) RECORDS_FIELD_NUMBER: _ClassVar[int] records: _containers.RepeatedCompositeFieldContainer[VectorEmbeddingRecord] def __init__(self, records: _Optional[_Iterable[_Union[VectorEmbeddingRecord, _Mapping]]] = ...) -> None: ... class VectorEmbeddingRecord(_message.Message): - __slots__ = ["id", "vector"] + __slots__ = ("id", "vector") ID_FIELD_NUMBER: _ClassVar[int] VECTOR_FIELD_NUMBER: _ClassVar[int] id: str @@ -355,7 +355,7 @@ class VectorEmbeddingRecord(_message.Message): def __init__(self, id: _Optional[str] = ..., vector: _Optional[_Union[Vector, _Mapping]] = ...) -> None: ... class QueryVectorsRequest(_message.Message): - __slots__ = ["vectors", "k", "allowed_ids", "include_embeddings", "segment_id"] + __slots__ = ("vectors", "k", "allowed_ids", "include_embeddings", "segment_id") VECTORS_FIELD_NUMBER: _ClassVar[int] K_FIELD_NUMBER: _ClassVar[int] ALLOWED_IDS_FIELD_NUMBER: _ClassVar[int] @@ -369,19 +369,19 @@ class QueryVectorsRequest(_message.Message): def __init__(self, vectors: _Optional[_Iterable[_Union[Vector, _Mapping]]] = ..., k: _Optional[int] = ..., allowed_ids: _Optional[_Iterable[str]] = ..., include_embeddings: bool = ..., segment_id: _Optional[str] = ...) -> None: ... class QueryVectorsResponse(_message.Message): - __slots__ = ["results"] + __slots__ = ("results",) RESULTS_FIELD_NUMBER: _ClassVar[int] results: _containers.RepeatedCompositeFieldContainer[VectorQueryResults] def __init__(self, results: _Optional[_Iterable[_Union[VectorQueryResults, _Mapping]]] = ...) -> None: ... class VectorQueryResults(_message.Message): - __slots__ = ["results"] + __slots__ = ("results",) RESULTS_FIELD_NUMBER: _ClassVar[int] results: _containers.RepeatedCompositeFieldContainer[VectorQueryResult] def __init__(self, results: _Optional[_Iterable[_Union[VectorQueryResult, _Mapping]]] = ...) -> None: ... class VectorQueryResult(_message.Message): - __slots__ = ["id", "distance", "vector"] + __slots__ = ("id", "distance", "vector") ID_FIELD_NUMBER: _ClassVar[int] DISTANCE_FIELD_NUMBER: _ClassVar[int] VECTOR_FIELD_NUMBER: _ClassVar[int] @@ -389,3 +389,25 @@ class VectorQueryResult(_message.Message): distance: float vector: Vector def __init__(self, id: _Optional[str] = ..., distance: _Optional[float] = ..., vector: _Optional[_Union[Vector, _Mapping]] = ...) -> None: ... + +class LocalSegmentMetadataTuple(_message.Message): + __slots__ = ("embedding_id", "hnsw_label", "seq_id") + EMBEDDING_ID_FIELD_NUMBER: _ClassVar[int] + HNSW_LABEL_FIELD_NUMBER: _ClassVar[int] + SEQ_ID_FIELD_NUMBER: _ClassVar[int] + embedding_id: str + hnsw_label: int + seq_id: int + def __init__(self, embedding_id: _Optional[str] = ..., hnsw_label: _Optional[int] = ..., seq_id: _Optional[int] = ...) -> None: ... + +class LocalSegmentMetadata(_message.Message): + __slots__ = ("tuples", "dimensionality", "total_elements_added", "max_seq_id") + TUPLES_FIELD_NUMBER: _ClassVar[int] + DIMENSIONALITY_FIELD_NUMBER: _ClassVar[int] + TOTAL_ELEMENTS_ADDED_FIELD_NUMBER: _ClassVar[int] + MAX_SEQ_ID_FIELD_NUMBER: _ClassVar[int] + tuples: _containers.RepeatedCompositeFieldContainer[LocalSegmentMetadataTuple] + dimensionality: int + total_elements_added: int + max_seq_id: int + def __init__(self, tuples: _Optional[_Iterable[_Union[LocalSegmentMetadataTuple, _Mapping]]] = ..., dimensionality: _Optional[int] = ..., total_elements_added: _Optional[int] = ..., max_seq_id: _Optional[int] = ...) -> None: ... diff --git a/chromadb/proto/coordinator_pb2.pyi b/chromadb/proto/coordinator_pb2.pyi index bd71112a65c..93c3ddd8b98 100644 --- a/chromadb/proto/coordinator_pb2.pyi +++ b/chromadb/proto/coordinator_pb2.pyi @@ -8,7 +8,7 @@ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Map DESCRIPTOR: _descriptor.FileDescriptor class CreateDatabaseRequest(_message.Message): - __slots__ = ["id", "name", "tenant"] + __slots__ = ("id", "name", "tenant") ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] TENANT_FIELD_NUMBER: _ClassVar[int] @@ -18,13 +18,13 @@ class CreateDatabaseRequest(_message.Message): def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., tenant: _Optional[str] = ...) -> None: ... class CreateDatabaseResponse(_message.Message): - __slots__ = ["status"] + __slots__ = ("status",) STATUS_FIELD_NUMBER: _ClassVar[int] status: _chroma_pb2.Status def __init__(self, status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class GetDatabaseRequest(_message.Message): - __slots__ = ["name", "tenant"] + __slots__ = ("name", "tenant") NAME_FIELD_NUMBER: _ClassVar[int] TENANT_FIELD_NUMBER: _ClassVar[int] name: str @@ -32,7 +32,7 @@ class GetDatabaseRequest(_message.Message): def __init__(self, name: _Optional[str] = ..., tenant: _Optional[str] = ...) -> None: ... class GetDatabaseResponse(_message.Message): - __slots__ = ["database", "status"] + __slots__ = ("database", "status") DATABASE_FIELD_NUMBER: _ClassVar[int] STATUS_FIELD_NUMBER: _ClassVar[int] database: _chroma_pb2.Database @@ -40,25 +40,25 @@ class GetDatabaseResponse(_message.Message): def __init__(self, database: _Optional[_Union[_chroma_pb2.Database, _Mapping]] = ..., status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class CreateTenantRequest(_message.Message): - __slots__ = ["name"] + __slots__ = ("name",) NAME_FIELD_NUMBER: _ClassVar[int] name: str def __init__(self, name: _Optional[str] = ...) -> None: ... class CreateTenantResponse(_message.Message): - __slots__ = ["status"] + __slots__ = ("status",) STATUS_FIELD_NUMBER: _ClassVar[int] status: _chroma_pb2.Status def __init__(self, status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class GetTenantRequest(_message.Message): - __slots__ = ["name"] + __slots__ = ("name",) NAME_FIELD_NUMBER: _ClassVar[int] name: str def __init__(self, name: _Optional[str] = ...) -> None: ... class GetTenantResponse(_message.Message): - __slots__ = ["tenant", "status"] + __slots__ = ("tenant", "status") TENANT_FIELD_NUMBER: _ClassVar[int] STATUS_FIELD_NUMBER: _ClassVar[int] tenant: _chroma_pb2.Tenant @@ -66,31 +66,31 @@ class GetTenantResponse(_message.Message): def __init__(self, tenant: _Optional[_Union[_chroma_pb2.Tenant, _Mapping]] = ..., status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class CreateSegmentRequest(_message.Message): - __slots__ = ["segment"] + __slots__ = ("segment",) SEGMENT_FIELD_NUMBER: _ClassVar[int] segment: _chroma_pb2.Segment def __init__(self, segment: _Optional[_Union[_chroma_pb2.Segment, _Mapping]] = ...) -> None: ... class CreateSegmentResponse(_message.Message): - __slots__ = ["status"] + __slots__ = ("status",) STATUS_FIELD_NUMBER: _ClassVar[int] status: _chroma_pb2.Status def __init__(self, status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class DeleteSegmentRequest(_message.Message): - __slots__ = ["id"] + __slots__ = ("id",) ID_FIELD_NUMBER: _ClassVar[int] id: str def __init__(self, id: _Optional[str] = ...) -> None: ... class DeleteSegmentResponse(_message.Message): - __slots__ = ["status"] + __slots__ = ("status",) STATUS_FIELD_NUMBER: _ClassVar[int] status: _chroma_pb2.Status def __init__(self, status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class GetSegmentsRequest(_message.Message): - __slots__ = ["id", "type", "scope", "collection"] + __slots__ = ("id", "type", "scope", "collection") ID_FIELD_NUMBER: _ClassVar[int] TYPE_FIELD_NUMBER: _ClassVar[int] SCOPE_FIELD_NUMBER: _ClassVar[int] @@ -102,7 +102,7 @@ class GetSegmentsRequest(_message.Message): def __init__(self, id: _Optional[str] = ..., type: _Optional[str] = ..., scope: _Optional[_Union[_chroma_pb2.SegmentScope, str]] = ..., collection: _Optional[str] = ...) -> None: ... class GetSegmentsResponse(_message.Message): - __slots__ = ["segments", "status"] + __slots__ = ("segments", "status") SEGMENTS_FIELD_NUMBER: _ClassVar[int] STATUS_FIELD_NUMBER: _ClassVar[int] segments: _containers.RepeatedCompositeFieldContainer[_chroma_pb2.Segment] @@ -110,7 +110,7 @@ class GetSegmentsResponse(_message.Message): def __init__(self, segments: _Optional[_Iterable[_Union[_chroma_pb2.Segment, _Mapping]]] = ..., status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class UpdateSegmentRequest(_message.Message): - __slots__ = ["id", "collection", "reset_collection", "metadata", "reset_metadata"] + __slots__ = ("id", "collection", "reset_collection", "metadata", "reset_metadata") ID_FIELD_NUMBER: _ClassVar[int] COLLECTION_FIELD_NUMBER: _ClassVar[int] RESET_COLLECTION_FIELD_NUMBER: _ClassVar[int] @@ -124,13 +124,13 @@ class UpdateSegmentRequest(_message.Message): def __init__(self, id: _Optional[str] = ..., collection: _Optional[str] = ..., reset_collection: bool = ..., metadata: _Optional[_Union[_chroma_pb2.UpdateMetadata, _Mapping]] = ..., reset_metadata: bool = ...) -> None: ... class UpdateSegmentResponse(_message.Message): - __slots__ = ["status"] + __slots__ = ("status",) STATUS_FIELD_NUMBER: _ClassVar[int] status: _chroma_pb2.Status def __init__(self, status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class CreateCollectionRequest(_message.Message): - __slots__ = ["id", "name", "metadata", "dimension", "get_or_create", "tenant", "database"] + __slots__ = ("id", "name", "metadata", "dimension", "get_or_create", "tenant", "database") ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] METADATA_FIELD_NUMBER: _ClassVar[int] @@ -148,7 +148,7 @@ class CreateCollectionRequest(_message.Message): def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., metadata: _Optional[_Union[_chroma_pb2.UpdateMetadata, _Mapping]] = ..., dimension: _Optional[int] = ..., get_or_create: bool = ..., tenant: _Optional[str] = ..., database: _Optional[str] = ...) -> None: ... class CreateCollectionResponse(_message.Message): - __slots__ = ["collection", "created", "status"] + __slots__ = ("collection", "created", "status") COLLECTION_FIELD_NUMBER: _ClassVar[int] CREATED_FIELD_NUMBER: _ClassVar[int] STATUS_FIELD_NUMBER: _ClassVar[int] @@ -158,7 +158,7 @@ class CreateCollectionResponse(_message.Message): def __init__(self, collection: _Optional[_Union[_chroma_pb2.Collection, _Mapping]] = ..., created: bool = ..., status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class DeleteCollectionRequest(_message.Message): - __slots__ = ["id", "tenant", "database"] + __slots__ = ("id", "tenant", "database") ID_FIELD_NUMBER: _ClassVar[int] TENANT_FIELD_NUMBER: _ClassVar[int] DATABASE_FIELD_NUMBER: _ClassVar[int] @@ -168,13 +168,13 @@ class DeleteCollectionRequest(_message.Message): def __init__(self, id: _Optional[str] = ..., tenant: _Optional[str] = ..., database: _Optional[str] = ...) -> None: ... class DeleteCollectionResponse(_message.Message): - __slots__ = ["status"] + __slots__ = ("status",) STATUS_FIELD_NUMBER: _ClassVar[int] status: _chroma_pb2.Status def __init__(self, status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class GetCollectionsRequest(_message.Message): - __slots__ = ["id", "name", "tenant", "database", "limit", "offset"] + __slots__ = ("id", "name", "tenant", "database", "limit", "offset") ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] TENANT_FIELD_NUMBER: _ClassVar[int] @@ -190,7 +190,7 @@ class GetCollectionsRequest(_message.Message): def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., tenant: _Optional[str] = ..., database: _Optional[str] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ...) -> None: ... class GetCollectionsResponse(_message.Message): - __slots__ = ["collections", "status"] + __slots__ = ("collections", "status") COLLECTIONS_FIELD_NUMBER: _ClassVar[int] STATUS_FIELD_NUMBER: _ClassVar[int] collections: _containers.RepeatedCompositeFieldContainer[_chroma_pb2.Collection] @@ -198,7 +198,7 @@ class GetCollectionsResponse(_message.Message): def __init__(self, collections: _Optional[_Iterable[_Union[_chroma_pb2.Collection, _Mapping]]] = ..., status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class UpdateCollectionRequest(_message.Message): - __slots__ = ["id", "name", "dimension", "metadata", "reset_metadata"] + __slots__ = ("id", "name", "dimension", "metadata", "reset_metadata") ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] DIMENSION_FIELD_NUMBER: _ClassVar[int] @@ -212,13 +212,13 @@ class UpdateCollectionRequest(_message.Message): def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., dimension: _Optional[int] = ..., metadata: _Optional[_Union[_chroma_pb2.UpdateMetadata, _Mapping]] = ..., reset_metadata: bool = ...) -> None: ... class UpdateCollectionResponse(_message.Message): - __slots__ = ["status"] + __slots__ = ("status",) STATUS_FIELD_NUMBER: _ClassVar[int] status: _chroma_pb2.Status def __init__(self, status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class Notification(_message.Message): - __slots__ = ["id", "collection_id", "type", "status"] + __slots__ = ("id", "collection_id", "type", "status") ID_FIELD_NUMBER: _ClassVar[int] COLLECTION_ID_FIELD_NUMBER: _ClassVar[int] TYPE_FIELD_NUMBER: _ClassVar[int] @@ -230,19 +230,19 @@ class Notification(_message.Message): def __init__(self, id: _Optional[int] = ..., collection_id: _Optional[str] = ..., type: _Optional[str] = ..., status: _Optional[str] = ...) -> None: ... class ResetStateResponse(_message.Message): - __slots__ = ["status"] + __slots__ = ("status",) STATUS_FIELD_NUMBER: _ClassVar[int] status: _chroma_pb2.Status def __init__(self, status: _Optional[_Union[_chroma_pb2.Status, _Mapping]] = ...) -> None: ... class GetLastCompactionTimeForTenantRequest(_message.Message): - __slots__ = ["tenant_id"] + __slots__ = ("tenant_id",) TENANT_ID_FIELD_NUMBER: _ClassVar[int] tenant_id: _containers.RepeatedScalarFieldContainer[str] def __init__(self, tenant_id: _Optional[_Iterable[str]] = ...) -> None: ... class TenantLastCompactionTime(_message.Message): - __slots__ = ["tenant_id", "last_compaction_time"] + __slots__ = ("tenant_id", "last_compaction_time") TENANT_ID_FIELD_NUMBER: _ClassVar[int] LAST_COMPACTION_TIME_FIELD_NUMBER: _ClassVar[int] tenant_id: str @@ -250,21 +250,21 @@ class TenantLastCompactionTime(_message.Message): def __init__(self, tenant_id: _Optional[str] = ..., last_compaction_time: _Optional[int] = ...) -> None: ... class GetLastCompactionTimeForTenantResponse(_message.Message): - __slots__ = ["tenant_last_compaction_time"] + __slots__ = ("tenant_last_compaction_time",) TENANT_LAST_COMPACTION_TIME_FIELD_NUMBER: _ClassVar[int] tenant_last_compaction_time: _containers.RepeatedCompositeFieldContainer[TenantLastCompactionTime] def __init__(self, tenant_last_compaction_time: _Optional[_Iterable[_Union[TenantLastCompactionTime, _Mapping]]] = ...) -> None: ... class SetLastCompactionTimeForTenantRequest(_message.Message): - __slots__ = ["tenant_last_compaction_time"] + __slots__ = ("tenant_last_compaction_time",) TENANT_LAST_COMPACTION_TIME_FIELD_NUMBER: _ClassVar[int] tenant_last_compaction_time: TenantLastCompactionTime def __init__(self, tenant_last_compaction_time: _Optional[_Union[TenantLastCompactionTime, _Mapping]] = ...) -> None: ... class FlushSegmentCompactionInfo(_message.Message): - __slots__ = ["segment_id", "file_paths"] + __slots__ = ("segment_id", "file_paths") class FilePathsEntry(_message.Message): - __slots__ = ["key", "value"] + __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str @@ -277,7 +277,7 @@ class FlushSegmentCompactionInfo(_message.Message): def __init__(self, segment_id: _Optional[str] = ..., file_paths: _Optional[_Mapping[str, _chroma_pb2.FilePaths]] = ...) -> None: ... class FlushCollectionCompactionRequest(_message.Message): - __slots__ = ["tenant_id", "collection_id", "log_position", "collection_version", "segment_compaction_info"] + __slots__ = ("tenant_id", "collection_id", "log_position", "collection_version", "segment_compaction_info") TENANT_ID_FIELD_NUMBER: _ClassVar[int] COLLECTION_ID_FIELD_NUMBER: _ClassVar[int] LOG_POSITION_FIELD_NUMBER: _ClassVar[int] @@ -291,7 +291,7 @@ class FlushCollectionCompactionRequest(_message.Message): def __init__(self, tenant_id: _Optional[str] = ..., collection_id: _Optional[str] = ..., log_position: _Optional[int] = ..., collection_version: _Optional[int] = ..., segment_compaction_info: _Optional[_Iterable[_Union[FlushSegmentCompactionInfo, _Mapping]]] = ...) -> None: ... class FlushCollectionCompactionResponse(_message.Message): - __slots__ = ["collection_id", "collection_version", "last_compaction_time"] + __slots__ = ("collection_id", "collection_version", "last_compaction_time") COLLECTION_ID_FIELD_NUMBER: _ClassVar[int] COLLECTION_VERSION_FIELD_NUMBER: _ClassVar[int] LAST_COMPACTION_TIME_FIELD_NUMBER: _ClassVar[int] diff --git a/chromadb/proto/logservice_pb2.py b/chromadb/proto/logservice_pb2.py index 7159c2e0cd0..5162962819d 100644 --- a/chromadb/proto/logservice_pb2.py +++ b/chromadb/proto/logservice_pb2.py @@ -1,6 +1,7 @@ # -*- coding: utf-8 -*- # Generated by the protocol buffer compiler. DO NOT EDIT! # source: chromadb/proto/logservice.proto +# Protobuf Python Version: 4.25.1 """Generated protocol buffer code.""" from google.protobuf import descriptor as _descriptor from google.protobuf import descriptor_pool as _descriptor_pool @@ -20,8 +21,8 @@ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals) _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'chromadb.proto.logservice_pb2', _globals) if _descriptor._USE_C_DESCRIPTORS == False: - DESCRIPTOR._options = None - DESCRIPTOR._serialized_options = b'Z7github.com/chroma-core/chroma/go/pkg/proto/logservicepb' + _globals['DESCRIPTOR']._options = None + _globals['DESCRIPTOR']._serialized_options = b'Z7github.com/chroma-core/chroma/go/pkg/proto/logservicepb' _globals['_PUSHLOGSREQUEST']._serialized_start=72 _globals['_PUSHLOGSREQUEST']._serialized_end=154 _globals['_PUSHLOGSRESPONSE']._serialized_start=156 diff --git a/chromadb/proto/logservice_pb2.pyi b/chromadb/proto/logservice_pb2.pyi index c9c78d4ca4b..ac0b8428e24 100644 --- a/chromadb/proto/logservice_pb2.pyi +++ b/chromadb/proto/logservice_pb2.pyi @@ -7,7 +7,7 @@ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Map DESCRIPTOR: _descriptor.FileDescriptor class PushLogsRequest(_message.Message): - __slots__ = ["collection_id", "records"] + __slots__ = ("collection_id", "records") COLLECTION_ID_FIELD_NUMBER: _ClassVar[int] RECORDS_FIELD_NUMBER: _ClassVar[int] collection_id: str @@ -15,13 +15,13 @@ class PushLogsRequest(_message.Message): def __init__(self, collection_id: _Optional[str] = ..., records: _Optional[_Iterable[_Union[_chroma_pb2.OperationRecord, _Mapping]]] = ...) -> None: ... class PushLogsResponse(_message.Message): - __slots__ = ["record_count"] + __slots__ = ("record_count",) RECORD_COUNT_FIELD_NUMBER: _ClassVar[int] record_count: int def __init__(self, record_count: _Optional[int] = ...) -> None: ... class PullLogsRequest(_message.Message): - __slots__ = ["collection_id", "start_from_offset", "batch_size", "end_timestamp"] + __slots__ = ("collection_id", "start_from_offset", "batch_size", "end_timestamp") COLLECTION_ID_FIELD_NUMBER: _ClassVar[int] START_FROM_OFFSET_FIELD_NUMBER: _ClassVar[int] BATCH_SIZE_FIELD_NUMBER: _ClassVar[int] @@ -33,7 +33,7 @@ class PullLogsRequest(_message.Message): def __init__(self, collection_id: _Optional[str] = ..., start_from_offset: _Optional[int] = ..., batch_size: _Optional[int] = ..., end_timestamp: _Optional[int] = ...) -> None: ... class LogRecord(_message.Message): - __slots__ = ["log_offset", "record"] + __slots__ = ("log_offset", "record") LOG_OFFSET_FIELD_NUMBER: _ClassVar[int] RECORD_FIELD_NUMBER: _ClassVar[int] log_offset: int @@ -41,13 +41,13 @@ class LogRecord(_message.Message): def __init__(self, log_offset: _Optional[int] = ..., record: _Optional[_Union[_chroma_pb2.OperationRecord, _Mapping]] = ...) -> None: ... class PullLogsResponse(_message.Message): - __slots__ = ["records"] + __slots__ = ("records",) RECORDS_FIELD_NUMBER: _ClassVar[int] records: _containers.RepeatedCompositeFieldContainer[LogRecord] def __init__(self, records: _Optional[_Iterable[_Union[LogRecord, _Mapping]]] = ...) -> None: ... class CollectionInfo(_message.Message): - __slots__ = ["collection_id", "first_log_offset", "first_log_ts"] + __slots__ = ("collection_id", "first_log_offset", "first_log_ts") COLLECTION_ID_FIELD_NUMBER: _ClassVar[int] FIRST_LOG_OFFSET_FIELD_NUMBER: _ClassVar[int] FIRST_LOG_TS_FIELD_NUMBER: _ClassVar[int] @@ -57,17 +57,17 @@ class CollectionInfo(_message.Message): def __init__(self, collection_id: _Optional[str] = ..., first_log_offset: _Optional[int] = ..., first_log_ts: _Optional[int] = ...) -> None: ... class GetAllCollectionInfoToCompactRequest(_message.Message): - __slots__ = [] + __slots__ = () def __init__(self) -> None: ... class GetAllCollectionInfoToCompactResponse(_message.Message): - __slots__ = ["all_collection_info"] + __slots__ = ("all_collection_info",) ALL_COLLECTION_INFO_FIELD_NUMBER: _ClassVar[int] all_collection_info: _containers.RepeatedCompositeFieldContainer[CollectionInfo] def __init__(self, all_collection_info: _Optional[_Iterable[_Union[CollectionInfo, _Mapping]]] = ...) -> None: ... class UpdateCollectionLogOffsetRequest(_message.Message): - __slots__ = ["collection_id", "log_offset"] + __slots__ = ("collection_id", "log_offset") COLLECTION_ID_FIELD_NUMBER: _ClassVar[int] LOG_OFFSET_FIELD_NUMBER: _ClassVar[int] collection_id: str @@ -75,5 +75,5 @@ class UpdateCollectionLogOffsetRequest(_message.Message): def __init__(self, collection_id: _Optional[str] = ..., log_offset: _Optional[int] = ...) -> None: ... class UpdateCollectionLogOffsetResponse(_message.Message): - __slots__ = [] + __slots__ = () def __init__(self) -> None: ...