nimcode 0.3.3__tar.gz → 0.3.4__tar.gz

{nimcode-0.3.3 → nimcode-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nimcode
-Version: 0.3.3
+Version: 0.3.4
 Summary: A standalone, robust coding agent for NVIDIA NIM models.
 Author: Autonomous Agent
 Requires-Dist: httpx>=0.27.0

{nimcode-0.3.3 → nimcode-0.3.4}/README.md

@@ -121,6 +121,20 @@ NimCode creates a `.nimcode` directory inside your projects. This directory acts
 - `.nimcode/skills/`: Any custom guidelines, framework rules, or memories you teach the agent.
 - `.nimcode/history/`: File backups for `/undo` capabilities.
 
+### Advanced Configuration (Timeouts)
+
+NimCode's limits and timeouts can be customized through the `.nimcode/settings.json` file. Setting any timeout to `0` will completely disable it (infinite timeout).
+
+```json
+{
+    "timeout_command": 1200,    // Max time for bash commands (in seconds)
+    "timeout_llm": 120,         // Max time for LLM API responses (in seconds)
+    "timeout_format": 10,       // Max time for auto-formatters like black/prettier
+    "timeout_browser": 15000,   // Max time for browser operations (in milliseconds)
+    "timeout_updater": 3        // Max time for update checks
+}
+```
+
 ## 🛡️ Requirements
 
 - Python 3.8+

{nimcode-0.3.3 → nimcode-0.3.4}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="nimcode",
-    version="0.3.3",
+    version="0.3.4",
     packages=find_packages(where="src"),
     package_dir={"": "src"},
     install_requires=[

{nimcode-0.3.3 → nimcode-0.3.4}/src/nimcode/agent.py

@@ -53,7 +53,12 @@ class Agent:
         self.settings = load_settings()
         self.model = model or self.settings.get("model", "meta/llama-3.1-70b-instruct")
         api_base_url = self.settings.get("api_base_url", "https://integrate.api.nvidia.com/v1")
-        self.client = NimClient(api_key=api_key, base_url=api_base_url, model=self.model)
+        self.client = NimClient(
+            api_key=api_key, 
+            base_url=api_base_url, 
+            model=self.model,
+            timeout=self.settings.get("timeout_llm", 120.0)
+        )
         
         # Initialize MCP Manager
         self.mcp = MCPManager(self.settings)
@@ -338,8 +343,10 @@ class Agent:
                             if file_path.endswith(".py"):
                                 import subprocess
                                 try:
-                                    subprocess.run(["black", "-q", file_path], capture_output=True, timeout=10)
-                                    lint = subprocess.run(["flake8", "--select=E9,F821,F822,F823", file_path], capture_output=True, text=True, timeout=10)
+                                    t_fmt = self.settings.get("timeout_format", 10)
+                                    t_fmt = None if t_fmt == 0 else t_fmt
+                                    subprocess.run(["black", "-q", file_path], capture_output=True, timeout=t_fmt)
+                                    lint = subprocess.run(["flake8", "--select=E9,F821,F822,F823", file_path], capture_output=True, text=True, timeout=t_fmt)
                                     if lint.returncode != 0:
                                         import sys
                                         if sys.stdin.isatty():
@@ -357,7 +364,9 @@ class Agent:
                             elif file_path.endswith((".js", ".ts", ".jsx", ".tsx")):
                                 import subprocess
                                 try:
-                                    subprocess.run(["npx", "--yes", "prettier", "--write", file_path], capture_output=True, timeout=10)
+                                    t_fmt = self.settings.get("timeout_format", 10)
+                                    t_fmt = None if t_fmt == 0 else t_fmt
+                                    subprocess.run(["npx", "--yes", "prettier", "--write", file_path], capture_output=True, timeout=t_fmt)
                                 except Exception:
                                     pass
                 except KeyboardInterrupt:

{nimcode-0.3.3 → nimcode-0.3.4}/src/nimcode/config.py

@@ -10,7 +10,12 @@ def load_settings() -> Dict[str, Any]:
     settings = {
         "model": "meta/llama-3.1-70b-instruct",
         "api_base_url": "https://integrate.api.nvidia.com/v1",
-        "mcp_servers": {}
+        "mcp_servers": {},
+        "timeout_command": 1200,    # Timeout for bash commands (seconds)
+        "timeout_llm": 120,         # Timeout for LLM API calls (seconds)
+        "timeout_format": 10,       # Timeout for formatting/linting tools (seconds)
+        "timeout_browser": 15000,   # Timeout for browser actions (milliseconds)
+        "timeout_updater": 3        # Timeout for update checks (seconds)
     }
     
     # Global settings

{nimcode-0.3.3 → nimcode-0.3.4}/src/nimcode/nim_client.py

@@ -8,10 +8,11 @@ from typing import List, Dict, Any, Optional, AsyncGenerator
 logger = logging.getLogger(__name__)
 
 class NimClient:
-    def __init__(self, api_key: str, base_url: str = "https://integrate.api.nvidia.com/v1", model: str = "meta/llama-3.1-70b-instruct"):
+    def __init__(self, api_key: str, base_url: str = "https://integrate.api.nvidia.com/v1", model: str = "meta/llama-3.1-70b-instruct", timeout: float = 120.0):
         self.api_key = api_key
         self.base_url = base_url.rstrip("/")
         self.model = model
+        self.timeout = None if timeout == 0 else timeout
         self.headers = {
             "Authorization": f"Bearer {self.api_key}",
             "Content-Type": "application/json",
@@ -39,7 +40,7 @@ class NimClient:
                 f"{self.base_url}/chat/completions",
                 headers=self.headers,
                 json=payload,
-                timeout=60.0
+                timeout=self.timeout
             )
             response.raise_for_status()
             data = response.json()
@@ -70,7 +71,7 @@ class NimClient:
                     f"{self.base_url}/chat/completions",
                     headers=self.headers,
                     json=payload,
-                    timeout=60.0
+                    timeout=self.timeout
                 )
                 response.raise_for_status()
                 data = response.json()
@@ -98,7 +99,7 @@ class NimClient:
             chunk_yielded = False
             try:
                 async with httpx.AsyncClient() as client:
-                    async with client.stream("POST", f"{self.base_url}/chat/completions", headers=self.headers, json=payload, timeout=120.0) as response:
+                    async with client.stream("POST", f"{self.base_url}/chat/completions", headers=self.headers, json=payload, timeout=self.timeout) as response:
                         if response.status_code in [408, 429, 500, 502, 503, 504, 529]:
                             raise httpx.HTTPStatusError(f"Temporary server error {response.status_code}", request=response.request, response=response)
                         

{nimcode-0.3.3 → nimcode-0.3.4}/src/nimcode/tools.py

@@ -239,6 +239,8 @@ class ToolRegistry:
             return f"Started task '{task_id}' in the background. Check logs at {log_file}. Use /tasks to manage."
             
         try:
+            t_cmd = settings.get("timeout_command", 1200)
+            t_cmd = None if t_cmd == 0 else t_cmd
             result = subprocess.run(
                 command,
                 shell=True,
@@ -246,7 +248,7 @@ class ToolRegistry:
                 stdout=subprocess.PIPE,
                 stderr=subprocess.PIPE,
                 text=True,
-                timeout=1200
+                timeout=t_cmd
             )
             out = result.stdout + "\n" + result.stderr
             out = out.strip()
@@ -257,7 +259,7 @@ class ToolRegistry:
             
             return out if out else "Command executed successfully with no output."
         except subprocess.TimeoutExpired:
-            return "Error: Command timed out after 1200 seconds."
+            return f"Error: Command timed out after {t_cmd} seconds."
         except Exception as e:
             return f"Error running bash: {e}"
 
@@ -565,11 +567,16 @@ class ToolRegistry:
         try:
             from playwright.sync_api import sync_playwright
             import base64
+            from .config import load_settings
+            
+            settings = load_settings()
+            t_browser = settings.get("timeout_browser", 15000)
+            t_browser = 0 if t_browser == 0 else t_browser
             
             with sync_playwright() as p:
                 browser = p.chromium.launch(headless=True)
                 page = browser.new_page()
-                page.goto(url, wait_until="networkidle", timeout=15000)
+                page.goto(url, wait_until="networkidle", timeout=t_browser)
                 
                 screenshot_bytes = page.screenshot(type="jpeg", quality=50)
                 b64_img = base64.b64encode(screenshot_bytes).decode('utf-8')

{nimcode-0.3.3 → nimcode-0.3.4}/src/nimcode/updater.py

@@ -13,7 +13,12 @@ class AutoUpdater:
         Returns the latest version string if > CURRENT_VERSION, else None.
         """
         try:
-            async with httpx.AsyncClient(timeout=3.0) as client:
+            from .config import load_settings
+            settings = load_settings()
+            t_upd = settings.get("timeout_updater", 3.0)
+            t_upd = None if t_upd == 0 else t_upd
+            
+            async with httpx.AsyncClient(timeout=t_upd) as client:
                 response = await client.get("https://pypi.org/pypi/nimcode/json")
                 if response.status_code == 200:
                     data = response.json()

{nimcode-0.3.3 → nimcode-0.3.4}/src/nimcode.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nimcode
-Version: 0.3.3
+Version: 0.3.4
 Summary: A standalone, robust coding agent for NVIDIA NIM models.
 Author: Autonomous Agent
 Requires-Dist: httpx>=0.27.0