codereader 0.7.0__tar.gz → 0.7.2__tar.gz

{codereader-0.7.0 → codereader-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codereader
-Version: 0.7.0
+Version: 0.7.2
 Summary: Local code readability grader using LLMs
 License: GPLv3
 License-File: LICENSE
@@ -159,6 +159,8 @@ models:
     type: ollama
     model: qwen2.5-coder
     weight: 1.0
+    runner_config:
+      no_think: true # to disable deep thinking to shorten
 
 settings:
   log_path: readability_log.txt

{codereader-0.7.0 → codereader-0.7.2}/README.md

@@ -135,6 +135,8 @@ models:
     type: ollama
     model: qwen2.5-coder
     weight: 1.0
+    runner_config:
+      no_think: true # to disable deep thinking to shorten
 
 settings:
   log_path: readability_log.txt

{codereader-0.7.0 → codereader-0.7.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codereader"
-version = "0.7.0"
+version = "0.7.2"
 description = "Local code readability grader using LLMs"
 authors = [
     {name = "Jason Liu",email = "Liujason2003@gmail.com"}

{codereader-0.7.0 → codereader-0.7.2}/src/codereader/runners/runner.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import asyncio
 import json
 import re
+import os
 
 from dataclasses import dataclass
 from typing import Any, Dict, List, Optional, Protocol
@@ -89,13 +90,15 @@ async def run_subprocess(
     loop = asyncio.get_running_loop()
     start = loop.time()
 
+    clean_env = {**os.environ, "TERM": "dumb", "NO_COLOR": "1", **(env or {})}
+
     try:
         proc = await asyncio.create_subprocess_exec(
             *cmd,
             stdin=asyncio.subprocess.PIPE if stdin_text is not None else None,
             stdout=asyncio.subprocess.PIPE,
             stderr=asyncio.subprocess.PIPE,
-            env=env,
+            env=clean_env,
         )
     except FileNotFoundError as e:
         dur = loop.time() - start
@@ -184,6 +187,16 @@ def _try_parse_from(text: str, start: int) -> Optional[Dict[str, Any]]:
     return None
 
 
+def clean_output(text: str):
+    text = re.sub(r"\x1b\[[0-9;]*[A-Za-z]", "", text)
+
+    text = re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL)
+
+    text = re.sub(r"Thinking\.\.\..*?\.\.\.done thinking\.", "", text, flags=re.DOTALL)
+
+    return text
+
+
 def extract_json_object(text: str) -> Optional[Dict[str, Any]]:
     """
     Robustly extract the first valid JSON object from model output.
@@ -199,7 +212,7 @@ def extract_json_object(text: str) -> Optional[Dict[str, Any]]:
     - Nested objects: {"subscores": {"a": 1}}
     """
 
-    text = re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL)
+    text = clean_output(text)
 
     fenced = re.search(r"```(?:json)?\s*(\{.*?\})\s*```", text, re.DOTALL)
     if fenced: