mycode-sdk 0.9.3__tar.gz → 0.9.5__tar.gz

{mycode_sdk-0.9.3 → mycode_sdk-0.9.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mycode-sdk
-Version: 0.9.3
+Version: 0.9.5
 Summary: Lightweight Python SDK for building AI agents.
 Project-URL: Homepage, https://github.com/legibet/mycode
 Project-URL: Repository, https://github.com/legibet/mycode
@@ -18,11 +18,11 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development
 Requires-Python: >=3.12
-Requires-Dist: anthropic>=0.106.0
-Requires-Dist: google-genai>=2.8.0
-Requires-Dist: griffelib>=2.0.0
-Requires-Dist: openai>=2.41.0
-Requires-Dist: pydantic>=2.13.0
+Requires-Dist: anthropic>=0.116.0
+Requires-Dist: google-genai>=2.11.0
+Requires-Dist: griffelib>=2.1.0
+Requires-Dist: openai>=2.45.0
+Requires-Dist: pydantic>=2.13.4
 Description-Content-Type: text/markdown
 
 # mycode-sdk

{mycode_sdk-0.9.3 → mycode_sdk-0.9.5}/pyproject.toml

@@ -1,10 +1,10 @@
 [build-system]
-requires = ["hatchling"]
+requires = ["hatchling>=1.31.0"]
 build-backend = "hatchling.build"
 
 [project]
 name = "mycode-sdk"
-version = "0.9.3"
+version = "0.9.5"
 description = "Lightweight Python SDK for building AI agents."
 readme = "README.md"
 requires-python = ">=3.12"
@@ -23,11 +23,11 @@ classifiers = [
 ]
 keywords = ["agent", "llm", "anthropic", "openai", "gemini", "sdk"]
 dependencies = [
-    "anthropic>=0.106.0",
-    "google-genai>=2.8.0",
-    "griffelib>=2.0.0",
-    "openai>=2.41.0",
-    "pydantic>=2.13.0",
+    "anthropic>=0.116.0",
+    "google-genai>=2.11.0",
+    "griffelib>=2.1.0",
+    "openai>=2.45.0",
+    "pydantic>=2.13.4",
 ]
 
 [project.urls]

{mycode_sdk-0.9.3 → mycode_sdk-0.9.5}/src/mycode/agent.py

@@ -152,16 +152,10 @@ class Agent:
         # (e.g. bash spill files) live next to the session JSONL; without a
         # session, fall back to a tempdir scoped to ``session_id``.
         if session_dir is not None:
-            tool_output_dir = session_dir / self.session_id / "tool-output"
+            self.tool_output_dir = session_dir / self.session_id / "tool-output"
         else:
-            tool_output_dir = Path(tempfile.gettempdir()) / "mycode" / self.session_id / "tool-output"
+            self.tool_output_dir = Path(tempfile.gettempdir()) / "mycode" / self.session_id / "tool-output"
         self.tools = ToolExecutor(tools)
-        self.tool_ctx = ToolContext(
-            executor=self.tools,
-            cwd=self.cwd,
-            tool_output_dir=tool_output_dir,
-            supports_image_input=False,
-        )
 
         self.refresh_capabilities(
             max_tokens=max_tokens,
@@ -201,7 +195,6 @@ class Agent:
         self.supports_reasoning: bool | None = meta.supports_reasoning
         self.supports_image_input: bool = bool(meta.supports_image_input)
         self.supports_pdf_input: bool = bool(meta.supports_pdf_input)
-        self.tool_ctx.supports_image_input = self.supports_image_input
 
     def cancel(self) -> None:
         """Request cancellation of the in-flight turn."""
@@ -330,11 +323,11 @@ class Agent:
             output = "\n".join([*output_parts, "error: cancelled"]) if output_parts else "error: cancelled"
             yield self._error_done(tool_id, output)
             return
-        else:
-            try:
-                result = await task
-            except Exception as exc:  # pragma: no cover - defensive
-                result = ToolExecutionResult(output=f"error: {exc}", is_error=True)
+
+        try:
+            result = await task
+        except Exception as exc:  # pragma: no cover - defensive
+            result = ToolExecutionResult(output=f"error: {exc}", is_error=True)
 
         yield await self._finish_tool_call(tool_id, hook_ctx, result)
 
@@ -360,13 +353,11 @@ class Agent:
         *,
         emit: Callable[[str], None] | None = None,
     ) -> ToolContext:
-        """Build a per-call ToolContext from the base context."""
-
         return ToolContext(
             executor=self.tools,
-            cwd=self.tool_ctx.cwd,
-            tool_output_dir=self.tool_ctx.tool_output_dir,
-            supports_image_input=self.tool_ctx.supports_image_input,
+            cwd=self.cwd,
+            tool_output_dir=self.tool_output_dir,
+            supports_image_input=self.supports_image_input,
             tool_call_id=tool_id,
             emit=emit,
         )
@@ -402,15 +393,12 @@ class Agent:
 
         provider_stream: AsyncIterator[ProviderStreamEvent] = adapter.stream_turn(request)
 
-        async def next_provider_event() -> ProviderStreamEvent:
-            return await anext(provider_stream)
-
         try:
             while True:
                 if self._cancel_event.is_set():
                     raise asyncio.CancelledError
 
-                self._provider_event_task = asyncio.create_task(next_provider_event())
+                self._provider_event_task = asyncio.ensure_future(anext(provider_stream))
                 try:
                     yield await self._provider_event_task
                 except StopAsyncIteration:
@@ -529,7 +517,10 @@ class Agent:
         adapter = get_provider_adapter(self.provider)
 
         turn_number = 0
-        while self.max_turns is None or turn_number < self.max_turns:
+        while True:
+            if self.max_turns is not None and turn_number >= self.max_turns:
+                yield Event("error", {"message": "max_turns reached"})
+                return
             turn_number += 1
             if self._cancel_event.is_set():
                 yield Event("error", {"message": "cancelled"})
@@ -632,13 +623,15 @@ class Agent:
 
             total_tokens = meta.get("total_tokens")
             if total_tokens:
-                payload: dict[str, Any] = {
-                    "total_tokens": total_tokens,
-                    "model": meta.get("model") or self.model,
-                    "provider": meta.get("provider") or self.provider,
-                    "context_window": meta["context_window"],
-                }
-                yield Event("usage", payload)
+                yield Event(
+                    "usage",
+                    {
+                        "total_tokens": total_tokens,
+                        "model": meta.get("model") or self.model,
+                        "provider": meta.get("provider") or self.provider,
+                        "context_window": meta["context_window"],
+                    },
+                )
 
             tool_calls = [
                 block
@@ -665,11 +658,10 @@ class Agent:
                             )
                         )
 
