neuro-simulator 0.1.3__py3-none-any.whl → 0.2.0__py3-none-any.whl

neuro_simulator/__init__.py

@@ -1,10 +1 @@
-# neuro_simulator/__init__.py
-
-# 导出主要模块以便于使用
-from .builtin_agent import initialize_builtin_agent, get_builtin_response, reset_builtin_agent_memory
-
-__all__ = [
-    "initialize_builtin_agent",
-    "get_builtin_response",
-    "reset_builtin_agent_memory"
-]
+# neuro_simulator package root

neuro_simulator/agent/__init__.py

@@ -1,8 +1 @@
-# agent/__init__.py
-"""
-Agent module for Neuro Simulator Server
-"""
-
-from .core import Agent
-
-__all__ = ["Agent"]
+# neuro_simulator.agent package

neuro_simulator/agent/base.py

@@ -0,0 +1,43 @@
+# agent/base.py
+"""Base classes for Neuro Simulator Agent"""
+
+from abc import ABC, abstractmethod
+from typing import List, Dict, Any, Optional
+
+
+class BaseAgent(ABC):
+    """Abstract base class for all agents"""
+    
+    @abstractmethod
+    async def initialize(self):
+        """Initialize the agent"""
+        pass
+    
+    @abstractmethod
+    async def reset_memory(self):
+        """Reset agent memory"""
+        pass
+    
+    @abstractmethod
+    async def get_response(self, chat_messages: List[Dict[str, str]]) -> Dict[str, Any]:
+        """Get response from the agent
+        
+        Args:
+            chat_messages: List of message dictionaries with 'username' and 'text' keys
+            
+        Returns:
+            Dictionary containing processing details including tool executions and final response
+        """
+        pass
+    
+    @abstractmethod
+    async def process_messages(self, messages: List[Dict[str, str]]) -> Dict[str, Any]:
+        """Process messages and generate a response
+        
+        Args:
+            messages: List of message dictionaries with 'username' and 'text' keys
+            
+        Returns:
+            Dictionary containing processing details including tool executions and final response
+        """
+        pass