feat : zero shot router for other lang

This commit is contained in:
martin legrand
2025-03-24 20:22:30 +01:00
parent a289ddf1fd
commit af168d2c57
7 changed files with 62 additions and 16 deletions
+6 -1
View File
@@ -17,7 +17,12 @@ class BrowserAgent(Agent):
self.tools = {
"web_search": searxSearch(),
}
self.role = "Web search and navigation"
self.role = {
"en": "Web search and navigation",
"fr": "Recherche et navigation web",
"zh": "网络搜索和导航",
"es": "Búsqueda y navegación web"
}
self.type = "browser_agent"
self.browser = Browser()
self.current_page = ""
+6 -1
View File
@@ -18,7 +18,12 @@ class CasualAgent(Agent):
"file_finder": FileFinder(),
"bash": BashInterpreter()
}
self.role = "talk"
self.role = {
"en": "talking",
"fr": "discuter",
"zh": "聊天",
"es": "discutir"
}
self.type = "casual_agent"
def process(self, prompt, speech_module) -> str:
+6 -1
View File
@@ -20,7 +20,12 @@ class CoderAgent(Agent):
"go": GoInterpreter(),
"file_finder": FileFinder()
}
self.role = "Coding task"
self.role = {
"en": "Coding task",
"fr": "Tâche de codage",
"zh": "编码任务",
"es": "Tarea de codificación"
}
self.type = "code_agent"
def process(self, prompt, speech_module) -> str:
+6 -1
View File
@@ -14,7 +14,12 @@ class FileAgent(Agent):
"file_finder": FileFinder(),
"bash": BashInterpreter()
}
self.role = "find and read files"
self.role = {
"en": "find and read files",
"fr": "trouver et lire des fichiers",
"zh": "查找和读取文件",
"es": "encontrar y leer archivos"
}
self.type = "file_agent"
def process(self, prompt, speech_module) -> str:
+6 -1
View File
@@ -21,7 +21,12 @@ class PlannerAgent(Agent):
"file": FileAgent(model, name, prompt_path, provider),
"web": BrowserAgent(model, name, prompt_path, provider)
}
self.role = "Research, setup and code"
self.role = {
"en": "Research, setup and code",
"fr": "Recherche, configuration et codage",
"zh": "研究,设置和编码",
"es": "Investigación, configuración y code"
}
self.type = "planner_agent"
def parse_agent_tasks(self, text):