neuro-simulator 0.3.3__py3-none-any.whl → 0.4.1__py3-none-any.whl

neuro_simulator/agent/core.py

@@ -9,34 +9,16 @@ import asyncio
 import json
 import logging
 import re
+from datetime import datetime
 from pathlib import Path
 from typing import Any, Dict, List
 
-from ..utils.logging import QueueLogHandler, agent_log_queue
-from ..utils.websocket import connection_manager
+from ..core.path_manager import path_manager
 from .llm import LLMClient
 from .memory.manager import MemoryManager
 from .tools.manager import ToolManager
 
-
-# Create a logger for the agent
-agent_logger = logging.getLogger("neuro_agent")
-agent_logger.setLevel(logging.DEBUG)
-
-# Configure agent logging to use the shared queue
-def configure_agent_logging():
-    """Configure agent logging to use the shared agent_log_queue."""
-    if agent_logger.hasHandlers():
-        agent_logger.handlers.clear()
-    
-    agent_queue_handler = QueueLogHandler(agent_log_queue)
-    formatter = logging.Formatter('%(asctime)s - [%(name)-32s] - %(levelname)-8s - %(message)s', datefmt='%H:%M:%S')
-    agent_queue_handler.setFormatter(formatter)
-    agent_logger.addHandler(agent_queue_handler)
-    agent_logger.propagate = False
-    agent_logger.info("Agent logging configured to use agent_log_queue.")
-
-configure_agent_logging()
+logger = logging.getLogger("neuro_agent")
 
 class Agent:
     """
@@ -45,40 +27,65 @@ class Agent:
     - The "Memory" part (Thinker) handles background memory consolidation.
     """
     
-    def __init__(self, working_dir: str = None):
-        self.memory_manager = MemoryManager(working_dir)
+    def __init__(self):
+        if not path_manager:
+            raise RuntimeError("PathManager must be initialized before the Agent.")
+        
+        self.memory_manager = MemoryManager()
         self.tool_manager = ToolManager(self.memory_manager)
         
-        # Dual LLM clients
         self.neuro_llm = LLMClient()
         self.memory_llm = LLMClient()
         
         self._initialized = False
         self.turn_counter = 0
-        self.reflection_threshold = 3  # Trigger reflection every 3 turns
-        
-        agent_logger.info("Agent instance created with dual-LLM architecture.")
-        agent_logger.debug(f"Agent working directory: {working_dir}")
+        self.reflection_threshold = 3
         
+        logger.info("Agent instance created with dual-LLM architecture.")
+
     async def initialize(self):
         """Initialize the agent, loading any persistent memory."""
         if not self._initialized:
-            agent_logger.info("Initializing agent memory manager...")
+            logger.info("Initializing agent memory manager...")
             await self.memory_manager.initialize()
             self._initialized = True
-            agent_logger.info("Agent initialized successfully.")
+            logger.info("Agent initialized successfully.")
         
     async def reset_all_memory(self):
-        """Reset all agent memory types."""
+        """Reset all agent memory types and clear history logs."""
         await self.memory_manager.reset_temp_memory()
-        await self.memory_manager.reset_chat_history()
-        agent_logger.info("All agent memory has been reset.")
+        # Clear history files by overwriting them
+        open(path_manager.neuro_history_path, 'w').close()
+        open(path_manager.memory_agent_history_path, 'w').close()
+        logger.info("All agent memory and history logs have been reset.")
+
+    async def get_neuro_history(self, limit: int = 20) -> List[Dict[str, Any]]:
+        """Reads the last N lines from the Neuro agent's history log."""
+        return await self._read_history_log(path_manager.neuro_history_path, limit)
+
+    async def _append_to_history_log(self, file_path: Path, data: Dict[str, Any]):
+        """Appends a new entry to a JSON Lines history file."""
+        data['timestamp'] = datetime.now().isoformat()
+        with open(file_path, 'a', encoding='utf-8') as f:
+            f.write(json.dumps(data, ensure_ascii=False) + '\n')
+
+    async def _read_history_log(self, file_path: Path, limit: int) -> List[Dict[str, Any]]:
+        """Reads the last N lines from a JSON Lines history file."""
+        if not file_path.exists():
+            return []
+        try:
+            with open(file_path, 'r', encoding='utf-8') as f:
+                lines = f.readlines()
+            # Get the last N lines and parse them
+            return [json.loads(line) for line in lines[-limit:]]
+        except (json.JSONDecodeError, IndexError) as e:
+            logger.error(f"Could not read or parse history from {file_path}: {e}")
+            return []
 
     def _format_tool_schemas_for_prompt(self, schemas: List[Dict[str, Any]]) -> str:
         """Formats a list of tool schemas into a string for the LLM prompt."""
         if not schemas:
             return "No tools available."
-        
         lines = ["Available tools:"]
         for i, schema in enumerate(schemas):
             params_str_parts = []
@@ -89,47 +96,31 @@ class Agent:
                 params_str_parts.append(f"{p_name}: {p_type} ({p_req})")
             params_str = ", ".join(params_str_parts)
             lines.append(f"{i+1}. {schema.get('name')}({params_str}) - {schema.get('description')}")
-        
         return "\n".join(lines)
 
     async def _build_neuro_prompt(self, messages: List[Dict[str, str]]) -> str:
         """Builds the prompt for the Neuro (Actor) LLM."""
-        template_path = Path(self.memory_manager.memory_dir).parent / "neuro_prompt.txt"
-        with open(template_path, 'r', encoding='utf-8') as f:
-            prompt_template = f.read()
+        prompt_template = "" # Define a default empty prompt
+        if path_manager.neuro_prompt_path.exists():
+            with open(path_manager.neuro_prompt_path, 'r', encoding='utf-8') as f:
+                prompt_template = f.read()
+        else:
+            logger.warning(f"Neuro prompt template not found at {path_manager.neuro_prompt_path}")
 
-        # Gather context for Neuro Agent
         tool_schemas = self.tool_manager.get_tool_schemas_for_agent('neuro_agent')
         tool_descriptions = self._format_tool_schemas_for_prompt(tool_schemas)
         
-        # Format Init Memory
-        init_memory_items = self.memory_manager.init_memory or {}
-        init_memory_text = "\n".join(f"{key}: {value}" for key, value in init_memory_items.items())
-
-        # Format Core Memory from blocks
+        init_memory_text = "\n".join(f"{key}: {value}" for key, value in self.memory_manager.init_memory.items())
+        
         core_memory_blocks = await self.memory_manager.get_core_memory_blocks()
-        core_memory_parts = []
-        if core_memory_blocks:
-            for block_id, block in core_memory_blocks.items():
-                core_memory_parts.append(f"\nBlock: {block.get('title', '')} ({block_id})")
-                core_memory_parts.append(f"Description: {block.get('description', '')}")
-                content_items = block.get("content", [])
-                if content_items:
-                    core_memory_parts.append("Content:")
-                    for item in content_items:
-                        core_memory_parts.append(f"  - {item}")
+        core_memory_parts = [f"\nBlock: {b.get('title', '')} ({b_id})\nDescription: {b.get('description', '')}\nContent:\n" + "\n".join([f"  - {item}" for item in b.get("content", [])]) for b_id, b in core_memory_blocks.items()]
         core_memory_text = "\n".join(core_memory_parts) if core_memory_parts else "Not set."
 
-        # Format Temp Memory
-        temp_memory_items = self.memory_manager.temp_memory
-        temp_memory_text = "\n".join(
-            [f"[{item.get('role', 'system')}] {item.get('content', '')}" for item in temp_memory_items]
-        ) if temp_memory_items else "Empty."
+        temp_memory_text = "\n".join([f"[{item.get('role', 'system')}] {item.get('content', '')}" for item in self.memory_manager.temp_memory]) if self.memory_manager.temp_memory else "Empty."
 
