Function convert_to_cpp_doc

python/zvec/model/convert.py:20–46

(doc: Doc, collection_schema: CollectionSchema)

def convert_to_cpp_doc(doc: Doc, collection_schema: CollectionSchema) -> _Doc:
    if not doc or not collection_schema:
        return None

    _doc = _Doc()

    # set pk
    _doc.set_pk(doc.id)

    # set scalar fields
    for k, v in doc.fields.items():
        field_schema = collection_schema.field(k)
        if not field_schema:
            raise ValueError(
                f"schema validate failed: {k} not found in collection schema"
            )
        _doc.set_any(k, field_schema._get_object(), v)

    # set vector fields
    for k, v in doc.vectors.items():
        vector_schema = collection_schema.vector(k)
        if not vector_schema:
            raise ValueError(
                f"schema validate failed: {k} not found in collection schema"
            )
        _doc.set_any(k, vector_schema._get_object(), v)
    return _doc


def convert_to_py_doc(doc: _Doc, collection_schema: CollectionSchema) -> Doc: