Spaces:
Running
Running
Remove EnhancedResearchAgent
Browse files
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": {}}
|