Method set_query_vector

python/zvec/executor/query_executor.py:226–249

Resolve a Query vector and set it on a native query object.

(
        self, query: Query, search_query, collection: _Collection
    )

            search_query.fts = fts

    def set_query_vector(
        self, query: Query, search_query, collection: _Collection
    ) -> None:
        """Resolve a Query vector and set it on a native query object."""
        vector_schema = self._schema.vector(query.field_name)
        if vector_schema is None:
            raise ValueError(f"Vector field '{query.field_name}' not found in schema")

        if query.has_vector():
            vec_data = query.vector
        elif query.has_id():
            fetched = collection.Fetch([query.id])
            doc = next(iter(fetched.values()), None)
            if not doc:
                raise ValueError(f"Document with id '{query.id}' not found")
            vec_data = doc.get_any(vector_schema.name, vector_schema.data_type)
        else:
            raise ValueError("Group by query requires a vector or document id")

        target_dtype = DTYPE_MAP.get(vector_schema.data_type.value)
        search_query.set_vector(
            vector_schema._get_object(),
            convert_to_numpy(vec_data, target_dtype) if target_dtype else vec_data,
        )

    def _build_search_query(
        self, ctx: QueryContext, query: Query, collection: _Collection