refactor(agent): align chat schema with full run agent input

2026-03-03 15:29:46 +08:00
parent 5e169251fe
commit 17e6de177c
3 changed files with 70 additions and 0 deletions
@@ -1,10 +1,23 @@
 from __future__ import annotations

+from typing import Any
 from uuid import UUID

 from pydantic import BaseModel, Field


+class RunAgentInput(BaseModel):
+    threadId: str
+    runId: str
+    parentRunId: str | None = None
+    state: dict[str, Any]
+    messages: list[dict[str, Any]]
+    tools: list[dict[str, Any]]
+    context: list[dict[str, Any]]
+    forwardedProps: dict[str, Any]
+    resume: dict[str, Any] | None = None
+
+
 class AgentChatRunRequest(BaseModel):
    message: str = Field(min_length=1, max_length=8000)
    session_id: UUID | None = None