-                        if self._cancel_event.is_set():
-                            tool_result_message = build_message("user", tool_results)
-                            self.messages.append(tool_result_message)
-                            await persist(tool_result_message)
-                            return
+                    if self._cancel_event.is_set():
+                        # Skip remaining tool calls; the results collected so
+                        # far are still persisted below.
+                        break
 
                 tool_result_message = build_message("user", tool_results)
                 self.messages.append(tool_result_message)
@@ -693,11 +685,7 @@ class Agent:
                     )
 
             if not tool_calls:
-                break
-
-        else:
-            yield Event("error", {"message": "max_turns reached"})
-            return
+                return
 
     def run(
         self,

{mycode_sdk-0.9.3 → mycode_sdk-0.9.5}/src/mycode/messages.py

@@ -25,18 +25,20 @@ ContentBlock = dict[str, Any]
 ConversationMessage = dict[str, Any]
 
 
-def text_block(text: str, *, meta: dict[str, Any] | None = None) -> ContentBlock:
-    block: ContentBlock = {"type": "text", "text": text}
+def _set_meta(block: dict[str, Any], meta: dict[str, Any] | None) -> dict[str, Any]:
+    """Attach a copy of ``meta`` under the shared ``meta`` key, if non-empty."""
+
     if meta:
         block["meta"] = dict(meta)
     return block
 
 
+def text_block(text: str, *, meta: dict[str, Any] | None = None) -> ContentBlock:
+    return _set_meta({"type": "text", "text": text}, meta)
+
+
 def thinking_block(text: str, *, meta: dict[str, Any] | None = None) -> ContentBlock:
-    block: ContentBlock = {"type": "thinking", "text": text}
-    if meta:
-        block["meta"] = dict(meta)
-    return block
+    return _set_meta({"type": "thinking", "text": text}, meta)
 
 
 def image_block(
@@ -49,9 +51,7 @@ def image_block(
     block: ContentBlock = {"type": "image", "data": data, "mime_type": mime_type}
     if name:
         block["name"] = name
-    if meta:
-        block["meta"] = dict(meta)
-    return block
+    return _set_meta(block, meta)
 
 
 def document_block(
@@ -64,9 +64,7 @@ def document_block(
     block: ContentBlock = {"type": "document", "data": data, "mime_type": mime_type}
     if name:
         block["name"] = name
-    if meta:
-        block["meta"] = dict(meta)
-    return block
+    return _set_meta(block, meta)
 
 
 def tool_use_block(
@@ -82,9 +80,7 @@ def tool_use_block(
         "name": name,
         "input": dict(input or {}),
     }
-    if meta:
-        block["meta"] = dict(meta)
-    return block
+    return _set_meta(block, meta)
 
 
 def tool_result_block(
@@ -114,9 +110,7 @@ def tool_result_block(
         block["metadata"] = dict(metadata)
     if content:
         block["content"] = [dict(item) for item in content]
-    if meta:
-        block["meta"] = dict(meta)
-    return block
+    return _set_meta(block, meta)
 
 
 def user_text_message(text: str, *, meta: dict[str, Any] | None = None) -> ConversationMessage:
@@ -129,10 +123,7 @@ def build_message(
     *,
     meta: dict[str, Any] | None = None,
 ) -> ConversationMessage:
-    message: ConversationMessage = {"role": role, "content": blocks}
-    if meta:
-        message["meta"] = dict(meta)
-    return message
+    return _set_meta({"role": role, "content": blocks}, meta)
 
 
 def assistant_message(

{mycode_sdk-0.9.3 → mycode_sdk-0.9.5}/src/mycode/models.py

@@ -66,16 +66,26 @@ def resolve_model_metadata(
     *,
     provider: str,
     model: str,
-    **overrides: Any,
+    context_window: int | None = None,
+    max_output_tokens: int | None = None,
+    supports_reasoning: bool | None = None,
+    supports_image_input: bool | None = None,
+    supports_pdf_input: bool | None = None,
 ) -> ModelMetadata:
     """Return catalog metadata for ``(provider, model)`` with non-None overrides layered on top.
 
-    Override keys must match :class:`ModelMetadata` fields. Missing overrides
-    and an absent catalog entry both leave the corresponding fields at ``None``
-    so callers can apply their own fallback defaults.
+    Missing overrides and an absent catalog entry both leave the corresponding
+    fields at ``None`` so callers can apply their own fallback defaults.
     """
 
     base = lookup_model_metadata(provider_type=provider, model=model) or ModelMetadata(provider=provider, model=model)
+    overrides = {
+        "context_window": context_window,
+        "max_output_tokens": max_output_tokens,
+        "supports_reasoning": supports_reasoning,
+        "supports_image_input": supports_image_input,
+        "supports_pdf_input": supports_pdf_input,
+    }
     return replace(base, **{k: v for k, v in overrides.items() if v is not None})