mycode-sdk 0.7.1__tar.gz → 0.7.3__tar.gz

{mycode_sdk-0.7.1 → mycode_sdk-0.7.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mycode-sdk
-Version: 0.7.1
+Version: 0.7.3
 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.7.1 → mycode_sdk-0.7.3}/pyproject.toml

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

{mycode_sdk-0.7.1 → mycode_sdk-0.7.3}/src/mycode/agent.py

@@ -11,6 +11,7 @@ import asyncio
 import logging
 import os
 import tempfile
+import time
 from collections.abc import AsyncIterator, Awaitable, Callable, Sequence
 from dataclasses import dataclass, field
 from pathlib import Path
@@ -493,6 +494,8 @@ class Agent:
                 return
 
             assistant_message: ConversationMessage | None = None
+            thinking_started_at: float | None = None
+            thinking_duration_ms: int | None = None
             request = ProviderRequest(
                 provider=self.provider,
                 model=self.model,
@@ -518,12 +521,17 @@ class Agent:
                     if provider_event.type == "thinking_delta":
                         delta_text = str(provider_event.data.get("text") or "")
                         if delta_text:
+                            if thinking_started_at is None:
+                                thinking_started_at = time.monotonic()
                             yield Event("reasoning", {"delta": delta_text})
                         continue
 
                     if provider_event.type == "text_delta":
                         delta_text = str(provider_event.data.get("text") or "")
                         if delta_text:
+                            if thinking_started_at is not None and thinking_duration_ms is None:
+                                thinking_duration_ms = max(0, int((time.monotonic() - thinking_started_at) * 1000))
+                                yield Event("reasoning_done", {"duration_ms": thinking_duration_ms})
                             yield Event("text", {"delta": delta_text})
                         continue
 
@@ -533,6 +541,10 @@ class Agent:
                     if provider_event.type != "message_done":
                         continue
 
+                    if thinking_started_at is not None and thinking_duration_ms is None:
+                        thinking_duration_ms = max(0, int((time.monotonic() - thinking_started_at) * 1000))
+                        yield Event("reasoning_done", {"duration_ms": thinking_duration_ms})
+
                     message = provider_event.data.get("message")
                     if isinstance(message, dict):
                         assistant_message = message
@@ -549,6 +561,15 @@ class Agent:
                 yield Event("error", {"message": "provider produced no assistant message"})
                 return
 
+            if thinking_duration_ms is not None:
+                for block in reversed(assistant_message.get("content") or []):
+                    if not isinstance(block, dict) or block.get("type") != "thinking":
+                        continue
+                    raw_meta = block.get("meta")
+                    meta = cast(dict[str, Any], raw_meta) if isinstance(raw_meta, dict) else {}
+                    block["meta"] = {**meta, "duration_ms": thinking_duration_ms}
+                    break
+
             self.messages.append(assistant_message)
             await persist(assistant_message)
 

{mycode_sdk-0.7.1 → mycode_sdk-0.7.3}/src/mycode/messages.py

@@ -15,6 +15,8 @@ Metadata contract:
   `provider`, `model`, `provider_message_id`, `stop_reason`, `usage`
 - provider-specific assistant message extras live under `meta.native`
 - provider-specific block replay hints live under `block.meta.native`
+- local display metadata, such as `block.meta.duration_ms`, is never sent
+  upstream; provider adapters must explicitly project only supported fields
 """
 
 from __future__ import annotations

{mycode_sdk-0.7.1 → mycode_sdk-0.7.3}/src/mycode/providers/base.py

@@ -205,7 +205,7 @@ def repair_messages_for_replay(
                 block_type = raw_block.get("type")
                 if block_type in {"text", "thinking"}:
                     text = str(raw_block.get("text") or "")
-                    if text:
+                    if text or get_native_meta(raw_block):
                         content.append(dict(raw_block))
                     continue
 

{mycode_sdk-0.7.1 → mycode_sdk-0.7.3}/src/mycode/providers/openai_chat.py

@@ -79,9 +79,10 @@ class OpenAIChatAdapter(ProviderAdapter):
 
                 delta = choice.delta
                 reasoning_delta, reasoning_meta_update = self._extract_reasoning_delta(delta)
+                if reasoning_meta_update:
+                    thinking_native_meta.update(reasoning_meta_update)
                 if reasoning_delta:
                     thinking_parts.append(reasoning_delta)
-                    thinking_native_meta.update(reasoning_meta_update)
                     yield ProviderStreamEvent("thinking_delta", {"text": reasoning_delta})
 
                 if delta.content:
@@ -104,7 +105,7 @@ class OpenAIChatAdapter(ProviderAdapter):
             raise ValueError(str(exc)) from exc
 
         blocks = []
-        if thinking_parts:
+        if thinking_parts or thinking_native_meta:
             blocks.append(
                 thinking_block(
                     "".join(thinking_parts),
@@ -286,32 +287,52 @@ class OpenAIChatAdapter(ProviderAdapter):
         reasoning_field = str(native_meta.get("reasoning_field") or "")
         if reasoning_field == "reasoning_details":
             return {"reasoning_details": native_meta.get("reasoning_details") or []}
+        if reasoning_field == "reasoning":
+            return {"reasoning": thinking_text or None}
+        if reasoning_field == "reasoning_content":
+            return {"reasoning_content": thinking_text or None}
         return {"reasoning_content": thinking_text} if thinking_text else {}
 
     def _extract_reasoning_delta(self, delta: Any) -> tuple[str, dict[str, Any]]:
         # Third-party providers surface reasoning through non-standard extras.
         # We check both the delta root and model_extra to cover both patterns.
-        # Known fields: reasoning_content (Moonshot/MiniMax chat), reasoning_details (some others).
+        # Known fields: reasoning, reasoning_content, reasoning_details.
         for source in (delta, getattr(delta, "model_extra", None) or {}):
             if isinstance(source, dict):
+                has_reasoning = "reasoning" in source
+                reasoning = source.get("reasoning")
+                has_reasoning_content = "reasoning_content" in source
                 reasoning_content = source.get("reasoning_content")
+                has_reasoning_details = "reasoning_details" in source
                 reasoning_details = source.get("reasoning_details")
             else:
+                has_reasoning = hasattr(source, "reasoning")
+                reasoning = getattr(source, "reasoning", None)
+                has_reasoning_content = hasattr(source, "reasoning_content")
                 reasoning_content = getattr(source, "reasoning_content", None)
+                has_reasoning_details = hasattr(source, "reasoning_details")
                 reasoning_details = getattr(source, "reasoning_details", None)
 
-            if isinstance(reasoning_content, str) and reasoning_content:
-                return reasoning_content, {"reasoning_field": "reasoning_content"}
+            if has_reasoning:
+                return (
+                    reasoning if isinstance(reasoning, str) else "",
+                    {"reasoning_field": "reasoning"},
+                )
 
-            if isinstance(reasoning_details, list) and reasoning_details:
+            if has_reasoning_content:
+                return (
+                    reasoning_content if isinstance(reasoning_content, str) else "",
+                    {"reasoning_field": "reasoning_content"},
+                )
+
+            if has_reasoning_details and isinstance(reasoning_details, list):
                 reasoning_text = "".join(
                     str(item.get("text") or "") for item in reasoning_details if isinstance(item, dict)
                 )
-                if reasoning_text:
-                    return reasoning_text, {
-                        "reasoning_field": "reasoning_details",
-                        "reasoning_details": reasoning_details,
-                    }
+                return reasoning_text, {
+                    "reasoning_field": "reasoning_details",
+                    "reasoning_details": reasoning_details,
+                }
 
         return "", {}