fix: correct default value

Signed-off-by: Chojan Shang <chojan.shang@vesoft.com>

Author: Chojan Shang

README.md

@@ -34,53 +34,60 @@ Prefer a step-by-step walkthrough? Read the [Quickstart guide](docs/quickstart.m
 
 Embed the agent inside another Python process without spawning your own pipes:
 
+```python
+import asyncio
+import sys
+from pathlib import Path
+
+from acp import spawn_agent_process
+from acp.schema import InitializeRequest, NewSessionRequest, PromptRequest, TextContentBlock
+
+
+async def main() -> None:
+    agent_script = Path("examples/echo_agent.py")
+    async with spawn_agent_process(lambda _agent: YourClient(), sys.executable, str(agent_script)) as (conn, _proc):
+        await conn.initialize(InitializeRequest(protocolVersion=1))
+        session = await conn.newSession(NewSessionRequest(cwd=str(agent_script.parent), mcpServers=[]))
+        await conn.prompt(
+            PromptRequest(
+                sessionId=session.sessionId,
+                prompt=[TextContentBlock(type="text", text="Hello!")],
+            )
+        )
+
+
+asyncio.run(main())
+```
+
+`spawn_client_process` mirrors this pattern for the inverse direction.
+
+### Minimal agent sketch
+
 ```python
 import asyncio
 
 from acp import (
     Agent,
     AgentSideConnection,
-    AuthenticateRequest,
-    AuthenticateResponse,
-    CancelNotification,
     InitializeRequest,
     InitializeResponse,
-    LoadSessionRequest,
-    LoadSessionResponse,
     NewSessionRequest,
     NewSessionResponse,
     PromptRequest,
     PromptResponse,
-    SetSessionModeRequest,
-    SetSessionModeResponse,
-    stdio_streams,
-    PROTOCOL_VERSION,
-)
-from acp.schema import (
-    AgentCapabilities,
-    AgentMessageChunk,
-    McpCapabilities,
-    PromptCapabilities,
     SessionNotification,
-    TextContentBlock,
+    stdio_streams,
 )
+from acp.schema import TextContentBlock, AgentMessageChunk
+
 
 class EchoAgent(Agent):
     def __init__(self, conn):
         self._conn = conn
 
     async def initialize(self, params: InitializeRequest) -> InitializeResponse:
-        mcp_caps: McpCapabilities = McpCapabilities(http=False, sse=False)
-        prompt_caps: PromptCapabilities = PromptCapabilities(audio=False, embeddedContext=False, image=False)
-        agent_caps: AgentCapabilities = AgentCapabilities(
-            loadSession=False,
-            mcpCapabilities=mcp_caps,
-            promptCapabilities=prompt_caps,
-        )
-        return InitializeResponse(
-            protocolVersion=PROTOCOL_VERSION,
-            agentCapabilities=agent_caps,
-        )
+        return InitializeResponse(protocolVersion=params.protocolVersion)
+
     async def newSession(self, params: NewSessionRequest) -> NewSessionResponse:
         return NewSessionResponse(sessionId="sess-1")
 

examples/echo_agent.py

@@ -1,57 +1,16 @@
 import asyncio
 
-from acp import (
-    Agent,
-    AgentSideConnection,
-    AuthenticateRequest,
-    AuthenticateResponse,
-    CancelNotification,
-    InitializeRequest,
-    InitializeResponse,
-    LoadSessionRequest,
-    LoadSessionResponse,
-    NewSessionRequest,
-    NewSessionResponse,
-    PromptRequest,
-    PromptResponse,
-    SetSessionModeRequest,
-    SetSessionModeResponse,
-    stdio_streams,
-    PROTOCOL_VERSION,
-)
-from acp.schema import (
-    AgentCapabilities,
-    AgentMessageChunk,
-    McpCapabilities,
-    PromptCapabilities,
-    SessionNotification,
-    TextContentBlock,
-)
+from acp import Agent, AgentSideConnection, PromptRequest, PromptResponse, SessionNotification, stdio_streams
+from acp.schema import AgentMessageChunk, TextContentBlock
 
 
 class EchoAgent(Agent):
     def __init__(self, conn):
         self._conn = conn
 
-    async def initialize(self, params: InitializeRequest) -> InitializeResponse:
-        mcp_caps: McpCapabilities = McpCapabilities(http=False, sse=False)
-        prompt_caps: PromptCapabilities = PromptCapabilities(audio=False, embeddedContext=False, image=False)
-        agent_caps: AgentCapabilities = AgentCapabilities(
-            loadSession=False,
-            mcpCapabilities=mcp_caps,
-            promptCapabilities=prompt_caps,
-        )
-        return InitializeResponse(
-            protocolVersion=PROTOCOL_VERSION,
-            agentCapabilities=agent_caps,
-        )
-
-    async def newSession(self, params: NewSessionRequest) -> NewSessionResponse:
-        return NewSessionResponse(sessionId="sess-1")
-
     async def prompt(self, params: PromptRequest) -> PromptResponse:
         for block in params.prompt:
-            text = block.get("text", "") if isinstance(block, dict) else getattr(block, "text", "")
+            text = getattr(block, "text", "")
             await self._conn.sessionUpdate(
                 SessionNotification(
                     sessionId=params.sessionId,

scripts/gen_schema.py

@@ -74,12 +74,24 @@
 )
 
 DEFAULT_VALUE_OVERRIDES: tuple[tuple[str, str, str], ...] = (
-    ("AgentCapabilities", "mcpCapabilities", "None"),
-    ("AgentCapabilities", "promptCapabilities", "None"),
-    ("ClientCapabilities", "fs", "None"),
-    ("ClientCapabilities", "terminal", "None"),
-    ("InitializeRequest", "clientCapabilities", "None"),
-    ("InitializeResponse", "agentCapabilities", "None"),
+    ("AgentCapabilities", "mcpCapabilities", "McpCapabilities(http=False, sse=False)"),
+    (
+        "AgentCapabilities",
+        "promptCapabilities",
+        "PromptCapabilities(audio=False, embeddedContext=False, image=False)",
+    ),
+    ("ClientCapabilities", "fs", "FileSystemCapability(readTextFile=False, writeTextFile=False)"),
+    ("ClientCapabilities", "terminal", "False"),
+    (
+        "InitializeRequest",
+        "clientCapabilities",
+        "ClientCapabilities(fs=FileSystemCapability(readTextFile=False, writeTextFile=False), terminal=False)",
+    ),
+    (
+        "InitializeResponse",
+        "agentCapabilities",
+        "AgentCapabilities(loadSession=False, mcpCapabilities=McpCapabilities(http=False, sse=False), promptCapabilities=PromptCapabilities(audio=False, embeddedContext=False, image=False))",
+    ),
 )
 
 

src/acp/schema.py

@@ -402,11 +402,11 @@ class AgentCapabilities(BaseModel):
     mcpCapabilities: Annotated[
         Optional[McpCapabilities],
         Field(description="MCP capabilities supported by the agent."),
-    ] = None
+    ] = McpCapabilities(http=False, sse=False)
     promptCapabilities: Annotated[
         Optional[PromptCapabilities],
         Field(description="Prompt capabilities supported by the agent."),
-    ] = None
+    ] = PromptCapabilities(audio=False, embeddedContext=False, image=False)
 
 
 class Annotations(BaseModel):
@@ -476,11 +476,11 @@ class ClientCapabilities(BaseModel):
         Field(
             description="File system capabilities supported by the client.\nDetermines which file operations the agent can request."
         ),
-    ] = None
+    ] = FileSystemCapability(readTextFile=False, writeTextFile=False)
     terminal: Annotated[
         Optional[bool],
         Field(description="Whether the Client support all `terminal/*` methods."),
-    ] = None
+    ] = False
 
 
 class TextContentBlock(BaseModel):
@@ -575,7 +575,7 @@ class InitializeRequest(BaseModel):
     clientCapabilities: Annotated[
         Optional[ClientCapabilities],
         Field(description="Capabilities supported by the client."),
-    ] = None
+    ] = ClientCapabilities(fs=FileSystemCapability(readTextFile=False, writeTextFile=False), terminal=False)
     protocolVersion: Annotated[
         int,
         Field(
@@ -594,7 +594,11 @@ class InitializeResponse(BaseModel):
     agentCapabilities: Annotated[
         Optional[AgentCapabilities],
         Field(description="Capabilities supported by the agent."),
-    ] = None
+    ] = AgentCapabilities(
+        loadSession=False,
+        mcpCapabilities=McpCapabilities(http=False, sse=False),
+        promptCapabilities=PromptCapabilities(audio=False, embeddedContext=False, image=False),
+    )
     authMethods: Annotated[
         Optional[List[AuthMethod]],
         Field(description="Authentication methods supported by the agent."),