mycode-sdk 0.10.0__tar.gz → 0.11.1__tar.gz

{mycode_sdk-0.10.0 → mycode_sdk-0.11.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mycode-sdk
-Version: 0.10.0
+Version: 0.11.1
 Summary: Lightweight Python SDK for building AI agents.
 Project-URL: Homepage, https://github.com/legibet/mycode
 Project-URL: Repository, https://github.com/legibet/mycode
@@ -71,7 +71,7 @@ print(result.text)
 
 ## Multi-turn conversations
 
-Call `achat()` or `run()` again on the same `Agent` — history accumulates automatically:
+Call `achat()` or `run()` again on the same `Agent` to continue the conversation:
 
 ```python
 agent = Agent(model="claude-sonnet-4-6", api_key="...")
@@ -117,7 +117,7 @@ agent = Agent(
 )
 ```
 
-Construct another `Agent` with the same `(session_dir, session_id)` later to resume the conversation — the history is loaded automatically.
+Construct another `Agent` with the same `(session_dir, session_id)` to load the conversation history.
 
 ## Built-in tools
 
@@ -167,6 +167,8 @@ def summarize_file(ctx: ToolContext, path: str) -> str:
     return result.output.splitlines()[0] if result.output else ""
 ```
 
+Async tools use `await ctx.aread()`, `await ctx.awrite()`, `await ctx.aedit()`, and `await ctx.abash()`. Use `await ctx.acall(name, args)` to call another registered tool by name.
+
 ## Tool hooks
 
 Inspect or replace tool calls before they run. Return `None` from `before_tool` to let the tool execute, or a `ToolExecutionResult` to skip it:

{mycode_sdk-0.10.0 → mycode_sdk-0.11.1}/README.md

@@ -44,7 +44,7 @@ print(result.text)
 
 ## Multi-turn conversations
 
-Call `achat()` or `run()` again on the same `Agent` — history accumulates automatically:
+Call `achat()` or `run()` again on the same `Agent` to continue the conversation:
 
 ```python
 agent = Agent(model="claude-sonnet-4-6", api_key="...")
@@ -90,7 +90,7 @@ agent = Agent(
 )
 ```
 
-Construct another `Agent` with the same `(session_dir, session_id)` later to resume the conversation — the history is loaded automatically.
+Construct another `Agent` with the same `(session_dir, session_id)` to load the conversation history.
 
 ## Built-in tools
 
@@ -140,6 +140,8 @@ def summarize_file(ctx: ToolContext, path: str) -> str:
     return result.output.splitlines()[0] if result.output else ""
 ```
 
+Async tools use `await ctx.aread()`, `await ctx.awrite()`, `await ctx.aedit()`, and `await ctx.abash()`. Use `await ctx.acall(name, args)` to call another registered tool by name.
+
 ## Tool hooks
 
 Inspect or replace tool calls before they run. Return `None` from `before_tool` to let the tool execute, or a `ToolExecutionResult` to skip it:

{mycode_sdk-0.10.0 → mycode_sdk-0.11.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "mycode-sdk"
-version = "0.10.0"
+version = "0.11.1"
 description = "Lightweight Python SDK for building AI agents."
 readme = "README.md"
 requires-python = ">=3.12"

{mycode_sdk-0.10.0 → mycode_sdk-0.11.1}/src/mycode/agent.py

@@ -87,6 +87,7 @@ class Agent:
         compact_threshold: float | None = None,
         reasoning_effort: str | None = None,
         supports_reasoning: bool | None = None,
+        supports_reasoning_effort: bool = False,
         supports_image_input: bool | None = None,
         supports_pdf_input: bool | None = None,
         system: str = "",
@@ -127,11 +128,16 @@ class Agent:
         self.temperature = float(temperature)
         self.compact_threshold = compact_threshold if compact_threshold is not None else DEFAULT_COMPACT_THRESHOLD
         self.reasoning_effort = reasoning_effort
+        # Whether the endpoint accepts the effort knob. Comes from provider
+        # config, not the catalog, so it stays out of refresh_capabilities.
+        self.supports_reasoning_effort = supports_reasoning_effort
 
         self.system = system
         self.hooks = hooks or Hooks()
         self._cancel_event = asyncio.Event()
+        self._event_loop: asyncio.AbstractEventLoop | None = None
         self._provider_event_task: asyncio.Future[ProviderStreamEvent] | None = None
+        self._active_tool_task: asyncio.Task[ToolExecutionResult] | None = None
 
         # History resolution:
         # - messages is None → auto-resume from disk if the session exists
@@ -201,10 +207,24 @@ class Agent:
     def cancel(self) -> None:
         """Request cancellation of the in-flight turn."""
 
+        try:
+            running_loop = asyncio.get_running_loop()
+        except RuntimeError:
+            running_loop = None
+
+        loop = self._event_loop
+        if loop is not None and loop.is_running() and running_loop is not loop:
+            loop.call_soon_threadsafe(self._cancel_in_loop)
+            return
+        self._cancel_in_loop()
+
+    def _cancel_in_loop(self) -> None:
         self._cancel_event.set()
         self.tools.cancel_active()
         if self._provider_event_task and not self._provider_event_task.done():
             self._provider_event_task.cancel()
+        if self._active_tool_task and not self._active_tool_task.done():
+            self._active_tool_task.cancel()
 
     def clear(self) -> None:
         """Drop the in-memory conversation history."""
@@ -215,6 +235,23 @@ class Agent:
     # Tool execution
     # ------------------------------------------------------------------
 
+    async def _execute_tool(
+        self,
+        spec: ToolSpec,
+        args: dict[str, Any],
+        ctx: ToolContext,
+    ) -> ToolExecutionResult:
+        if not spec.is_async:
+            return await self.tools.aexecute(spec.name, args, ctx)
+
+        task = asyncio.create_task(self.tools.aexecute(spec.name, args, ctx))
+        self._active_tool_task = task
+        try:
+            return await task
+        finally:
+            if self._active_tool_task is task:
+                self._active_tool_task = None
+
     async def _run_tool_call(self, tool_use: dict[str, Any]) -> AsyncIterator[Event]:
         """Run one tool call and emit the standard tool events."""
 
@@ -261,13 +298,23 @@ class Agent:
             return
 
         if spec.streams_output:
-            async for event in self._run_streaming_tool(tool_id=tool_id, name=name, args=args, hook_ctx=hook_ctx):
+            async for event in self._run_streaming_tool(
+                tool_id=tool_id,
+                spec=spec,
+                args=args,
+                hook_ctx=hook_ctx,
+            ):
                 yield event
             return
 
         try:
             ctx = self._ctx_for_call(tool_id)
-            result = await asyncio.to_thread(self.tools.execute, name, args, ctx)
+            result = await self._execute_tool(spec, args, ctx)
+        except asyncio.CancelledError:
+            if self._cancel_event.is_set():
+                yield self._error_done(tool_id, "error: cancelled")
+                return
+            raise
         except Exception as exc:  # pragma: no cover - defensive
             result = ToolExecutionResult(output=f"error: {exc}", is_error=True)
 
@@ -277,7 +324,7 @@ class Agent:
         self,
         *,
         tool_id: str,
-        name: str,
+        spec: ToolSpec,
         args: dict[str, Any],
         hook_ctx: ToolHookContext,
     ) -> AsyncIterator[Event]:
@@ -291,13 +338,13 @@ class Agent:
 
         ctx = self._ctx_for_call(tool_id, emit=on_output)
 
-        async def run_in_thread() -> ToolExecutionResult:
+        async def run_tool() -> ToolExecutionResult:
             try:
-                return await asyncio.to_thread(self.tools.execute, name, args, ctx)
+                return await self._execute_tool(spec, args, ctx)
             finally:
                 loop.call_soon_threadsafe(output_queue.put_nowait, None)
 
-        task = asyncio.create_task(run_in_thread())
+        task = asyncio.create_task(run_tool())
         was_cancelled = False
         output_parts: list[str] = []
 
@@ -319,8 +366,8 @@ class Agent:
                 output_parts.append(output)
                 yield Event("tool_output", {"tool_use_id": tool_id, "output": output})
 
-        if was_cancelled:
-            with suppress(Exception):
+        if was_cancelled or (self._cancel_event.is_set() and task.cancelled()):
+            with suppress(asyncio.CancelledError, Exception):
                 await task
             output = "\n".join([*output_parts, "error: cancelled"]) if output_parts else "error: cancelled"
             yield self._error_done(tool_id, output)
@@ -490,6 +537,7 @@ class Agent:
         async def persist(message: ConversationMessage) -> None:
             await self._persist_message(message, on_persist)
 
+        self._event_loop = asyncio.get_running_loop()
         self._cancel_event.clear()
 
         user_message: ConversationMessage
@@ -742,6 +790,7 @@ class Agent:
         :meth:`cancel` stops the summary request.
         """
 
+        self._event_loop = asyncio.get_running_loop()
         self._cancel_event.clear()
         adapter = get_provider_adapter(self.provider)
         return await self._compact(adapter, on_persist)