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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mycode-sdk
-Version: 0.9.5
+Version: 0.10.0
 Summary: Lightweight Python SDK for building AI agents.
 Project-URL: Homepage, https://github.com/legibet/mycode
 Project-URL: Repository, https://github.com/legibet/mycode

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

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

{mycode_sdk-0.9.5 → mycode_sdk-0.10.0}/src/mycode/__init__.py

@@ -4,6 +4,7 @@ from importlib import metadata
 
 from mycode.agent import Agent, Event, PersistCallback, RunResult
 from mycode.attachments import Attachment
+from mycode.compact import NothingToCompactError
 from mycode.hooks import AfterToolHook, BeforeToolHook, HookResult, Hooks, ToolHookContext
 from mycode.messages import (
     ContentBlock,
@@ -45,6 +46,7 @@ __all__ = [
     "BeforeToolHook",
     "HookResult",
     "Hooks",
+    "NothingToCompactError",
     "PersistCallback",
     "RunResult",
     "SessionStore",

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

@@ -23,7 +23,9 @@ from mycode.attachments import AttachmentLike, build_attachment_blocks
 from mycode.compact import (
     COMPACT_SUMMARY_PROMPT,
     DEFAULT_COMPACT_THRESHOLD,
+    NothingToCompactError,
     build_compact_event,
+    has_compactable_history,
     should_compact,
 )
 from mycode.hooks import Hooks, ToolHookContext
@@ -455,6 +457,23 @@ class Agent:
             block["meta"] = {**meta, "duration_ms": duration_ms}
             return
 
+    async def _persist_message(
+        self,
+        message: ConversationMessage,
+        on_persist: PersistCallback | None,
+    ) -> None:
+        """Persist one message: caller callback first, then the SDK session store."""
+
+        if on_persist is not None:
+            # Callers may need to write related records before the SDK
+            # appends this message to its own session log.
+            await on_persist(message)
+        if self._store is None:
+            return
+        if not self._store.session_exists(self.session_id):
+            await self._store.create_session(self.session_id, cwd=self.cwd)
+        await self._store.append_message(self.session_id, message)
+
     # ------------------------------------------------------------------
     # Public entry points
     # ------------------------------------------------------------------
@@ -469,15 +488,7 @@ class Agent:
         """Run the full agent loop for one user message."""
 
         async def persist(message: ConversationMessage) -> None:
-            if on_persist is not None:
-                # Callers may need to write related records before the SDK
-                # appends this message to its own session log.
-                await on_persist(message)
-            if self._store is None:
-                return
-            if not self._store.session_exists(self.session_id):
-                await self._store.create_session(self.session_id, cwd=self.cwd)
-            await self._store.append_message(self.session_id, message)
+            await self._persist_message(message, on_persist)
 
         self._cancel_event.clear()
 
@@ -671,7 +682,7 @@ class Agent:
                 return
             if should_compact(total_tokens, self.context_window, self.compact_threshold):
                 try:
-                    await self._compact(adapter, persist)
+                    await self._compact(adapter, on_persist)
                     yield Event("compact", {})
                 except asyncio.CancelledError:
                     yield Event("error", {"message": "cancelled"})
@@ -719,12 +730,47 @@ class Agent:
     # Context compaction
     # ------------------------------------------------------------------
 
+    async def acompact(
+        self,
+        *,
+        on_persist: PersistCallback | None = None,
+    ) -> ConversationMessage:
+        """Compact the conversation now and return the persisted compact marker.
+
+        Raises :class:`NothingToCompactError` when no new context follows the
+        latest compact marker, and :class:`asyncio.CancelledError` when
+        :meth:`cancel` stops the summary request.
+        """
+
+        self._cancel_event.clear()
+        adapter = get_provider_adapter(self.provider)
+        return await self._compact(adapter, on_persist)
+
+    def compact(
+        self,
+        *,
+        on_persist: PersistCallback | None = None,
+    ) -> ConversationMessage:
+        """Compact the conversation synchronously; see :meth:`acompact`."""
+
+        try:
+            asyncio.get_running_loop()
+        except RuntimeError:
+            pass
+        else:
+            raise RuntimeError("Agent.compact() cannot run inside an active event loop; use Agent.acompact() instead")
+
+        return asyncio.run(self.acompact(on_persist=on_persist))
+
     async def _compact(
         self,
         adapter: ProviderAdapter,
-        persist: PersistCallback,
-    ) -> None:
-        """Ask the provider for a summary, persist the compact event, append it."""
+        on_persist: PersistCallback | None,
+    ) -> ConversationMessage:
+        """Ask the provider for a summary, persist and append the compact marker."""
+
+        if not has_compactable_history(self.messages):
+            raise NothingToCompactError("nothing to compact")
 
         request = self._build_request(
             tools=[],
@@ -746,6 +792,9 @@ class Agent:
         if not summary_text:
             raise ValueError("compaction produced empty summary")
 
+        if self._cancel_event.is_set():
+            raise asyncio.CancelledError
+
         summary_total_tokens = (summary_message.get("meta") or {}).get("total_tokens")
         compact_event = build_compact_event(
             summary_text,
@@ -754,5 +803,6 @@ class Agent:
             total_tokens=summary_total_tokens,
         )
 
-        await persist(compact_event)
+        await self._persist_message(compact_event, on_persist)
         self.messages.append(compact_event)
+        return compact_event

{mycode_sdk-0.9.5 → mycode_sdk-0.10.0}/src/mycode/compact.py

@@ -8,6 +8,11 @@ from mycode.messages import ConversationMessage, build_message, text_block
 
 DEFAULT_COMPACT_THRESHOLD = 0.8
 
+
+class NothingToCompactError(ValueError):
+    """Raised when there is no new context to compact past the latest marker."""
+
+
 COMPACT_SUMMARY_PROMPT = """\
 Summarize this conversation to create a continuation document. \
 This summary will replace the full conversation history, so it must \
@@ -65,6 +70,20 @@ def should_compact(
     return last_total_tokens >= context_window * threshold
 
 
+def has_compactable_history(messages: list[ConversationMessage]) -> bool:
+    """True when at least one non-empty user/assistant message follows the latest compact marker."""
+
+    last_compact = -1
+    for i, message in enumerate(messages):
+        if message.get("role") == "compact":
+            last_compact = i
+
+    return any(
+        message.get("role") in ("user", "assistant") and message.get("content")
+        for message in messages[last_compact + 1 :]
+    )
+
+
 def build_compact_event(
     summary_text: str,
     *,

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

@@ -165,8 +165,8 @@ def flatten_message_text(message: ConversationMessage, *, include_thinking: bool
             continue
         raw_meta = block.get("meta")
         meta = raw_meta if isinstance(raw_meta, dict) else {}
-        # Attached file snapshots should not become session titles or history labels.
-        if meta.get("attachment"):
+        # Local payload blocks should not become session titles or history labels.
+        if meta.get("attachment") or meta.get("skill_snapshot"):
             continue
         btype = block.get("type")
         if btype == "text" or (include_thinking and btype == "thinking"):

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

@@ -2,9 +2,9 @@
 
 from __future__ import annotations
 
-import functools
 import json
 from dataclasses import dataclass, replace
+from functools import cache
 from pathlib import Path
 from typing import Any
 
@@ -30,14 +30,14 @@ class ModelMetadata:
     supports_pdf_input: bool | None = None
 
 
-@functools.cache
+@cache
 def load_models_catalog() -> dict[str, Any] | None:
     """Load the bundled model catalog from disk once per process."""
 
     try:
         data = json.loads(_MODELS_CATALOG_PATH.read_text(encoding="utf-8"))
-    except Exception:
-        data = None
+    except (OSError, UnicodeError, json.JSONDecodeError):
+        return None
     return data if isinstance(data, dict) else None
 
 

{mycode_sdk-0.9.5 → mycode_sdk-0.10.0}/src/mycode/providers/anthropic_like.py

@@ -111,8 +111,9 @@ class AnthropicLikeAdapter(ProviderAdapter):
             if not isinstance(content, list):
                 return
 
-            blocks: list[dict[str, Any]] = [cast(dict[str, Any], block) for block in content if isinstance(block, dict)]
-            for block in reversed(blocks):
+            for block in reversed(content):
+                if not isinstance(block, dict):
+                    continue
                 block_type = str(block.get("type") or "")
                 if block_type not in {"text", "image", "document", "tool_result"}:
                     continue

{mycode_sdk-0.9.5 → mycode_sdk-0.10.0}/src/mycode/providers/base.py

@@ -88,7 +88,7 @@ def parse_tool_call_input(raw_arguments: str) -> tuple[dict[str, Any], dict[str,
     try:
         parsed = json.loads(raw_arguments)
     except json.JSONDecodeError:
-        parsed = None
+        return {}, {"raw_arguments": raw_arguments}
     if isinstance(parsed, dict):
         return parsed, {}
     return {}, {"raw_arguments": raw_arguments}

{mycode_sdk-0.9.5 → mycode_sdk-0.10.0}/src/mycode/providers/gemini.py

@@ -84,7 +84,7 @@ class GoogleGeminiAdapter(ProviderAdapter):
                 finish_message = getattr(candidate, "finish_message", None) or finish_message
 
                 for part in getattr(getattr(candidate, "content", None), "parts", None) or []:
-                    for event in self._consume_part(blocks, part):
+                    if event := self._consume_part(blocks, part):
                         yield event
         except APIError as exc:
             raise ValueError(str(exc)) from exc
@@ -223,7 +223,7 @@ class GoogleGeminiAdapter(ProviderAdapter):
         return contents
 
     def _build_config(self, request: ProviderRequest) -> types.GenerateContentConfig:
-        tools: list[types.Tool | Any] | None = None
+        tools: list[types.Tool] | None = None
         if request.tools:
             tools = [
                 types.Tool(
@@ -262,7 +262,7 @@ class GoogleGeminiAdapter(ProviderAdapter):
             thinking_config=thinking_config,
         )
 
-    def _consume_part(self, blocks: list[dict[str, Any]], part: Any) -> list[ProviderStreamEvent]:
+    def _consume_part(self, blocks: list[dict[str, Any]], part: Any) -> ProviderStreamEvent | None:
         native_part = _to_json(part) or {}
         if native_part.get("thought") is False:
             native_part.pop("thought", None)
@@ -278,12 +278,12 @@ class GoogleGeminiAdapter(ProviderAdapter):
                     meta={"native": {"part": native_part}},
                 )
             )
-            return []
+            return None
 
         text = getattr(part, "text", None)
         if text is None or text == "":
             if not native_part.get("thought_signature"):
-                return []
+                return None
 
             # Gemini may put the final thought signature into an empty-text part.
             # Keep it as a separate empty block so replay preserves the original
@@ -291,7 +291,7 @@ class GoogleGeminiAdapter(ProviderAdapter):
             part_meta = {"native": {"part": native_part}}
             is_thought = getattr(part, "thought", False)
             blocks.append(thinking_block("", meta=part_meta) if is_thought else text_block("", meta=part_meta))
-            return []
+            return None
 
         is_thought = bool(getattr(part, "thought", False))
         event = ProviderStreamEvent("thinking_delta" if is_thought else "text_delta", {"text": str(text)})
@@ -310,10 +310,10 @@ class GoogleGeminiAdapter(ProviderAdapter):
                     last_part["text"] = f"{last_part.get('text') or ''}{text}"
                     if current_signature and not last_signature:
                         last_part["thought_signature"] = current_signature
-                    return [event]
+                    return event
 
         part_meta = {"native": {"part": native_part}}
         blocks.append(
             thinking_block(str(text), meta=part_meta) if is_thought else text_block(str(text), meta=part_meta)
         )
-        return [event]
+        return event

{mycode_sdk-0.9.5 → mycode_sdk-0.10.0}/src/mycode/providers/openai_chat.py

@@ -9,7 +9,7 @@ from typing import Any, override
 
 from openai import APIError, AsyncOpenAI
 
-from mycode.messages import assistant_message, text_block, thinking_block, tool_use_block
+from mycode.messages import ConversationMessage, assistant_message, text_block, thinking_block, tool_use_block
 from mycode.providers.base import (
     DEFAULT_REQUEST_TIMEOUT,
     ProviderAdapter,
@@ -174,7 +174,7 @@ class OpenAIChatAdapter(ProviderAdapter):
             },
         }
 
-    def _serialize_message(self, message: dict[str, Any]) -> list[dict[str, Any]]:
+    def _serialize_message(self, message: ConversationMessage) -> list[dict[str, Any]]:
         """Convert one canonical message into Chat Completions wire messages."""
 
         role = str(message.get("role") or "user")

{mycode_sdk-0.9.5 → mycode_sdk-0.10.0}/src/mycode/providers/openai_responses.py

@@ -102,9 +102,8 @@ class OpenAIResponsesAdapter(ProviderAdapter):
             raise ValueError(str(exc)) from exc
 
     def _build_request_payload(self, request: ProviderRequest) -> dict[str, Any]:
-        prepared_messages = self.prepare_messages(request)
         input_items: list[dict[str, Any]] = []
-        for message in prepared_messages:
+        for message in self.prepare_messages(request):
             role = message.get("role")
             if role == "user":
                 input_items.extend(self._serialize_user_message(message))
@@ -262,7 +261,7 @@ class OpenAIResponsesAdapter(ProviderAdapter):
         response: Any,
         *,
         output_items: list[Any] | None = None,
-    ) -> dict[str, Any]:
+    ) -> ConversationMessage:
         raw_output = output_items if output_items is not None else (getattr(response, "output", None) or [])
         dumped_output_items = dump_model(raw_output)
         blocks: list[dict[str, Any]] = []
@@ -357,8 +356,8 @@ def _normalize_strict_schema(schema: Any) -> None:
 
     properties = schema.get("properties")
     if isinstance(properties, dict):
-        required_names = {str(name) for name in schema.get("required", []) if isinstance(name, str)}
-        for name, property_schema in list(properties.items()):
+        required_names = {name for name in schema.get("required", []) if isinstance(name, str)}
+        for name, property_schema in properties.items():
             _normalize_strict_schema(property_schema)
             if name not in required_names:
                 properties[name] = _nullable_schema(property_schema)

{mycode_sdk-0.9.5 → mycode_sdk-0.10.0}/src/mycode/session.py

@@ -139,7 +139,7 @@ class SessionStore:
         meta = _project_meta(meta)
         self.meta_path(session_id).write_text(json.dumps(meta, indent=2, ensure_ascii=False), encoding="utf-8")
         index = self._read_index()
-        index[session_id] = dict(meta)
+        index[session_id] = meta
         self._write_index(index)
 
     def _read_index(self) -> SessionIndex:
@@ -161,7 +161,7 @@ class SessionStore:
                 continue
             meta = self._read_meta(entry.name)
             if meta is not None:
-                index[entry.name] = dict(meta)
+                index[entry.name] = meta
         self._write_index(index)
         return index
 
@@ -246,9 +246,7 @@ class SessionStore:
 
     async def delete_session(self, session_id: str) -> None:
         def delete() -> None:
-            sdir = self.session_dir(session_id)
-            if sdir.exists():
-                shutil.rmtree(sdir, ignore_errors=True)
+            shutil.rmtree(self.session_dir(session_id), ignore_errors=True)
             index = self._read_index()
             index.pop(session_id, None)
             self._write_index(index)
@@ -285,8 +283,7 @@ class SessionStore:
 
         def append() -> None:
             with self.messages_path(session_id).open("a", encoding="utf-8") as handle:
-                handle.write(json.dumps(message, ensure_ascii=False))
-                handle.write("\n")
+                handle.write(json.dumps(message, ensure_ascii=False) + "\n")
 
             meta = self._read_meta(session_id)
             if meta is None:

{mycode_sdk-0.9.5 → mycode_sdk-0.10.0}/src/mycode/tools.py

@@ -129,12 +129,6 @@ class ToolContext:
 
         return self.executor.execute(name, args, self)
 
-    def track_proc(self, proc: subprocess.Popen[str]) -> None:
-        self.executor.track_proc(proc)
-
-    def untrack_proc(self, proc: subprocess.Popen[str]) -> None:
-        self.executor.untrack_proc(proc)
-
 
 # ---------------------------------------------------------------------------
 # ToolExecutor
@@ -202,6 +196,9 @@ class ToolExecutor:
 # agents may run concurrently in the same process); this module-level set is
 # a shutdown-time safety net exposed as ``cancel_all_tools``.
 
+_ACTIVE_PROCS: set[subprocess.Popen[str]] = set()
+_ACTIVE_PROCS_LOCK = threading.Lock()
+
 
 def cancel_all_tools() -> None:
     """Terminate every running bash subprocess in the current process."""
@@ -213,10 +210,6 @@ def cancel_all_tools() -> None:
         _kill_proc_tree(proc)
 
 
-_ACTIVE_PROCS: set[subprocess.Popen[str]] = set()
-_ACTIVE_PROCS_LOCK = threading.Lock()
-
-
 def _kill_proc_tree(proc: subprocess.Popen[str]) -> None:
     try:
         if os.name == "posix":
@@ -479,8 +472,8 @@ def read_tool(
             ],
         )
 
-    start_line = offset if isinstance(offset, int) and offset > 0 else 1
-    line_limit = limit if isinstance(limit, int) and limit > 0 else DEFAULT_MAX_LINES
+    start_line = offset if offset is not None and offset > 0 else 1
+    line_limit = limit if limit is not None and limit > 0 else DEFAULT_MAX_LINES
     lines: list[str] = []
     total_lines = 0
     next_offset: int | None = None
@@ -764,7 +757,7 @@ def bash_tool(ctx: ToolContext, command: str, timeout: int | None = None) -> Too
     again by :func:`truncate_text` to the display limits.
     """
 
-    timeout_seconds = int(timeout) if isinstance(timeout, int) and timeout > 0 else BASH_TIMEOUT_SECONDS
+    timeout_seconds = timeout if timeout is not None and timeout > 0 else BASH_TIMEOUT_SECONDS
 
     proc: subprocess.Popen[str] | None = None
     log_path = ctx.tool_output_dir / f"bash-{ctx.tool_call_id or 'call'}.log"
@@ -787,7 +780,7 @@ def bash_tool(ctx: ToolContext, command: str, timeout: int | None = None) -> Too
             bufsize=1,
             start_new_session=os.name == "posix",
         )
-        ctx.track_proc(proc)
+        ctx.executor.track_proc(proc)
 
         stdout = cast(TextIO, proc.stdout)
         output_queue: queue.Queue[str | None] = queue.Queue()
@@ -903,7 +896,7 @@ def bash_tool(ctx: ToolContext, command: str, timeout: int | None = None) -> Too
             with suppress(Exception):
                 log_file.close()
         if proc is not None:
-            ctx.untrack_proc(proc)
+            ctx.executor.untrack_proc(proc)
             if proc.poll() is None:
                 _kill_proc_tree(proc)