SYSTEM PROMPT

agent/agent.py

@@ -33,12 +33,14 @@ class Agent:
         api_key: str,
         model: str,
         max_iterations: int = 15,
+        system_prompt: str | None = None,
     ) -> None:
         self.client = OpenAI(base_url=base_url, api_key=api_key)
         self.model = model
         self._tools: list[Tool] = []
         self._final_tool_name: str | None = None
         self._max_iterations = max_iterations
+        self.system_prompt = system_prompt
 
     # ------------------------------------------------------------------
     # Tool registration
@@ -86,6 +88,12 @@ class Agent:
         """
         iteration = 0
 
+        # Inject system prompt once at the front if set
+        if self.system_prompt and (
+            not messages or messages[0].get("role") != "system"
+        ):
+            messages.insert(0, {"role": "system", "content": self.system_prompt})
+
         while True:
             iteration += 1
             if iteration > self._max_iterations:
@@ -123,10 +131,8 @@ class Agent:
                     continue
                 delta = choices[0].get("delta", {})
 
-                # Reasoning (e.g. DeepSeek R1)
                 reasoning = delta.get("reasoning_content") or delta.get("reasoning")
                 if reasoning:
-                    print(reasoning, flush=True, end="")
                     yield {"type": "reasoning", "content": reasoning}
 
                 # Text content

agent/tools.py

@@ -1,6 +1,6 @@
 from markitdown import MarkItDown
 from typing import Any, Callable
-
+import requests
 
 class Tool:
     """A callable tool the agent can invoke."""
@@ -29,8 +29,14 @@ class Tool:
 
 def fetch_webpage(url: str) -> str:
     """Fetch a webpage and return its text content."""
-    md = MarkItDown()
-    return md.convert(url).text_content
+    try:
+        jina_ai_url = "https://r.jina.ai/"
+        response = requests.get(jina_ai_url + url)
+        response.raise_for_status()
+        return response.text
+    except Exception as e:
+        md = MarkItDown()
+        return md.convert(url).text_content
 
 FETCH_WEBPAGE_TOOL = Tool(
     name="fetch_webpage",

app.py

@@ -9,10 +9,26 @@ from pathlib import Path
 load_dotenv()
 gr.set_static_paths("static/")
 
+_SYSTEM_PROMPT = """\
+You are OpenMythos, a powerful AI agent specialized in cybersecurity-related tasks.
+
+You have access to tools that you can use to accomplish your goals.
+
+=== IMPORTANT: How to end the conversation ===
+You MUST call the `final_message` tool when you have completed your response and want to end.
+If you do NOT call `final_message`, you will be stuck in a loop:
+  - You respond → system waits for final_message → you did not call it
+  - → system sends your response back to you → you must respond again
+  - → this repeats until you call `final_message`
+To break out of the loop, simply call `final_message` with no arguments.
+Only call `final_message` when you are done or already responded or stuck in a loop.
+"""
+
 agent = Agent(
     base_url=os.getenv("OPENAI_BASE_URL"),
     api_key=os.getenv("OPENAI_API_KEY"),
     model=os.getenv("OPENAI_MODEL"),
+    system_prompt=_SYSTEM_PROMPT,
 )
 agent.register_tool(FETCH_WEBPAGE_TOOL)
 agent.register_final_message_tool()
@@ -182,7 +198,7 @@ class GradioEvents:
         except Exception as exc:
             display_messages.append({
                 "role": "assistant",
-                "content": f'<span style="color: var(--color-red-600)">{exc}</span>',
+                "content": f'<span style="color: var(--color-red-400)">{exc}</span>',
                 "metadata": {"title": "💥 Error"},
             })
             yield {

requirements.txt

@@ -1,3 +1,6 @@
 gradio==6.8.0
 openai==2.41.0
 python-dotenv==1.2.2
+cloudscraper>=1.2,<2
+trafilatura>=2.0,<3
+markitdown>=0.1,<1