graphai.api.embedding.schemas module

class graphai.api.embedding.schemas.EmbeddingRequestBase(*, model_type: Literal['all-MiniLM-L12-v2', 'Solon-embeddings-large-0.1'] = 'all-MiniLM-L12-v2', force: bool = False)

Bases: BaseModel

model_type: Literal['all-MiniLM-L12-v2', 'Solon-embeddings-large-0.1']

force: bool

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class graphai.api.embedding.schemas.EmbeddingFingerprintRequest(*, model_type: Literal['all-MiniLM-L12-v2', 'Solon-embeddings-large-0.1'] = 'all-MiniLM-L12-v2', force: bool = False, text: str)

Bases: EmbeddingRequestBase

text: str

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class graphai.api.embedding.schemas.EmbeddingRequest(*, model_type: Literal['all-MiniLM-L12-v2', 'Solon-embeddings-large-0.1'] = 'all-MiniLM-L12-v2', force: bool = False, text: List[str] | str, no_cache: bool = False)

Bases: EmbeddingRequestBase

text: List[str] | str

no_cache: bool

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class graphai.api.embedding.schemas.EmbeddingTaskResponse(*, result: str | None = None, text_too_large: bool, successful: bool, fresh: bool, device: str | None = None, model_type: Literal['all-MiniLM-L12-v2', 'Solon-embeddings-large-0.1'] | None = 'all-MiniLM-L12-v2')

Bases: BaseModel

result: str | None

text_too_large: bool

successful: bool

fresh: bool

device: str | None

model_type: Literal['all-MiniLM-L12-v2', 'Solon-embeddings-large-0.1'] | None

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class graphai.api.embedding.schemas.EmbeddingResponse(*, task_id: str, task_name: str | None = None, task_status: str, task_result: EmbeddingTaskResponse | List[EmbeddingTaskResponse] | None)

Bases: TaskStatusResponse

task_result: EmbeddingTaskResponse | List[EmbeddingTaskResponse] | None

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].