fix: allow camelCase for accessing

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

Author: frostming

scripts/gen_schema.py

@@ -6,6 +6,7 @@
 import re
 import subprocess
 import sys
+import textwrap
 from collections.abc import Callable
 from dataclasses import dataclass
 from pathlib import Path
@@ -327,11 +328,20 @@ def _ensure_custom_base_model(content: str) -> str:
         if not has_config:
             new_imports.append("ConfigDict")
         lines[idx] = "from pydantic import " + ", ".join(new_imports)
+        to_insert = textwrap.dedent("""\
+            class BaseModel(_BaseModel):
+                model_config = ConfigDict(populate_by_name=True)
+
+                def __getattr__(self, item: str) -> Any:
+                    if item.lower() != item:
+                        snake_cased = "".join("_" + c.lower() if c.isupper() and i > 0 else c.lower() for i, c in enumerate(item))
+                        return getattr(self, snake_cased)
+                    raise AttributeError(f"'{type(self).__name__}' object has no attribute '{item}'")
+        """)
         insert_idx = idx + 1
         lines.insert(insert_idx, "")
-        lines.insert(insert_idx + 1, "class BaseModel(_BaseModel):")
-        lines.insert(insert_idx + 2, "    model_config = ConfigDict(populate_by_name=True)")
-        lines.insert(insert_idx + 3, "")
+        for offset, line in enumerate(to_insert.splitlines(), 1):
+            lines.insert(insert_idx + offset, line)
         break
     return "\n".join(lines) + "\n"
 

src/acp/agent/connection.py

@@ -61,7 +61,7 @@ def __init__(
         agent = to_agent(cast(Client, self)) if callable(to_agent) else to_agent
         if not isinstance(input_stream, asyncio.StreamWriter) or not isinstance(output_stream, asyncio.StreamReader):
             raise TypeError(_AGENT_CONNECTION_ERROR)
-        handler = build_agent_router(agent)  # type: ignore[arg-type]
+        handler = build_agent_router(cast(Agent, agent))
         self._conn = Connection(handler, input_stream, output_stream, listening=listening, **connection_kwargs)
         if on_connect := getattr(agent, "on_connect", None):
             on_connect(self)

src/acp/client/connection.py

@@ -56,7 +56,7 @@ def __init__(
         if not isinstance(input_stream, asyncio.StreamWriter) or not isinstance(output_stream, asyncio.StreamReader):
             raise TypeError(_CLIENT_CONNECTION_ERROR)
         client = to_client(cast(Agent, self)) if callable(to_client) else to_client
-        handler = build_client_router(client)  # type: ignore[arg-type]
+        handler = build_client_router(cast(Client, client))
         self._conn = Connection(handler, input_stream, output_stream, **connection_kwargs)
         if on_connect := getattr(client, "on_connect", None):
             on_connect(self)

src/acp/schema.py

@@ -19,6 +19,12 @@
 class BaseModel(_BaseModel):
     model_config = ConfigDict(populate_by_name=True)
 
+    def __getattr__(self, item: str) -> Any:
+        if item.lower() != item:
+            snake_cased = "".join("_" + c.lower() if c.isupper() and i > 0 else c.lower() for i, c in enumerate(item))
+            return getattr(self, snake_cased)
+        raise AttributeError(f"'{type(self).__name__}' object has no attribute '{item}'")
+
 
 class Jsonrpc(Enum):
     field_2_0 = "2.0"