feat : support any language set in config

This commit is contained in:
martin legrand
2025-04-11 11:13:07 +02:00
parent e66f535dd3
commit 4198e932ca
5 changed files with 22 additions and 16 deletions
+2 -2
View File
@@ -20,10 +20,10 @@ class AgentRouter:
"""
AgentRouter is a class that selects the appropriate agent based on the user query.
"""
def __init__(self, agents: list):
def __init__(self, agents: list, supported_language: List[str] = ["en", "fr", "zh"]):
self.agents = agents
self.logger = Logger("router.log")
self.lang_analysis = LanguageUtility()
self.lang_analysis = LanguageUtility(supported_language=supported_language)
self.pipelines = self.load_pipelines()
self.talk_classifier = self.load_llm_router()
self.complexity_classifier = self.load_llm_router()