Update app.py

app.py

@@ -434,13 +434,13 @@ class BasicAgent:
 
         try:
         # Initialize all tool instances
-        self.search_tool = DuckDuckGoSearchTool()
-        self.wiki_search_tool = WikiSearchTool() # Ensure this class is defined/imported
-        self.str_reverse_tool = StringReverseTool() # Ensure this class is defined/imported
-        self.keywords_extract_tool = KeywordsExtractorTool() # Ensure this class is defined/imported
-        self.speech_to_text_tool = SpeechToTextTool() # Ensure this class is defined/imported
-        self.visit_webpage_tool = VisitWebpageTool() # Ensure this class is defined/imported
-        self.final_answer_tool = FinalAnswerTool()
+            self.search_tool = DuckDuckGoSearchTool()
+            self.wiki_search_tool = WikiSearchTool() # Ensure this class is defined/imported
+            self.str_reverse_tool = StringReverseTool() # Ensure this class is defined/imported
+            self.keywords_extract_tool = KeywordsExtractorTool() # Ensure this class is defined/imported
+            self.speech_to_text_tool = SpeechToTextTool() # Ensure this class is defined/imported
+            self.visit_webpage_tool = VisitWebpageTool() # Ensure this class is defined/imported
+            self.final_answer_tool = FinalAnswerTool()
         except Exception as e:
             raise RuntimeError(f"Error initializing tools: {e}")
 
@@ -452,7 +452,6 @@ class BasicAgent:
         self.code_llama_tool = CodeLlamaTool() # Ensure this class is defined/imported
         self.parse_excel_to_json_tool = ParseExcelToJsonTool()
         
-
         
         system_prompt_template = """
 You are my general AI assistant. Your task is to answer the question I asked.
@@ -466,40 +465,40 @@ If the answer is a comma-separated list, apply the above rules for each element
 
         # Create web agent with image analysis capability
         try:
-        self.web_agent = ToolCallingAgent(
-            tools=[
-                self.search_tool, # Use the initialized DuckDuckGoSearchTool instance
-                self.visit_webpage_tool,
-                self.image_analysis_tool
-            ],
-            model=self.model, # Use self.model
-            max_steps=10,
-            name="web_search_agent",
-            description="Runs web searches and analyzes images",
-        )
+            self.web_agent = ToolCallingAgent(
+                tools=[
+                    self.search_tool, # Use the initialized DuckDuckGoSearchTool instance
+                    self.visit_webpage_tool,
+                    self.image_analysis_tool
+                ],
+                model=self.model, # Use self.model
+                max_steps=10,
+                name="web_search_agent",
+                description="Runs web searches and analyzes images",
+            )
         except Exception as e:
             raise RuntimeError(f"Error initializing web agent: {e}")
 
         # Create main agent with all capabilities
         try:
-        self.agent = CodeAgent(
-            model=self.model, # Use self.model
-            tools=[
-                self.search_tool,
-                self.wiki_search_tool,
-                self.str_reverse_tool,
-                self.keywords_extract_tool,
-                self.speech_to_text_tool,
-                self.visit_webpage_tool,
-                self.final_answer_tool,
-                self.video_transcription_tool,
-                self.code_llama_tool,
-                self.parse_excel_to_json_tool,
-                self.image_analysis_tool, # Use the initialized instance
-                self.analyse_attachment_tool # Add the initialized attachment analysis tool
-            ],
-            add_base_tools=True # Consider what this adds, ensure it doesn't duplicate.
-        )
+            self.agent = CodeAgent(
+                model=self.model, # Use self.model
+                tools=[
+                    self.search_tool,
+                    self.wiki_search_tool,
+                    self.str_reverse_tool,
+                    self.keywords_extract_tool,
+                    self.speech_to_text_tool,
+                    self.visit_webpage_tool,
+                    self.final_answer_tool,
+                    self.video_transcription_tool,
+                    self.code_llama_tool,
+                    self.parse_excel_to_json_tool,
+                    self.image_analysis_tool, # Use the initialized instance
+                    self.analyse_attachment_tool # Add the initialized attachment analysis tool
+                ],
+                add_base_tools=True # Consider what this adds, ensure it doesn't duplicate.
+            )
         except Exception as e:
             raise RuntimeError(f"Error initializing web agent: {e}")