codereader 0.7.2__tar.gz → 0.7.4__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codereader
-Version: 0.7.2
+Version: 0.7.4
 Summary: Local code readability grader using LLMs
 License: GPLv3
 License-File: LICENSE

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

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

{codereader-0.7.2 → codereader-0.7.4}/src/codereader/runners/ollama_runner.py

@@ -55,7 +55,7 @@ class OllamaRunner:
         """
         global _SERVER_CHECK
 
-        if _SERVER_CHECK is not None:
+        if _SERVER_CHECK is not None and _SERVER_CHECK.ok:
             return _SERVER_CHECK
 
         cmd = [self.ollama_bin, "list"]
@@ -66,7 +66,7 @@ class OllamaRunner:
             return res
 
         if _looks_like_ollama_server_down(res.stderr, res.stdout, res.error):
-            _SERVER_CHECK = RunnerResult(
+            return RunnerResult(
                 ok=False,
                 stdout=res.stdout,
                 stderr=res.stderr,
@@ -78,7 +78,6 @@ class OllamaRunner:
                     "Then retry your command."
                 ),
             )
-            return _SERVER_CHECK
 
         _SERVER_CHECK = res
         return res
@@ -118,10 +117,11 @@ class OllamaRunner:
     ) -> GradeResult:
         prompt = format_prompt(self.grade_prompt_template, tags, rules, language, code)
 
-        if self.no_think:
-            prompt = "/no_think\n" + prompt
-
-        cmd = [self.ollama_bin, "run", self.model]
+        cmd = (
+            [self.ollama_bin, "run"]
+            + (["--think=false"] if self.no_think else ["--hidethinking"])
+            + [self.model]
+        )
 
         res = await run_subprocess(cmd, stdin_text=prompt, timeout_s=timeout_s)
         print(f"[DEBUG {self.name}] raw output: {repr(res.stdout[:800])}", flush=True)