refactor: self.role for router

This commit is contained in:
martin legrand
2025-04-12 11:50:48 +02:00
parent 06e6b2798b
commit f4feb42dda
8 changed files with 27 additions and 35 deletions
+1 -5
View File
@@ -26,11 +26,7 @@ class BrowserAgent(Agent):
self.tools = {
"web_search": searxSearch(),
}
self.role = {
"en": "web",
"fr": "web",
"zh": "网络",
}
self.role = "web"
self.type = "browser_agent"
self.browser = browser
self.current_page = ""
+1 -5
View File
@@ -14,11 +14,7 @@ class CasualAgent(Agent):
super().__init__(name, prompt_path, provider, verbose, None)
self.tools = {
} # No tools for the casual agent
self.role = {
"en": "talk",
"fr": "discuter",
"zh": "聊天",
}
self.role = "talk"
self.type = "casual_agent"
def process(self, prompt, speech_module) -> str:
+1 -5
View File
@@ -24,11 +24,7 @@ class CoderAgent(Agent):
"file_finder": FileFinder()
}
self.work_dir = self.tools["file_finder"].get_work_dir()
self.role = {
"en": "code",
"fr": "codage",
"zh": "编码",
}
self.role = "code"
self.type = "code_agent"
+1 -5
View File
@@ -15,11 +15,7 @@ class FileAgent(Agent):
"bash": BashInterpreter()
}
self.work_dir = self.tools["file_finder"].get_work_dir()
self.role = {
"en": "files",
"fr": "fichiers",
"zh": "文件",
}
self.role = "files"
self.type = "file_agent"
def process(self, prompt, speech_module) -> str:
+1 -5
View File
@@ -24,11 +24,7 @@ class PlannerAgent(Agent):
"file": FileAgent(name, "prompts/base/file_agent.txt", provider, verbose=False),
"web": BrowserAgent(name, "prompts/base/browser_agent.txt", provider, verbose=False, browser=browser)
}
self.role = {
"en": "Complex Task",
"fr": "Tache complexe",
"zh": "复杂任务",
}
self.role = "planification"
self.type = "planner_agent"
def parse_agent_tasks(self, text):