Update definitions.py

definitions.py

@@ -305,16 +305,14 @@ class ChatSystem:
 
 class AutonomousAgent:
     """Autonomous agent for the system."""
-    def __init__(self):
+    def __init__(self, app):
+        self.app = app
         self.workspace_manager = WorkspaceManager(workspace_dir=os.getenv('WORKSPACE_DIR', 'workspace'))
         self.pipeline = self._initialize_pipeline()
-        self.refinement_loop = RefinementLoop(pipeline=self.pipeline)  # Access RefinementLoop directly
+        self.refinement_loop = RefinementLoop(pipeline=self.pipeline)
         self.interface = self.StreamlitInterface(self)
-        self.autonomous_agent = self.AutonomousAgent(self)
-        self.autonomous_agent.initialize_chat_system(self) # Initialize chat system after autonomous agent is created
         self.tools_repository = self._initialize_tool_repository()
-        self.autonomous_agent.initialize_tool_repository(self.tools_repository)
-        self._app = app
+        self.chat_system = ChatSystem(self) 
 
     def initialize_tool_repository(self, tool_repository: object) -> None:
         """
@@ -352,16 +350,18 @@ class AutonomousAgent:
             return self.SecurityAgent(agent_name)
         else:
             return None
+    
     class StreamlitInterface:
         """Streamlit UI integration for the Autonomous Agent system."""
     
-    def __init__(self, app: 'AutonomousAgentApp'):  # Use string forward reference
+    def __init__(self, app):
         self.app = app
-        self.chat_system = self.app.autonomous_agent.ChatSystem(self.app.autonomous_agent)
-        
+        self.chat_system = self.app.chat_system
+
+    
     class ChatSystem:
         """Manages the chat interaction between users and the autonomous system."""
-        def __init__(self, agent: 'AutonomousAgentApp.AutonomousAgent'):
+        def __init__(self, agent):
             self.agent = agent
             self.chat_history = []
             self.active_tasks = {}
@@ -372,8 +372,8 @@ class AutonomousAgent:
                 '/help': self.handle_help_command,
                 '/modify': self.handle_modify_command
             }
-            self.logger = logging.getLogger(__name__)  # Define the logger attribute
-
+            self.logger = logging.getLogger(__name__)
+            
         def render_chat_interface(self):
             """Render the chat interface in Streamlit sidebar."""
             with st.sidebar: