Function create_collection_with_metadata

tests/test-grpc.py:20–60

Test creating collection with metadata schema

(stub)

def create_collection_with_metadata(stub):
    """Test creating collection with metadata schema"""
    request = vector_service.CreateCollectionRequest(
        name="test_metadata_collection",
        description="Test collection with metadata schema",
        dense_vector=vector_service.DenseVectorOptions(dimension=128, enabled=True),
        sparse_vector=vector_service.SparseVectorOptions(enabled=False),
        tf_idf_options=vector_service.TFIDFOptions(enabled=False),
        metadata_schema=vector_service.MetadataSchema(
            fields=[
                vector_service.MetadataField(
                    name="category",
                    values=[
                        vector_service.FieldValue(string_value="electronics"),
                        vector_service.FieldValue(string_value="books"),
                        vector_service.FieldValue(string_value="clothing"),
                    ],
                ),
                vector_service.MetadataField(
                    name="rating",
                    values=[
                        vector_service.FieldValue(int_value=1),
                        vector_service.FieldValue(int_value=2),
                        vector_service.FieldValue(int_value=3),
                        vector_service.FieldValue(int_value=4),
                        vector_service.FieldValue(int_value=5),
                    ],
                ),
            ],
            supported_conditions=[
                vector_service.SupportedCondition(
                    op=vector_service.SupportedCondition.OperationType.AND,
                    field_names=["category", "rating"],
                )
            ],
        ),
        config=vector_service.CollectionConfig(
            max_vectors=1000000, replication_factor=1
        ),
    )
    return stub.CreateCollection(request)


def create_simple_collection(stub, name="test_collection", dimension=128):