azettl commited on
Commit
f11f112
·
verified ·
1 Parent(s): a1abb42

Remove EnhancedResearchAgent

Browse files
Files changed (1) hide show
  1. app.py +14 -4
app.py CHANGED
@@ -13,9 +13,6 @@ import threading
13
  import queue
14
  import uuid
15
  from gradio_consilium_roundtable import consilium_roundtable
16
- from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, InferenceClientModel, VisitWebpageTool, Tool
17
- from research_tools import EnhancedResearchAgent
18
- from enhanced_search_functions import ENHANCED_SEARCH_FUNCTIONS
19
  from openfloor import *
20
  from openfloor.manifest import *
21
  from openfloor.envelope import *
@@ -262,9 +259,22 @@ def update_session_api_keys(mistral_key, sambanova_key, session_id_state, reques
262
  class VisualConsensusEngine:
263
  def __init__(self, moderator_model: str = None, update_callback=None, session_id: str = None):
264
  self.moderator_model = moderator_model or MODERATOR_MODEL
265
- self.search_agent = EnhancedResearchAgent()
266
  self.update_callback = update_callback
267
  self.session_id = session_id
 
 
 
 
 
 
 
 
 
 
 
 
 
 
268
 
269
  # Get session-specific keys or fall back to global
270
  session = get_or_create_session_state(session_id) if session_id else {"api_keys": {}}
 
13
  import queue
14
  import uuid
15
  from gradio_consilium_roundtable import consilium_roundtable
 
 
 
16
  from openfloor import *
17
  from openfloor.manifest import *
18
  from openfloor.envelope import *
 
259
  class VisualConsensusEngine:
260
  def __init__(self, moderator_model: str = None, update_callback=None, session_id: str = None):
261
  self.moderator_model = moderator_model or MODERATOR_MODEL
 
262
  self.update_callback = update_callback
263
  self.session_id = session_id
264
+
265
+ # Create OpenFloor research agents
266
+ from research_tools import WebSearchTool, WikipediaSearchTool, ArxivSearchTool, GitHubSearchTool, SECSearchTool
267
+
268
+ self.research_agents = {
269
+ 'web_search': OpenFloorResearchAgent(WebSearchTool(), port=8001),
270
+ 'wikipedia': OpenFloorResearchAgent(WikipediaSearchTool(), port=8002),
271
+ 'arxiv': OpenFloorResearchAgent(ArxivSearchTool(), port=8003),
272
+ 'github': OpenFloorResearchAgent(GitHubSearchTool(), port=8004),
273
+ 'sec_edgar': OpenFloorResearchAgent(SECSearchTool(), port=8005)
274
+ }
275
+
276
+ # Available research agents for discovery
277
+ self.available_research_agents = list(self.research_agents.keys())
278
 
279
  # Get session-specific keys or fall back to global
280
  session = get_or_create_session_state(session_id) if session_id else {"api_keys": {}}