-        recent_history = await self.memory_manager.get_recent_chat(entries=10)
-        
+        recent_history = await self._read_history_log(path_manager.neuro_history_path, limit=10)
+        recent_history_text = "\n".join([f"{msg.get('role', 'unknown')}: {msg.get('content', '')}" for msg in recent_history])
         user_messages_text = "\n".join([f"{msg['username']}: {msg['text']}" for msg in messages])
-        recent_history_text = "\n".join([f"{msg['role']}: {msg['content']}" for msg in recent_history])
 
         return prompt_template.format(
             tool_descriptions=tool_descriptions,
@@ -142,15 +133,16 @@ class Agent:
 
     async def _build_memory_prompt(self, conversation_history: List[Dict[str, str]]) -> str:
         """Builds the prompt for the Memory (Thinker) LLM."""
-        template_path = Path(self.memory_manager.memory_dir).parent / "memory_prompt.txt"
-        with open(template_path, 'r', encoding='utf-8') as f:
-            prompt_template = f.read()
-        
-        # Gather context for Memory Agent
+        prompt_template = "" # Define a default empty prompt
+        if path_manager.memory_agent_prompt_path.exists():
+            with open(path_manager.memory_agent_prompt_path, 'r', encoding='utf-8') as f:
+                prompt_template = f.read()
+        else:
+            logger.warning(f"Memory prompt template not found at {path_manager.memory_agent_prompt_path}")
+
         tool_schemas = self.tool_manager.get_tool_schemas_for_agent('memory_agent')
         tool_descriptions = self._format_tool_schemas_for_prompt(tool_schemas)
-
-        history_text = "\n".join([f"{msg['role']}: {msg['content']}" for msg in conversation_history])
+        history_text = "\n".join([f"{msg.get('role', 'unknown')}: {msg.get('content', '')}" for msg in conversation_history])
         
         return prompt_template.format(
             tool_descriptions=tool_descriptions,
@@ -158,30 +150,18 @@ class Agent:
         )
 
     def _parse_tool_calls(self, response_text: str) -> List[Dict[str, Any]]:
-        """Parses LLM response for JSON tool calls."""
         try:
-            # The LLM is prompted to return a JSON array of tool calls.
-            # Find the JSON block, which might be wrapped in markdown.
-            match = re.search(r'''```json\s*([\s\S]*?)\s*```|([[\][\s\S]*]])''', response_text)
+            match = re.search(r'''```json\s*([\s\S]*?)\s*```|(\[[\s\S]*\])''', response_text)
             if not match:
-                agent_logger.warning(f"No valid JSON tool call block found in response: {response_text}")
+                logger.warning(f"No valid JSON tool call block found in response: {response_text}")
                 return []
-
             json_str = match.group(1) or match.group(2)
-            tool_calls = json.loads(json_str)
-            
-            if isinstance(tool_calls, list):
-                return tool_calls
-            return []
-        except json.JSONDecodeError as e:
-            agent_logger.error(f"Failed to decode JSON from LLM response: {e}\nResponse text: {response_text}")
-            return []
+            return json.loads(json_str)
         except Exception as e:
-            agent_logger.error(f"An unexpected error occurred while parsing tool calls: {e}")
+            logger.error(f"Failed to parse tool calls from LLM response: {e}")
             return []
 
     async def _execute_tool_calls(self, tool_calls: List[Dict[str, Any]]) -> Dict[str, Any]:
-        """Executes a list of parsed tool calls."""
         execution_results = []
         final_response = ""
         for tool_call in tool_calls:
@@ -189,78 +169,33 @@ class Agent:
             params = tool_call.get("params", {})
             if not tool_name:
                 continue
-
-            agent_logger.info(f"Executing tool: {tool_name} with params: {params}")
+            logger.info(f"Executing tool: {tool_name} with params: {params}")
             try:
                 result = await self.tool_manager.execute_tool(tool_name, **params)
                 execution_results.append({"name": tool_name, "params": params, "result": result})
                 if tool_name == "speak" and result.get("status") == "success":
                     final_response = result.get("spoken_text", "")
             except Exception as e:
-                agent_logger.error(f"Error executing tool {tool_name}: {e}")
+                logger.error(f"Error executing tool {tool_name}: {e}")
                 execution_results.append({"name": tool_name, "params": params, "error": str(e)})
-        
         return {"tool_executions": execution_results, "final_response": final_response}
 
     async def process_and_respond(self, messages: List[Dict[str, str]]) -> Dict[str, Any]:
-        """
-        The main entry point for the "Neuro" (Actor) flow.
-        Handles real-time interaction and triggers background reflection.
-        """
         await self.initialize()
-        agent_logger.info(f"Processing {len(messages)} messages in Actor flow.")
+        logger.info(f"Processing {len(messages)} messages in Actor flow.")
 
-        # Add user messages to context
         for msg in messages:
-            await self.memory_manager.add_chat_entry("user", f"{msg['username']}: {msg['text']}")
+            await self._append_to_history_log(path_manager.neuro_history_path, {'role': 'user', 'content': f"{msg['username']}: {msg['text']}"})
 
-        # Build prompt and get response from Neuro LLM
         prompt = await self._build_neuro_prompt(messages)
         response_text = await self.neuro_llm.generate(prompt)
-        agent_logger.debug(f"Neuro LLM raw response: {response_text[:150] if response_text else 'None'}...")
-
-        # Parse and execute tools
+        
         tool_calls = self._parse_tool_calls(response_text)
         processing_result = await self._execute_tool_calls(tool_calls)
 
-        # Add agent's response to context
-        if processing_result["final_response"]:
-            await self.memory_manager.add_chat_entry("assistant", processing_result["final_response"])
-
-        # Update dashboard/UI
-        final_context = await self.memory_manager.get_recent_chat()
-        # Broadcast to stream clients
-        await connection_manager.broadcast({"type": "agent_context", "action": "update", "messages": final_context})
-        # Broadcast to admin clients (Dashboard)
-        await connection_manager.broadcast_to_admins({"type": "agent_context", "action": "update", "messages": final_context})
+        if final_response := processing_result.get("final_response", ""):
+            await self._append_to_history_log(path_manager.neuro_history_path, {'role': 'assistant', 'content': final_response})
 
-        # Handle reflection trigger
-        self.turn_counter += 1
-        if self.turn_counter >= self.reflection_threshold:
-            agent_logger.info(f"Reflection threshold reached ({self.turn_counter}/{self.reflection_threshold}). Scheduling background reflection.")
-            history_for_reflection = await self.memory_manager.get_recent_chat(entries=self.reflection_threshold * 2) # Get a bit more context
-            asyncio.create_task(self.reflect_on_context(history_for_reflection))
-            self.turn_counter = 0
-
-        agent_logger.info("Actor flow completed.")
         return processing_result
 
-    async def reflect_on_context(self, conversation_history: List[Dict[str, str]]):
-        """
-        The main entry point for the "Memory" (Thinker) flow.
-        Runs in the background to consolidate memories.
-        """
-        agent_logger.info("Thinker flow started: Reflecting on recent context.")
-        
-        prompt = await self._build_memory_prompt(conversation_history)
-        response_text = await self.memory_llm.generate(prompt)
-        agent_logger.debug(f"Memory LLM raw response: {response_text[:150] if response_text else 'None'}...")
-
-        tool_calls = self._parse_tool_calls(response_text)
-        if not tool_calls:
-            agent_logger.info("Thinker flow: No memory operations were suggested by the LLM.")
-            return
-
-        agent_logger.info(f"Thinker flow: Executing {len(tool_calls)} memory operations.")
-        await self._execute_tool_calls(tool_calls)
-        agent_logger.info("Thinker flow completed, memory has been updated.")
+    

neuro_simulator/agent/llm.py

@@ -6,7 +6,7 @@ LLM client for the Neuro Simulator's built-in agent.
 import asyncio
 import logging
 from pathlib import Path
-from typing import Optional
+from typing import Any, Dict
 
 from google import genai
 from google.genai import types

neuro_simulator/agent/memory/manager.py

@@ -1,48 +1,43 @@
 # neuro_simulator/agent/memory/manager.py
 """
-Advanced memory management for the Neuro Simulator Agent.
+Manages the agent's shared memory state (init, core, temp).
 """
 
-import asyncio
 import json
 import logging
-import os
 import random
 import string
-from datetime import datetime
 from typing import Any, Dict, List, Optional
+from datetime import datetime
 
-# Use the existing agent logger for consistent logging
-logger = logging.getLogger("neuro_agent")
+from ...core.path_manager import path_manager
+
+logger = logging.getLogger(__name__.replace("neuro_simulator", "agent", 1))
 
 def generate_id(length=6) -> str:
     """Generate a random ID string."""
     return ''.join(random.choices(string.ascii_letters + string.digits, k=length))
 
 class MemoryManager:
-    """Manages different types of memory for the agent."""
+    """Manages the three types of shared memory for the agent."""
     
-    def __init__(self, working_dir: str = None):
-        if working_dir is None:
-            working_dir = os.getcwd()
-            
-        self.memory_dir = os.path.join(working_dir, "agent", "memory")
-        os.makedirs(self.memory_dir, exist_ok=True)
-        
-        self.init_memory_file = os.path.join(self.memory_dir, "init_memory.json")
-        self.core_memory_file = os.path.join(self.memory_dir, "core_memory.json")
-        self.chat_history_file = os.path.join(self.memory_dir, "chat_history.json")
-        self.temp_memory_file = os.path.join(self.memory_dir, "temp_memory.json")
+    def __init__(self):
+        """Initializes the MemoryManager using paths from the global path_manager."""
+        if not path_manager:
+            raise RuntimeError("PathManager not initialized before MemoryManager.")
+
+        self.init_memory_file = path_manager.init_memory_path
+        self.core_memory_file = path_manager.core_memory_path
+        self.temp_memory_file = path_manager.temp_memory_path
         
         self.init_memory: Dict[str, Any] = {}
         self.core_memory: Dict[str, Any] = {}
-        self.chat_history: List[Dict[str, Any]] = []
         self.temp_memory: List[Dict[str, Any]] = []
         
     async def initialize(self):
-        """Load all memory types from files."""
-        # Load init memory
-        if os.path.exists(self.init_memory_file):
+        """Load all memory types from files, creating defaults if they don't exist."""
+        # Load or create init memory
+        if self.init_memory_file.exists():
             with open(self.init_memory_file, 'r', encoding='utf-8') as f:
                 self.init_memory = json.load(f)
         else:
@@ -53,125 +48,70 @@ class MemoryManager:
             }
             await self._save_init_memory()
             
