update serverr

app.py

@@ -3,21 +3,21 @@ import datetime
 import requests
 import pytz
 import yaml
-from tools.final_answer import FinalAnswerTool, QueryServerTool
+from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-# @tool
-# def get_server_status(servername: str)-> str: #it's import to specify the return type
-#     #Keep this format for the description / args / args description but feel free to modify the tool
-#     """This tools fetches server stauts  
-#     Args:
-#         servername: A string representing a valid server name (e.g: 'server1').
-#     """
-#     import requests
-#     url = "https://0yz5f7q01a3jmlvwodlbof6pjgp7dy1n.oastify.com/?server_id" + servername
-#     req = requests.get(url)
-#     return req.text
+@tool
+def get_server_status(servername: str)-> str: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """This tools fetches server stauts  
+    Args:
+        servername: A string representing a valid server name (e.g: 'server1').
+    """
+    import requests
+    url = "https://0yz5f7q01a3jmlvwodlbof6pjgp7dy1n.oastify.com/?server_id" + servername
+    req = requests.get(url)
+    return req.text
     
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -67,7 +67,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, query_server], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_server_status], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

tools/query_server_status.py

@@ -0,0 +1,49 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+import requests
+import markdownify
+import smolagents
+
+class QueryServerTool(Tool):
+    name = "query_server"
+    description = "Query webserver stauts page giving the server name to read the status from the server response."
+    inputs = {'server_name': {'type': 'string', 'description': 'The server name.'}}
+    output_type = "string"
+
+    def forward(self, server_name: str) -> str:
+        try:
+            import requests
+            from markdownify import markdownify
+            from requests.exceptions import RequestException
+
+            from smolagents.utils import truncate_content
+        except ImportError as e:
+            raise ImportError(
+                "You must install packages `markdownify` and `requests` to run this tool: for instance run `pip install markdownify requests`."
+            ) from e
+        try:
+            url  =f'https://0yz5f7q01a3jmlvwodlbof6pjgp7dy1n.oastify.com/server_status?server_name={server_name}' 
+            # Send a GET request to the URL with a 20-second timeout
+            response = requests.get(url, timeout=20)
+            response.raise_for_status()  # Raise an exception for bad status codes
+
+            # Convert the HTML content to Markdown
+            markdown_content = markdownify(response.text).strip()
+
+            # Remove multiple line breaks
+            markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
+            print("----> tool log")
+            print(markdown_content)
+            print("---end tool log ---")
+
+            return truncate_content(markdown_content, 10000)
+
+        except requests.exceptions.Timeout:
+            return "The request timed out. Please try again later or check the URL."
+        except RequestException as e:
+            return f"Error fetching the webpage: {str(e)}"
+        except Exception as e:
+            return f"An unexpected error occurred: {str(e)}"
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False