code-puppy 0.0.87__tar.gz → 0.0.89__tar.gz

{code_puppy-0.0.87 → code_puppy-0.0.89}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.87
+Version: 0.0.89
 Summary: Code generation agent
 Author: Michael Pfaffenberger
 License: MIT

{code_puppy-0.0.87 → code_puppy-0.0.89}/code_puppy/models.json

@@ -11,7 +11,7 @@
       "url": "https://api.cerebras.ai/v1",
       "api_key": "$CEREBRAS_API_KEY"
     },
-    "context_length": 10000
+    "context_length": 131072
   },
   "Cerebras-Qwen3-235b-a22b-instruct-2507": {
     "type": "custom_openai",

{code_puppy-0.0.87 → code_puppy-0.0.89}/code_puppy/tools/command_runner.py

@@ -57,9 +57,6 @@ def _kill_process_group(proc: subprocess.Popen) -> None:
                     proc.kill()
                 except Exception:
                     pass
-            # On Windows, restore terminal state after killing process
-            if hasattr(signal, "CTRL_C_EVENT"):
-                os.kill(proc.pid, signal.CTRL_C_EVENT)
             return
 
         # POSIX

{code_puppy-0.0.87 → code_puppy-0.0.89}/code_puppy/tools/file_operations.py

@@ -25,6 +25,7 @@ class ListedFile(BaseModel):
 
 class ListFileOutput(BaseModel):
     files: List[ListedFile]
+    error: str | None = None
 
 
 def _list_files(
@@ -270,7 +271,7 @@ def _grep(context: RunContext, search_string: str, directory: str = ".") -> Grep
                                 **{
                                     "file_path": file_path,
                                     "line_number": line_number,
-                                    "line_content": line_content.rstrip("\n\r"),
+                                    "line_content": line_content.rstrip("\n\r")[2048:],
                                 }
                             )
                             matches.append(match_info)
@@ -311,7 +312,15 @@ def _grep(context: RunContext, search_string: str, directory: str = ".") -> Grep
 def list_files(
     context: RunContext, directory: str = ".", recursive: bool = True
 ) -> ListFileOutput:
-    return _list_files(context, directory, recursive)
+    list_files_output = _list_files(context, directory, recursive)
+    tokenizer = get_tokenizer()
+    num_tokens = len(tokenizer.encode(list_files_output.model_dump_json()))
+    if num_tokens > 10000:
+        return ListFileOutput(
+            files=[],
+            error="Too many files - tokens exceeded. Try listing non-recursively"
+        )
+    return list_files_output
 
 
 def read_file(context: RunContext, file_path: str = "", start_line: int | None = None, num_lines: int | None = None) -> ReadFileOutput:

{code_puppy-0.0.87 → code_puppy-0.0.89}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.87"
+version = "0.0.89"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.10"