nanocode-cli 0.2.2__tar.gz → 0.2.3__tar.gz

{nanocode_cli-0.2.2/nanocode_cli.egg-info → nanocode_cli-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nanocode-cli
-Version: 0.2.2
+Version: 0.2.3
 Summary: A lightweight terminal-based AI coding assistant
 Author-email: hit9 <hit9@icloud.com>
 License-Expression: BSD-3-Clause

{nanocode_cli-0.2.2 → nanocode_cli-0.2.3}/nanocode.py

@@ -41,7 +41,7 @@ from prompt_toolkit.patch_stdout import patch_stdout
 
 JsonValue: TypeAlias = Any
 Json: TypeAlias = dict[str, JsonValue]
-__version__ = "0.2.2"
+__version__ = "0.2.3"
 
 
 class Error(Exception): ...
@@ -1995,6 +1995,8 @@ class ModelClient:
 
         self._record_usage(_json_dict(result.get("usage") if isinstance(result, dict) else None))
         content = self._message_content(result)
+        if content is None:
+            return self._invalid_model_response(self._format_missing_message_content(result))
         return self._parse_model_content(content)
 
     def _write_debug_prompt(self, *, activity: str, messages: list[Json]) -> str:
@@ -2024,13 +2026,9 @@ class ModelClient:
 
     def _parse_model_content(self, content: str) -> Json:
         text = content.strip()
-        if text.startswith("```"):
-            lines = text.splitlines()
-            if lines and lines[0].startswith("```"):
-                lines = lines[1:]
-            if lines and lines[-1].strip() == "```":
-                lines = lines[:-1]
-            text = "\n".join(lines).strip()
+        text = self._strip_leaked_think_tags(text)
+        text = self._strip_json_fence(text)
+        text = self._strip_leaked_think_tags(text)
         try:
             value = json.loads(text)
         except json.JSONDecodeError:
@@ -2039,15 +2037,49 @@ class ModelClient:
             return self._invalid_model_response(content)
         return value
 
+    def _strip_json_fence(self, text: str) -> str:
+        if not text.startswith("```"):
+            return text
+        lines = text.splitlines()
+        if lines and lines[0].startswith("```"):
+            lines = lines[1:]
+        if lines and lines[-1].strip() == "```":
+            lines = lines[:-1]
+        return "\n".join(lines).strip()
+
+    def _strip_leaked_think_tags(self, text: str) -> str:
+        text = text.strip()
+        while text.startswith("</think>"):
+            text = text[len("</think>") :].lstrip()
+        while text.startswith("<think>"):
+            end = text.find("</think>")
+            if end < 0:
+                return text
+            text = text[end + len("</think>") :].lstrip()
+            while text.startswith("</think>"):
+                text = text[len("</think>") :].lstrip()
+        return text
+
     def _invalid_model_response(self, content: str) -> Json:
+        guidance = ""
+        if self._looks_like_native_tool_call(content):
+            guidance = (
+                " Native tool_call syntax is not supported; return one JSON object with tool_calls entries like "
+                '{"name":"Read","intention":"...","args":["nanocode.py","0","100"]}.'
+            )
         return {
             "goal_reached": False,
             "tool_calls": None,
             "message_to_user": None,
             "_format_error": "Invalid model output: expected one JSON object matching the Output JSON schema. Return strict JSON only. Bad output: "
-            + _shorten(content),
+            + _shorten(content)
+            + guidance,
         }
 
+    def _looks_like_native_tool_call(self, content: str) -> bool:
+        text = self._strip_leaked_think_tags(content.strip())
+        return text.startswith("<tool_call>")
+
     def _chat_completions_url(self) -> str:
         url = self.session.api_url.rstrip("/")
         if url.endswith("/chat/completions"):
@@ -2061,7 +2093,7 @@ class ModelClient:
             return {"reasoning": {"effort": self.session.reasoning_effort}}
         return {}
 
-    def _message_content(self, result: JsonValue) -> str:
+    def _message_content(self, result: JsonValue) -> str | None:
         data = _json_dict(result)
         choices = _json_list(data.get("choices"))
         if not choices:
@@ -2069,9 +2101,18 @@ class ModelClient:
         message = _json_dict(_json_dict(choices[0]).get("message"))
         content = message.get("content")
         if not isinstance(content, str):
-            raise LLMError("API response missing message content")
+            return None
         return content
 
+    def _format_missing_message_content(self, result: JsonValue) -> str:
+        choice = _json_dict(_json_list(_json_dict(result).get("choices"))[0])
+        message = _json_dict(choice.get("message"))
+        details: Json = {
+            "finish_reason": choice.get("finish_reason"),
+            "message_keys": sorted(str(key) for key in message.keys()),
+        }
+        return "API response missing message content: " + json.dumps(details, ensure_ascii=False)
+
     def _record_usage(self, usage: Json) -> None:
         prompt_tokens = _json_int(usage.get("prompt_tokens"))
         completion_tokens = _json_int(usage.get("completion_tokens"))
@@ -2627,6 +2668,8 @@ class ConversationCompactor:
 
 @final
 class Agent:
+    MAX_CONSECUTIVE_FORMAT_ERRORS: ClassVar[int] = 3
+
     def __init__(self, session: Session):
         self.session = session
         self.prompt_builder = PromptBuilder(session)
@@ -2671,18 +2714,48 @@ class Agent:
         self.session.current.goal_reached = False
         self.maybe_auto_compact()
         self.session.append_conversation(UserMessage(content=user_input))
+        consecutive_format_errors = 0
 
         for _ in range(self.session.max_agent_steps):
             response = self.step()
             format_error = _json_str(response.get("_format_error"))
             if format_error:
+                consecutive_format_errors += 1
                 self.latest_tool_call_results = format_error
+                if consecutive_format_errors >= self.MAX_CONSECUTIVE_FORMAT_ERRORS:
+                    self._report_gate(
+                        on_message,
+                        "Stopped: model returned invalid output "
+                        + str(self.MAX_CONSECUTIVE_FORMAT_ERRORS)
+                        + " times in a row.",
+                        "Format_Gate: stopped after "
+                        + str(self.MAX_CONSECUTIVE_FORMAT_ERRORS)
+                        + " consecutive invalid model outputs. "
+                        + _shorten(format_error, 180),
+                    )
+                    raise LLMError(
+                        "model returned invalid output "
+                        + str(self.MAX_CONSECUTIVE_FORMAT_ERRORS)
+                        + " times in a row: "
+                        + _shorten(format_error, 300)
+                    )
+                self._report_gate(
+                    on_message,
+                    "Retrying: model returned invalid output.",
+                    "Format_Gate: retrying model response. " + _shorten(format_error, 180),
+                )
                 continue
+            consecutive_format_errors = 0
             tool_calls = _json_list(response.get("tool_calls"))
             summary_gate = self._format_tool_summary_gate(tool_calls)
             if summary_gate:
                 self.state_updater.latest_report = ""
                 self.latest_tool_call_results = summary_gate
+                self._report_gate(
+                    on_message,
+                    "Retrying: model needs to summarize the latest tool results.",
+                    self._compact_gate_report(summary_gate),
+                )
                 continue
             self.apply_response(response)
             if on_message is not None and self.state_updater.latest_report:
@@ -2705,13 +2778,35 @@ class Agent:
             if self.session.current.verification.status == VerificationStatus.REQUIRED:
                 self.session.current.goal_reached = False
                 self.latest_tool_call_results = self._format_verification_gate()
+                self._report_gate(
+                    on_message,
+                    "Retrying: verification is required before completion.",
+                    "Verification_Gate: retrying until verification is passed or blocked.",
+                )
                 continue
             if not self.session.current.goal_reached:
                 self.latest_tool_call_results = self._format_continuation_hint()
+                self._report_gate(
+                    on_message,
+                    "Continuing: goal is not complete yet.",
+                    "Continuation_Gate: goal not reached; retrying next useful action.",
+                )
                 continue
             return response
         raise LLMError("agent step limit reached")
 
+    def _report_gate(self, on_message: MessageCallback | None, message: str, debug_message: str) -> None:
+        if on_message is not None:
+            on_message(debug_message if self.session.debug else message)
+
+    def _compact_gate_report(self, gate: str) -> str:
+        lines = gate.splitlines()
+        headline = lines[0] if lines else "Gate"
+        details = [line for line in lines[1:] if line.startswith("- ")]
+        if details:
+            return headline + ": " + _shorten("; ".join(details[:3]), 220)
+        return headline
+
     def step(self) -> Json:
         response = self.request(self.build_system_prompt(), self.build_user_prompt(), activity="main")
         self.state_updater.apply_tool_call_summaries(response)

{nanocode_cli-0.2.2 → nanocode_cli-0.2.3/nanocode_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nanocode-cli
-Version: 0.2.2
+Version: 0.2.3
 Summary: A lightweight terminal-based AI coding assistant
 Author-email: hit9 <hit9@icloud.com>
 License-Expression: BSD-3-Clause

{nanocode_cli-0.2.2 → nanocode_cli-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "nanocode-cli"
-version = "0.2.2"
+version = "0.2.3"
 description = "A lightweight terminal-based AI coding assistant"
 readme = "README.md"
 requires-python = ">=3.11"