feat : vllm in server

sources/agents/browser_agent.py

@@ -112,7 +112,7 @@ class BrowserAgent(Agent):
 
         1. **Decide if the page answers the user’s query:**
           - If it does, take notes of useful information (Note: ...), include relevant link in note, then move to a new page.
-          - If it does and you completed user request, say {Action.REQUEST_EXIT}.
+          - If it does and no futher step would help with user request, say {Action.REQUEST_EXIT}.
           - If it doesn’t, say: Error: <why page don't help> then go back or navigate to another link.
         2. **Navigate to a link by either: **
           - Saying I will navigate to (write down the full URL) www.example.com/cats
@@ -145,7 +145,7 @@ class BrowserAgent(Agent):
         Action: {Action.GO_BACK.value}
 
         Example 3 (query answer found, enought notes taken):
-        Note: I found on <link> that ...<expand on information found>...
+        Note: I found on website www.example.com that ...<expand on information found>...
         Given this answer the user query I should exit the web browser.
         Action: {Action.REQUEST_EXIT.value}
 
@@ -161,8 +161,9 @@ class BrowserAgent(Agent):
         You previously took these notes:
         {notes}
         Do not Step-by-Step explanation. Write Notes or Error as a long paragraph followed by your action.
-        You might {Action.REQUEST_EXIT.value} if no more link are useful.
         If you conduct research do not exit until you have several notes.
+        Do not ever ask the user to conduct a task, do not ever exit expecting user intervention.
+        You should be Investigative, Curious and Skeptical.
         """
     
     def llm_decide(self, prompt: str, show_reasoning: bool = False) -> Tuple[str, str]:

sources/llm_provider.py

@@ -87,8 +87,8 @@ class Provider:
         except AttributeError as e:
             raise NotImplementedError(f"{str(e)}\nIs {self.provider_name} implemented ?")
         except Exception as e:
-            if "RemoteDisconnected" in str(e):
-                return f"{self.server_ip} seem offline. RemoteDisconnected error."
+            if "refused" in str(e):
+                return f"Server {self.server_ip} seem offline. Unable to answer."
             raise Exception(f"Provider {self.provider_name} failed: {str(e)}") from e
         return thought
 

sources/router.py

@@ -121,7 +121,9 @@ class AgentRouter:
             ("Write a Python function to merge two sorted lists", "LOW"),
             ("Organize my desktop files by extension and then write a script to list them", "HIGH"),
             ("Create a bash script to monitor disk space and alert via text file", "LOW"),
+            ("can you find vitess repo, clone it and install by following the readme", "HIGH"),
             ("Search X for posts about AI ethics and summarize them", "LOW"),
+            ("Can you follow the readme and install the project", "HIGH"),
             ("Find the latest research on renewable energy and build a web app to display it", "HIGH"),
             ("Write a C program to sort an array of integers", "LOW"),
             ("Create a Node.js server that queries a public API for traffic data and displays it", "HIGH"),