nimcode 0.1.20__tar.gz → 0.1.22__tar.gz

{nimcode-0.1.20 → nimcode-0.1.22}/PKG-INFO

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

{nimcode-0.1.20 → nimcode-0.1.22}/setup.py

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

{nimcode-0.1.20 → nimcode-0.1.22}/src/nimcode/agent.py

@@ -65,14 +65,7 @@ class Agent:
         # Base system prompt
         final_prompt = SYSTEM_PROMPT + self.mcp.get_system_prompt_additions()
         
-        # Inject Repo Map
-        try:
-            from .repo_map import RepoMapper
-            mapper = RepoMapper(os.getcwd())
-            repo_map = mapper.generate_map()
-            final_prompt += f"\n\n--- REPOSITORY MAP ---\n{repo_map}\n----------------------\n"
-        except Exception as e:
-            logger.error(f"Failed to generate repo map: {e}")
+        # We inject the Repo Map dynamically during the first run to avoid blocking startup.
         
         # Load .nimcoderules if present
         rules_path = os.path.join(os.getcwd(), ".nimcoderules")
@@ -237,6 +230,16 @@ class Agent:
 
     async def run_headless(self, query: str, max_turns: int = 5) -> str:
         """Runs the agent without terminal streaming, useful for background tasks."""
+        if len(self.messages) == 1:
+            try:
+                from .repo_map import RepoMapper
+                import asyncio
+                mapper = RepoMapper(os.getcwd())
+                repo_map = await asyncio.to_thread(mapper.generate_map)
+                self.messages[0]["content"] += f"\n\n--- REPOSITORY MAP ---\n{repo_map}\n----------------------\n"
+            except Exception as e:
+                logger.error(f"Failed to generate repo map: {e}")
+                
         self.messages.append({"role": "user", "content": query})
         turn = 0
         from .tools import ToolRegistry
@@ -282,9 +285,14 @@ class Agent:
             
         messages = [{"role": "system", "content": "You are a summarization AI."}, {"role": "user", "content": summary_prompt}]
         try:
+            # Use a robust default model for distillation to prevent 404 errors with unsupported endpoints
+            original_model = self.client.model
+            self.client.model = "meta/llama-3.1-8b-instruct"
             summary = await self.client.chat_one_shot(messages)
+            self.client.model = original_model
             system_msg["content"] += f"\n\n[PREVIOUS MEMORY SUMMARY]\n{summary}"
         except Exception as e:
+            self.client.model = original_model
             logger.error(f"Distillation failed: {e}")
             
         return [system_msg] + recent_msgs
@@ -294,6 +302,18 @@ class Agent:
         if hasattr(self.mcp, "connect_all"):
             await self.mcp.connect_all()
             
+        if len(self.messages) == 1:
+            try:
+                from rich.console import Console
+                Console().print("[dim italic]Indexing repository...[/dim italic]")
+                from .repo_map import RepoMapper
+                import asyncio
+                mapper = RepoMapper(os.getcwd())
+                repo_map = await asyncio.to_thread(mapper.generate_map)
+                self.messages[0]["content"] += f"\n\n--- REPOSITORY MAP ---\n{repo_map}\n----------------------\n"
+            except Exception as e:
+                logger.error(f"Failed to generate repo map: {e}")
+                
         if initial_prompt:
             self.messages.append({"role": "user", "content": initial_prompt})
         

{nimcode-0.1.20 → nimcode-0.1.22}/src/nimcode/repl.py

@@ -347,6 +347,7 @@ class NimcodeREPL:
                     table.add_column("Description", style="white")
                     
                     table.add_row("/help", "Show this help menu")
+                    table.add_row("/login", "Login with your NVIDIA NIM API Key")
                     table.add_row("/plan", "Enter planning mode (safe mode)")
                     table.add_row("/code", "Enter coding mode (all tools enabled)")
                     table.add_row("/trust", "Enter trust mode (runs all tools without permission)")
@@ -430,6 +431,16 @@ class NimcodeREPL:
                     self.permission_engine.mode = PermissionMode.DEFAULT
                     self.messages.append({"role": "system", "content": "You are now in planning mode. Use Read tools to explore. Then use the Write tool to write a markdown plan file inside the '.nimcode/plans/' directory (e.g., '.nimcode/plans/feature_x_plan.md'). Do NOT use Bash or Edit tools."})
                     continue
+                elif user_input.strip() == "/login":
+                    from .cli import run_login
+                    run_login()
+                    # Re-initialize client if api key changed
+                    from .config import load_settings
+                    settings = load_settings()
+                    if settings.get("api_key"):
+                        self.agent.client.api_key = settings.get("api_key")
+                        self.agent.client.headers["Authorization"] = f"Bearer {settings.get('api_key')}"
+                    continue
                 elif user_input.strip().startswith("/teleport"):
                     parts = user_input.strip().split(" ", 1)
                     if len(parts) > 1:
@@ -1132,7 +1143,7 @@ class NimcodeREPL:
                     continue
                 elif user_input.strip().startswith("/"):
                     import difflib
-                    valid_commands = ["/help", "/plan", "/code", "/trust", "/untrust", "/models", "/theme", "/clear", "/compact", "/commit", "/fix", "/exit", "/quit", "/config", "/alias", "/add", "/rewind", "/fork", "/testgen", "/vision", "/voice", "/index", "/research", "/mcp install", "/swarm", "/tdd", "/learn", "/cost", "/effort", "/thinking", "/grill-me", "/update", "/undo", "/teleport", "/buddy", "/ultraplan", "/bughunter", "/security-review", "/doctor", "/permissions", "/graph", "/guardian", "/thinkback", "/autofix-pr", "/terraform-god", "/sql-tune", "/decompile", "/sandbox"]
+                    valid_commands = ["/help", "/login", "/plan", "/code", "/trust", "/untrust", "/models", "/theme", "/clear", "/compact", "/commit", "/fix", "/exit", "/quit", "/config", "/alias", "/add", "/rewind", "/fork", "/testgen", "/vision", "/voice", "/index", "/research", "/mcp install", "/swarm", "/tdd", "/learn", "/cost", "/effort", "/thinking", "/grill-me", "/update", "/undo", "/teleport", "/buddy", "/ultraplan", "/bughunter", "/security-review", "/doctor", "/permissions", "/graph", "/guardian", "/thinkback", "/autofix-pr", "/terraform-god", "/sql-tune", "/decompile", "/sandbox"]
                     valid_commands.extend([f"/{cmd}" for cmd in self.plugin_manager.get_command_names()])
                     
                     cmd_parts = user_input.strip().split(" ", 1)

{nimcode-0.1.20 → nimcode-0.1.22}/src/nimcode.egg-info/PKG-INFO

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