nimcode 0.4.2__tar.gz → 0.4.3__tar.gz

{nimcode-0.4.2 → nimcode-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nimcode
-Version: 0.4.2
+Version: 0.4.3
 Summary: A standalone, robust coding agent for NVIDIA NIM models.
 Home-page: https://github.com/Batunash/nimcode
 Author: Autonomous Agent

nimcode-0.4.3/src/nimcode/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.4.3"

{nimcode-0.4.2 → nimcode-0.4.3}/src/nimcode/agent.py

@@ -327,7 +327,7 @@ class Agent:
         return "Subagent reached max turns before completing."
 
     async def _distill_memory(self) -> list:
-        system_msg = self.messages[0]
+        system_msg = dict(self.messages[0])
         recent_msgs = self.messages[-4:] if len(self.messages) > 4 else self.messages[1:]
         old_msgs = self.messages[1:-4] if len(self.messages) > 4 else []
         
@@ -335,16 +335,23 @@ class Agent:
             return self.messages
             
         summary_prompt = "You are a summarization AI. Please summarize the following conversation concisely. Focus on the final state, any completed goals, and any context needed for the next steps:\n\n"
+        
+        existing_summary_marker = "\n\n[PREVIOUS MEMORY SUMMARY]\n"
+        if existing_summary_marker in system_msg["content"]:
+            parts = system_msg["content"].split(existing_summary_marker, 1)
+            system_msg["content"] = parts[0]
+            summary_prompt += f"Previous summary to include:\n{parts[1]}\n\n"
+            
         for m in old_msgs:
             summary_prompt += f"{m['role'].upper()}: {str(m['content'])[:500]}...\n"
             
         try:
             summary = await self.client.chat_one_shot(summary_prompt)
-            system_msg["content"] += f"\n\n[PREVIOUS MEMORY SUMMARY]\n{summary}"
+            system_msg["content"] += f"{existing_summary_marker}{summary}"
+            return [system_msg] + recent_msgs
         except Exception as e:
             logger.error(f"Distillation failed: {e}")
-            
-        return [system_msg] + recent_msgs
+            return self.messages
 
     async def run(self, initial_prompt: str = None) -> None:
         """Main execution loop for NimCode agent."""

{nimcode-0.4.2 → nimcode-0.4.3}/src/nimcode.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nimcode
-Version: 0.4.2
+Version: 0.4.3
 Summary: A standalone, robust coding agent for NVIDIA NIM models.
 Home-page: https://github.com/Batunash/nimcode
 Author: Autonomous Agent

{nimcode-0.4.2 → nimcode-0.4.3}/tests/test_agent.py

@@ -99,8 +99,8 @@ async def test_distill_memory(agent):
     agent.client.chat_one_shot = AsyncMock(return_value="Summarized history")
     
     new_msgs = await agent._distill_memory()
-    assert "[PREVIOUS MEMORY SUMMARY]" in agent.messages[0]["content"]
-    assert "Summarized history" in agent.messages[0]["content"]
+    assert "[PREVIOUS MEMORY SUMMARY]" in new_msgs[0]["content"]
+    assert "Summarized history" in new_msgs[0]["content"]
     assert len(new_msgs) == 5
 
 @pytest.mark.asyncio

{nimcode-0.4.2 → nimcode-0.4.3}/tests/test_version.py

@@ -11,10 +11,10 @@ def test_version_format():
         assert part.isdigit(), f"Version part '{part}' is not a digit"
 
 
-def test_version_is_0_4_2():
-    """Current version should be 0.4.2."""
+def test_version_is_0_4_3():
+    """Current version should be 0.4.3."""
     from nimcode.__version__ import __version__
-    assert __version__ == "0.4.2"
+    assert __version__ == "0.4.3"
 
 
 def test_updater_uses_version():

nimcode-0.4.2/src/nimcode/__version__.py

@@ -1 +0,0 @@
-__version__ = "0.4.2"