refactor(api): scope uv to s2s-client and validate compatible responses

2026-10-06 18:23:08 -03:00
parent 92c53dc7ca
commit f8d5b7857a
39 changed files with 316 additions and 67 deletions
@@ -0,0 +1,125 @@
+"""Response contracts without imposing new assumptions on upstream JSON.
+
+Only fields constructed by this service have concrete types. Upstream-owned
+values remain Any, and unknown fields remain allowed at every modeled level.
+Responses are validated but serialized from their original payloads so omitted
+fields, numeric representations, key order, and unknown values stay unchanged.
+"""
+
+from typing import Any, Literal
+
+from pydantic import BaseModel, ConfigDict, RootModel
+
+
+class OpenResponse(BaseModel):
+    model_config = ConfigDict(extra="allow", strict=True)
+
+
+class ConfigResponse(OpenResponse):
+    search: bool
+    lb: bool
+    allowDirect: bool
+    s2sUrl: str
+    primaryPrompt: str
+    reasoningType: Literal["PROXY", "OPENAI", "NODE_RED"]
+    injectProfile: bool
+    debugPrompt: bool
+    startSessionMode: str
+    theme: str
+    apipilot: bool
+
+
+class SearchResult(OpenResponse):
+    # Serper values have historically passed through, including nulls, numbers,
+    # and objects. The service guarantees these keys, not their value types.
+    title: Any
+    snippet: Any
+    url: Any
+
+
+class SearchResponse(OpenResponse):
+    query: str
+    answer: Any
+    results: list[SearchResult]
+
+
+class ReportEntry(OpenResponse):
+    sourceScheduleLabel: Any
+    date: Any
+
+
+class ReportsResponse(OpenResponse):
+    count: int
+    wrappedLen: int
+    hints: list[str]
+    entries: list[ReportEntry]
+    rawCount: int | None = None
+    cap: int | None = None
+    skipped: bool | None = None
+    reason: str | None = None
+    truncated: bool | None = None
+    note: str | None = None
+
+
+class ToolInvocation(OpenResponse):
+    name: str
+    success: bool
+    result_length: int
+
+
+class ReasoningResponse(OpenResponse):
+    answer: str
+    log_only: bool | None = None
+    usage: Any = None
+    usage_pass2: Any = None
+    latency_ms: Any = None
+    intent: Any = None
+    caller_profile_keys: Any = None
+    debug: Any = None
+    apipilot_invoked: ToolInvocation | None = None
+
+
+class ToolDefinition(OpenResponse):
+    name: str
+    description: str
+    parameters: dict[str, Any]
+    requires_auth: bool
+
+
+class ToolCatalog(OpenResponse):
+    count: int
+    include_unauthorized: bool
+    tools: list[ToolDefinition]
+
+
+class ToolCatalogResponse(OpenResponse):
+    success: bool
+    data: ToolCatalog | None
+    error: Any
+
+
+class ToolExecutionResponse(OpenResponse):
+    success: bool
+    # The upstream may omit these fields or return any JSON value for them.
+    data: Any = None
+    error: Any = None
+
+
+class AllocationResponse(OpenResponse):
+    """LB-owned object: no new required fields or value-type restrictions."""
+
+    state: Any = None
+    session_id: Any = None
+    connect_url: Any = None
+    pending_timeout_s: Any = None
+    queue_id: Any = None
+    position: Any = None
+    poll_interval_s: Any = None
+
+
+class UserProfileResponse(RootModel[Any]):
+    """The profile endpoint relays objects, arrays, scalars, and null verbatim."""
+
+
+class QueueCancellationResponse(OpenResponse):
+    ok: bool