Spaces:
Running
Running
| """Typed response envelopes for the public OpenAPI contract.""" | |
| from __future__ import annotations | |
| from typing import Literal | |
| from pydantic import BaseModel, ConfigDict, JsonValue, RootModel | |
| from src.constants.data import KGChanges, Observation, StructuredData, TextChunk | |
| from src.constants.embeddings import Vector | |
| from src.constants.kg import Node, Predicate, Triple | |
| class IngestionAcceptedResponse(BaseModel): | |
| message: str | |
| task_id: str | |
| class LoginInfoResponse(BaseModel): | |
| is_system_pat: bool | |
| brain_id: str | None | |
| class BrainLifecycleResponse(BaseModel): | |
| success: bool | |
| brain_id: str | |
| operation: Literal["clear", "delete"] | |
| existed: bool | |
| cleared_backends: list[str] | |
| class StringListResponse(RootModel[list[str]]): | |
| pass | |
| class FlatHopNode(BaseModel): | |
| uuid: str | |
| labels: list[str] | |
| name: str | |
| class FlatHopPredicate(BaseModel): | |
| uuid: str | |
| name: str | |
| direction: str | None = None | |
| HopNode = Node | FlatHopNode | |
| HopPredicate = Predicate | FlatHopPredicate | |
| SecondDegreeHop = tuple[ | |
| HopNode, | |
| list[tuple[HopPredicate, HopNode, list[tuple[HopPredicate, HopNode]]]], | |
| ] | |
| class HopListResponse(RootModel[list[SecondDegreeHop]]): | |
| pass | |
| class RelationshipListResponse(BaseModel): | |
| message: str | |
| relationships: list[Triple] | |
| total: int | |
| class EntityListResponse(BaseModel): | |
| message: str | |
| entities: list[Node] | |
| total: int | |
| class StructuredDataItemResponse(BaseModel): | |
| message: str | |
| data: StructuredData | |
| class StructuredDataListResponse(BaseModel): | |
| message: str | |
| data: list[StructuredData] | |
| count: int | |
| total: int | |
| class TypeListResponse(BaseModel): | |
| message: str | |
| types: list[str] | |
| count: int | |
| class TextChunkListResponse(BaseModel): | |
| message: str | |
| data: list[TextChunk] | |
| total: int | |
| class ObservationItemResponse(BaseModel): | |
| message: str | |
| observation: Observation | |
| class ObservationListResponse(BaseModel): | |
| message: str | |
| observations: list[Observation] | |
| count: int | |
| class LabelListResponse(BaseModel): | |
| message: str | |
| labels: list[str] | |
| count: int | |
| class ChangelogItemResponse(BaseModel): | |
| message: str | |
| changelog: KGChanges | |
| class ChangelogListResponse(BaseModel): | |
| message: str | |
| changelogs: list[KGChanges] | |
| count: int | |
| class VectorStoreInfo(BaseModel): | |
| name: str | |
| dimension: int | |
| class VectorStoreListResponse(BaseModel): | |
| stores: list[VectorStoreInfo] | |
| class VectorListResponse(BaseModel): | |
| message: str | |
| store: str | |
| vectors: list[Vector] | |
| total: int | |
| class TaskStateResponse(BaseModel): | |
| """Task payload with stable state fields and pipeline-specific extensions.""" | |
| model_config = ConfigDict(extra="allow") | |
| task_id: str | None = None | |
| id: str | None = None | |
| status: str | |
| stage: str | None = None | |
| error: str | None = None | |
| result: JsonValue | None = None | |
| class TaskListResponse(BaseModel): | |
| tasks: list[TaskStateResponse] | |