Doc: readme; Feat : language utility class, small fix

sources/router.py

@@ -9,6 +9,7 @@ from sources.agents.agent import Agent
 from sources.agents.code_agent import CoderAgent
 from sources.agents.casual_agent import CasualAgent
 from sources.agents.planner_agent import PlannerAgent
+from sources.agents.browser_agent import BrowserAgent
 from sources.utility import pretty_print
 
 class AgentRouter:
@@ -30,7 +31,7 @@ class AgentRouter:
         else:
             return "cpu"
 
-    def classify_text(self, text: str, threshold: float = 0.5) -> list:
+    def classify_text(self, text: str, threshold: float = 0.4) -> list:
         """
         Classify the text into labels (agent roles).
         Args:
@@ -41,8 +42,8 @@ class AgentRouter:
         """
         first_sentence = None
         for line in text.split("\n"):
-                first_sentence = line.strip()
-                break
+            first_sentence = line.strip()
+            break
         if first_sentence is None:
             first_sentence = text
         result = self.pipeline(first_sentence, self.labels, threshold=threshold)
@@ -67,30 +68,50 @@ class AgentRouter:
 
 if __name__ == "__main__":
     agents = [
-        CoderAgent("deepseek-r1:14b", "agent1", "../prompts/coder_agent.txt", "server"),
-        CasualAgent("deepseek-r1:14b", "agent2", "../prompts/casual_agent.txt", "server"),
-        PlannerAgent("deepseek-r1:14b", "agent3", "../prompts/planner_agent.txt", "server")
+        CasualAgent("deepseek-r1:14b", "jarvis", "../prompts/casual_agent.txt", None),
+        BrowserAgent("deepseek-r1:14b", "browser", "../prompts/planner_agent.txt", None),
+        CoderAgent("deepseek-r1:14b", "coder", "../prompts/coder_agent.txt", None)
     ]
     router = AgentRouter(agents)
-    
-    texts = ["""
-    Write a python script to check if the device on my network is connected to the internet
-    """,
-    """
-    Hey could you search the web for the latest news on the stock market ?
-    """,
-    """
-    hey can you give give a list of the files in the current directory ?
-    """,
-    """
-    Make a cool game to illustrate the current relation between USA and europe
-    """
+    texts = [
+        "Write a python script to check if the device on my network is connected to the internet",
+        #"Peut tu écrire un script python qui vérifie si l'appareil sur mon réseau est connecté à internet?",
+        #"写一个Python脚本，检查我网络上的设备是否连接到互联网",
+        "Hey could you search the web for the latest news on the tesla stock market ?",
+        #"嘿，你能搜索网页上关于股票市场的最新新闻吗？",
+        #"Yo, cherche sur internet comment va tesla en bourse.",
+        "I would like you to search for weather api and then make an app using this API",
+        #"我想让你搜索天气API，然后用这个API做一个应用程序",
+        #"J'aimerais que tu cherche une api météo et que l'utilise pour faire une application",
+        "Plan a 3-day trip to New York, including flights and hotels.",
+        #"计划一次为期3天的纽约之旅，包括机票和酒店。",
+        #"Planifie un trip de 3 jours à Paris, y compris les vols et hotels.",
+        "Find me the latest research papers on AI.",
+        #"给我找最新的AI研究论文。",
+        #"Trouve moi les derniers papiers de recherche en IA",
+        "Help me write a C++ program to sort an array",
+        #"帮我写一个C++程序来排序数组",
+        #"Aide moi à faire un programme c++ pour trier une array.",
+        "What’s the weather like today? Oh, and can you find a good weather app?",
+        #"今天天气怎么样？哦，你还能找到一个好的天气应用程序吗？",
+        #"La météo est comment aujourd'hui ? oh et trouve moi une bonne appli météo tant que tu y est.",
+        "Can you debug this Java code? It’s not working.",
+        #"你能调试这段Java代码吗？它不起作用。",
+        #"Peut tu m'aider à debugger ce code java, ça marche pas",
+        "What's the latest brainrot on the internet ?",
+        #"互联网上最新的“脑残”是什么？",
+        #"Quel est la dernière connerie sur internet ?",
+        "i would like to setup a new AI project, index as mark2",
+        #"我想建立一个新的 AI 项目，索引为 Mark2",
+        #"Je voudrais configurer un nouveau projet d'IA, index Mark2",
+        "Hey, can you find the old_project.zip file somewhere on my drive?",
+        #"嘿，你能在我驱动器上找到old_project.zip文件吗？",
+        #"Hé trouve moi le old_project.zip, il est quelque part sur mon disque.",
+        "Tell me a funny story",
+        #"给我讲一个有趣的故事",
+        #"Raconte moi une histoire drole"
     ]
-
     for text in texts:
-        print(text)
-        results = router.classify_text(text)
-        for result in results:
-            print(result["label"], "=>", result["score"])
+        print("Input text:", text)
         agent = router.select_agent(text)
-        print("Selected agent role:", agent.role)
+        print()