caikit.interfaces.nlp.data_model.embedding_vectors

Data structures for embedding vector representations

Attributes

log
error

Classes

EmbeddingResult	Result from text embedding task
EmbeddingResults	Results from text embeddings task

Module Contents

caikit.interfaces.nlp.data_model.embedding_vectors.log

caikit.interfaces.nlp.data_model.embedding_vectors.error

class caikit.interfaces.nlp.data_model.embedding_vectors.EmbeddingResult

Bases: caikit.core.DataObjectBase

Result from text embedding task

result: py_to_proto.dataclass_to_proto.Annotated[caikit.interfaces.common.data_model.Vector1D, FieldNumber(1)]

producer_id: py_to_proto.dataclass_to_proto.Annotated[caikit.interfaces.common.data_model.ProducerId, FieldNumber(2)]

input_token_count: py_to_proto.dataclass_to_proto.Annotated[int | None, FieldNumber(3)]

class caikit.interfaces.nlp.data_model.embedding_vectors.EmbeddingResults

Bases: caikit.core.DataObjectBase

Results from text embeddings task

results: py_to_proto.dataclass_to_proto.Annotated[caikit.interfaces.common.data_model.ListOfVector1D, FieldNumber(1)]

producer_id: py_to_proto.dataclass_to_proto.Annotated[caikit.interfaces.common.data_model.ProducerId, FieldNumber(2)]

input_token_count: py_to_proto.dataclass_to_proto.Annotated[int | None, FieldNumber(3)]