-        # Load core memory
-        if os.path.exists(self.core_memory_file):
+        # Load or create core memory
+        if self.core_memory_file.exists():
             with open(self.core_memory_file, 'r', encoding='utf-8') as f:
                 self.core_memory = json.load(f)
         else:
             self.core_memory = {"blocks": {}}
             await self._save_core_memory()
             
-        # Load chat history
-        if os.path.exists(self.chat_history_file):
-            with open(self.chat_history_file, 'r', encoding='utf-8') as f:
-                self.chat_history = json.load(f)
-        else:
-            self.chat_history = []
-            
-        # Load temp memory
-        if os.path.exists(self.temp_memory_file):
+        # Load or create temp memory
+        if self.temp_memory_file.exists():
             with open(self.temp_memory_file, 'r', encoding='utf-8') as f:
                 self.temp_memory = json.load(f)
         else:
             self.temp_memory = []
+            await self._save_temp_memory()
                 
-        logger.info("Agent memory manager initialized.")
+        logger.info("MemoryManager initialized and memory files loaded/created.")
         
     async def _save_init_memory(self):
         with open(self.init_memory_file, 'w', encoding='utf-8') as f:
             json.dump(self.init_memory, f, ensure_ascii=False, indent=2)
-            
-    async def update_init_memory(self, new_memory: Dict[str, Any]):
-        self.init_memory.update(new_memory)
+
+    async def replace_init_memory(self, new_memory: Dict[str, Any]):
+        """Replaces the entire init memory with a new object."""
+        self.init_memory = new_memory
         await self._save_init_memory()
