fix: overrides in gen_schema.py

Signed-off-by: Frost Ming <me@frostming.com>

Author: frostming

scripts/gen_schema.py

@@ -85,7 +85,7 @@
     ("PermissionOption", "kind", "PermissionOptionKind", False),
     ("PlanEntry", "priority", "PlanEntryPriority", False),
     ("PlanEntry", "status", "PlanEntryStatus", False),
-    ("PromptResponse", "stopReason", "StopReason", False),
+    ("PromptResponse", "stop_reason", "StopReason", False),
     ("ToolCallProgress", "kind", "ToolKind", True),
     ("ToolCallProgress", "status", "ToolCallStatus", True),
     ("ToolCallStart", "kind", "ToolKind", True),
@@ -95,23 +95,23 @@
 )
 
 DEFAULT_VALUE_OVERRIDES: tuple[tuple[str, str, str], ...] = (
-    ("AgentCapabilities", "mcpCapabilities", "McpCapabilities(http=False, sse=False)"),
+    ("AgentCapabilities", "mcp_capabilities", "McpCapabilities()"),
     (
         "AgentCapabilities",
-        "promptCapabilities",
-        "PromptCapabilities(audio=False, embeddedContext=False, image=False)",
+        "prompt_capabilities",
+        "PromptCapabilities()",
     ),
-    ("ClientCapabilities", "fs", "FileSystemCapability(readTextFile=False, writeTextFile=False)"),
+    ("ClientCapabilities", "fs", "FileSystemCapability()"),
     ("ClientCapabilities", "terminal", "False"),
     (
         "InitializeRequest",
-        "clientCapabilities",
-        "ClientCapabilities(fs=FileSystemCapability(readTextFile=False, writeTextFile=False), terminal=False)",
+        "client_capabilities",
+        "ClientCapabilities()",
     ),
     (
         "InitializeResponse",
-        "agentCapabilities",
-        "AgentCapabilities(loadSession=False, mcpCapabilities=McpCapabilities(http=False, sse=False), promptCapabilities=PromptCapabilities(audio=False, embeddedContext=False, image=False))",
+        "agent_capabilities",
+        "AgentCapabilities()",
     ),
 )
 

scripts/gen_signature.py

@@ -86,7 +86,7 @@ def _to_param_def(self, name: str, field: FieldInfo) -> tuple[ast.arg, ast.expr
         ann = field.annotation
         if field.default is PydanticUndefined:
             default = None
-        elif isinstance(field.default, dict):
+        elif isinstance(field.default, dict | BaseModel):
             default = ast.Constant(None)
         else:
             default = ast.Constant(value=field.default)
@@ -128,6 +128,9 @@ def _format_annotation(self, annotation: t.Any) -> ast.expr:
 
 
 def gen_signature(source_dir: Path) -> None:
+    import importlib
+
+    importlib.reload(schema)  # Ensure schema is up to date
     for source_file in source_dir.rglob("*.py"):
         transformer = NodeTransformer()
         transformer.transform(source_file)

src/acp/schema.py

@@ -623,15 +623,15 @@ class AgentCapabilities(BaseModel):
             alias="mcpCapabilities",
             description="MCP capabilities supported by the agent.",
         ),
-    ] = {"http": False, "sse": False}
+    ] = McpCapabilities()
     # Prompt capabilities supported by the agent.
     prompt_capabilities: Annotated[
         Optional[PromptCapabilities],
         Field(
             alias="promptCapabilities",
             description="Prompt capabilities supported by the agent.",
         ),
-    ] = {"audio": False, "embeddedContext": False, "image": False}
+    ] = PromptCapabilities()
 
 
 class AgentErrorMessage(BaseModel):
@@ -731,7 +731,7 @@ class ClientCapabilities(BaseModel):
         Field(
             description="File system capabilities supported by the client.\nDetermines which file operations the agent can request."
         ),
-    ] = FileSystemCapability(readTextFile=False, writeTextFile=False)
+    ] = FileSystemCapability()
     # Whether the Client support all `terminal/*` methods.
     terminal: Annotated[
         Optional[bool],
@@ -870,7 +870,7 @@ class InitializeRequest(BaseModel):
             alias="clientCapabilities",
             description="Capabilities supported by the client.",
         ),
-    ] = {"fs": {"readTextFile": False, "writeTextFile": False}, "terminal": False}
+    ] = ClientCapabilities()
     # Information about the Client name and version sent to the Agent.
     #
     # Note: in future versions of the protocol, this will be required.
@@ -906,15 +906,7 @@ class InitializeResponse(BaseModel):
             alias="agentCapabilities",
             description="Capabilities supported by the agent.",
         ),
-    ] = {
-        "loadSession": False,
-        "mcpCapabilities": {"http": False, "sse": False},
-        "promptCapabilities": {
-            "audio": False,
-            "embeddedContext": False,
-            "image": False,
-        },
-    }
+    ] = AgentCapabilities()
     # Information about the Agent name and version sent to the Client.
     #
     # Note: in future versions of the protocol, this will be required.
@@ -1031,7 +1023,7 @@ class PromptResponse(BaseModel):
     ] = None
     # Indicates why the agent stopped processing the turn.
     stop_reason: Annotated[
-        str,
+        StopReason,
         Field(
             alias="stopReason",
             description="Indicates why the agent stopped processing the turn.",