nimcode 0.1.14__tar.gz → 0.1.16__tar.gz

{nimcode-0.1.14 → nimcode-0.1.16}/PKG-INFO

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

{nimcode-0.1.14 → nimcode-0.1.16}/setup.py

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

{nimcode-0.1.14 → nimcode-0.1.16}/src/nimcode/repo_map.py

@@ -13,7 +13,7 @@ class RepoMapper:
         try:
             with open(filepath, "r", encoding="utf-8") as f:
                 content = f.read()
-            tree = ast.parse(content)
+            tree = ast.parse(content, filename=filepath)
             
             lines = []
             for node in tree.body:
@@ -37,11 +37,24 @@ class RepoMapper:
         map_lines = []
         map_lines.append(f"Repository Map for: {self.root_dir}\n")
         
+        file_count = 0
+        max_files = 1000
+        
         for root, dirs, files in os.walk(self.root_dir):
-            dirs[:] = [d for d in dirs if d not in self.exclude_dirs]
+            # Exclude specific dirs, hidden dirs, and common system folders
+            dirs[:] = [
+                d for d in dirs 
+                if d not in self.exclude_dirs 
+                and not d.startswith('.')
+                and d.lower() not in ('appdata', 'windows', 'program files', 'program files (x86)', 'temp')
+            ]
             
             for file in files:
                 if file.endswith(".py"):
+                    if file_count >= max_files:
+                        map_lines.append("... (Repository Map Truncated: Too many Python files) ...\n")
+                        return "\n".join(map_lines)
+                        
                     full_path = os.path.join(root, file)
                     rel_path = os.path.relpath(full_path, self.root_dir)
                     map_lines.append(f"FILE: {rel_path}")
@@ -49,5 +62,6 @@ class RepoMapper:
                     file_map = self._parse_python_file(full_path)
                     map_lines.append(file_map)
                     map_lines.append("")
+                    file_count += 1
                     
         return "\n".join(map_lines)

{nimcode-0.1.14 → nimcode-0.1.16}/src/nimcode/stdio_server.py

@@ -59,6 +59,35 @@ class StdioServer:
             
         self.agent._stream_response = stdio_stream_response
 
+        # Override rich.console.Console.print to capture slash command outputs and general prints
+        from rich.console import Console
+        original_print = Console.print
+        
+        def patched_print(console_self, *objects, **kwargs):
+            # Try to extract raw string from objects
+            text_parts = []
+            for obj in objects:
+                if isinstance(obj, str):
+                    text_parts.append(obj)
+                elif hasattr(obj, "__rich_console__") or hasattr(obj, "__rich_measure__"):
+                    # For complex rich objects (like Markdown), we could render it to text
+                    try:
+                        with console_self.capture() as capture:
+                            original_print(console_self, obj, **kwargs)
+                        text_parts.append(capture.get())
+                    except Exception:
+                        text_parts.append(str(obj))
+                else:
+                    text_parts.append(str(obj))
+                    
+            if text_parts:
+                content = " ".join(text_parts)
+                # Strip excessive ANSI codes if capture includes them, but we want plain text or markdown
+                # For simplicity, just send it to the UI
+                self.send_message({"type": "info", "content": content})
+                
+        Console.print = patched_print
+
     def send_message(self, message: dict):
         """Send a JSON message to original stdout."""
         try:

{nimcode-0.1.14 → nimcode-0.1.16}/src/nimcode.egg-info/PKG-INFO

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