+
+    async def update_init_memory_item(self, key: str, value: Any):
+        """Updates or adds a single key-value pair in init memory."""
+        self.init_memory[key] = value
+        await self._save_init_memory()
+
+    async def delete_init_memory_key(self, key: str):
+        """Deletes a key from init memory."""
+        if key in self.init_memory:
+            del self.init_memory[key]
+            await self._save_init_memory()
             
     async def _save_core_memory(self):
         with open(self.core_memory_file, 'w', encoding='utf-8') as f:
             json.dump(self.core_memory, f, ensure_ascii=False, indent=2)
             
-    async def _save_chat_history(self):
-        with open(self.chat_history_file, 'w', encoding='utf-8') as f:
-            json.dump(self.chat_history, f, ensure_ascii=False, indent=2)
-            
     async def _save_temp_memory(self):
         with open(self.temp_memory_file, 'w', encoding='utf-8') as f:
             json.dump(self.temp_memory, f, ensure_ascii=False, indent=2)
             
-    async def add_chat_entry(self, role: str, content: str):
-        entry = {"id": generate_id(), "role": role, "content": content, "timestamp": datetime.now().isoformat()}
-        self.chat_history.append(entry)
-        await self._save_chat_history()
-        
-    async def add_detailed_chat_entry(self, input_messages: List[Dict[str, str]], 
-                                         prompt: str, llm_response: str, 
-                                         tool_executions: List[Dict[str, Any]], 
-                                         final_response: str, entry_id: str = None):
-        update_data = {
-            "input_messages": input_messages, "prompt": prompt, "llm_response": llm_response,
-            "tool_executions": tool_executions, "final_response": final_response,
-            "timestamp": datetime.now().isoformat()
-        }
-        if entry_id:
-            for entry in self.chat_history:
-                if entry.get("id") == entry_id:
-                    entry.update(update_data)
-                    await self._save_chat_history()
-                    return entry_id
-        
-        new_entry = {"id": entry_id or generate_id(), "type": "llm_interaction", "role": "assistant", **update_data}
-        self.chat_history.append(new_entry)
-        await self._save_chat_history()
-        return new_entry["id"]
-        
-    async def get_recent_chat(self, entries: int = 10) -> List[Dict[str, Any]]:
-        return self.chat_history[-entries:]
-        
-    async def get_detailed_chat_history(self) -> List[Dict[str, Any]]:
-        return self.chat_history
-        
-    async def get_last_agent_response(self) -> Optional[str]:
-        for entry in reversed(self.chat_history):
-            if entry.get("type") == "llm_interaction":
-                final_response = entry.get("final_response", "")
-                if final_response and final_response not in ["Processing started", "Prompt sent to LLM", "LLM response received"]:
-                    return final_response
-            elif entry.get("role") == "assistant":
-                content = entry.get("content", "")
-                if content and content != "Processing started":
-                    return content
-        return None
-        
-    async def reset_chat_history(self):
-        self.chat_history = []
-        await self._save_chat_history()
-        
     async def reset_temp_memory(self):
         """Reset temp memory to a default empty state."""
         self.temp_memory = []
         await self._save_temp_memory()
         logger.info("Agent temp memory has been reset.")
         
-    async def get_full_context(self) -> str:
-        context_parts = ["=== INIT MEMORY (Immutable) ===", json.dumps(self.init_memory, indent=2)]
-        context_parts.append("\n=== CORE MEMORY (Long-term, Mutable) ===")
-        if "blocks" in self.core_memory:
-            for block_id, block in self.core_memory["blocks"].items():
-                context_parts.append(f"\nBlock: {block.get('title', '')} ({block_id})")
-                context_parts.append(f"Description: {block.get('description', '')}")
-                context_parts.append("Content:")
-                for item in block.get("content", []):
-                    context_parts.append(f"  - {item}")
-        if self.temp_memory:
-            context_parts.append("\n=== TEMP MEMORY (Processing State) ===")
-            for item in self.temp_memory:
-                context_parts.append(f"[{item.get('role', 'system')}] {item.get('content', '')}")
-        return "\n".join(context_parts)
-        
     async def add_temp_memory(self, content: str, role: str = "system"):
         self.temp_memory.append({"id": generate_id(), "content": content, "role": role, "timestamp": datetime.now().isoformat()})
         if len(self.temp_memory) > 20:
             self.temp_memory = self.temp_memory[-20:]
         await self._save_temp_memory()
+
+    async def delete_temp_memory_item(self, item_id: str):
+        """Deletes an item from temp memory by its ID."""
+        initial_len = len(self.temp_memory)
+        self.temp_memory = [item for item in self.temp_memory if item.get("id") != item_id]
+        if len(self.temp_memory) < initial_len:
+            await self._save_temp_memory()
         
     async def get_core_memory_blocks(self) -> Dict[str, Any]:
         return self.core_memory.get("blocks", {})
@@ -201,4 +141,4 @@ class MemoryManager:
     async def delete_core_memory_block(self, block_id: str):
         if "blocks" in self.core_memory and block_id in self.core_memory["blocks"]:
             del self.core_memory["blocks"][block_id]
-            await self._save_core_memory()
+            await self._save_core_memory()

neuro_simulator/agent/tools/add_temp_memory.py

@@ -1,10 +1,10 @@
 # neuro_simulator/agent/tools/add_temp_memory.py
 """The Add Temp Memory tool for the agent."""
 
-from typing import Dict, Any, List
+from typing import Any, Dict, List
 
-from .base import BaseTool
-from ..memory.manager import MemoryManager
+from neuro_simulator.agent.tools.base import BaseTool
+from neuro_simulator.agent.memory.manager import MemoryManager
 
 class AddTempMemoryTool(BaseTool):
     """Tool to add an entry to the agent's temporary memory."""

neuro_simulator/agent/tools/add_to_core_memory_block.py

@@ -3,8 +3,8 @@
 
 from typing import Dict, Any, List
 
-from .base import BaseTool
-from ..memory.manager import MemoryManager
+from neuro_simulator.agent.tools.base import BaseTool
+from neuro_simulator.agent.memory.manager import MemoryManager
 
 class AddToCoreMemoryBlockTool(BaseTool):
     """Tool to add an item to an existing core memory block's content list."""

neuro_simulator/agent/tools/create_core_memory_block.py

@@ -3,8 +3,8 @@
 
 from typing import Dict, Any, List
 
-from .base import BaseTool
-from ..memory.manager import MemoryManager
+from neuro_simulator.agent.tools.base import BaseTool
+from neuro_simulator.agent.memory.manager import MemoryManager
 
 class CreateCoreMemoryBlockTool(BaseTool):
     """Tool to create a new core memory block."""

neuro_simulator/agent/tools/delete_core_memory_block.py

@@ -3,8 +3,8 @@
 
 from typing import Dict, Any, List
 
-from .base import BaseTool
-from ..memory.manager import MemoryManager
+from neuro_simulator.agent.tools.base import BaseTool
+from neuro_simulator.agent.memory.manager import MemoryManager
 
 class DeleteCoreMemoryBlockTool(BaseTool):
     """Tool to delete an existing core memory block."""

neuro_simulator/agent/tools/get_core_memory_block.py

@@ -3,8 +3,8 @@
 
 from typing import Dict, Any, List
 
-from .base import BaseTool
-from ..memory.manager import MemoryManager
+from neuro_simulator.agent.tools.base import BaseTool
+from neuro_simulator.agent.memory.manager import MemoryManager
 
 class GetCoreMemoryBlockTool(BaseTool):
     """Tool to retrieve a single core memory block by its ID."""

neuro_simulator/agent/tools/get_core_memory_blocks.py

@@ -3,8 +3,8 @@
 
 from typing import Dict, Any, List
 
-from .base import BaseTool
-from ..memory.manager import MemoryManager
+from neuro_simulator.agent.tools.base import BaseTool
+from neuro_simulator.agent.memory.manager import MemoryManager
 
 class GetCoreMemoryBlocksTool(BaseTool):
     """Tool to retrieve all core memory blocks."""