Add the agent mcp

app.py

@@ -1,5 +1,7 @@
 # %%
+import requests
 from bs4 import BeautifulSoup
+import gradio as gr
 
 def parse_news_item(html: str) -> dict:
     soup = BeautifulSoup(html, "html.parser")
@@ -29,9 +31,6 @@ def parse_news_item(html: str) -> dict:
 
 
 # %%
-import requests
-from bs4 import BeautifulSoup
-
 def search_news(keyword, page=1):
     """
     Fetch news articles related to a keyword from udn.com.
@@ -98,6 +97,35 @@ def get_content(url):
         'text': text_content
     }
 
+# %%
+from smolagents import Tool, CodeAgent, LiteLLMModel, ToolCollection, ActionStep, FinalAnswerStep
+import os
+
+model = LiteLLMModel("openrouter/qwen/qwen-2.5-coder-32b-instruct:free", api_key=os.environ["OPENROUTER_API_KEY"])
+url = "https://robin0307-newsmcp.hf.space/gradio_api/mcp/sse"
+server_parameters = {"url": url, "transport": "sse"}
+
+def newsAgent(task: str) -> str:
+    """
+    News Agent to handle the news task.
+    
+    Args:
+        task: The task description.
+
+    Returns:
+        The result of the Task.
+    """
+    result = ""
+    with ToolCollection.from_mcp(server_parameters, trust_remote_code=True) as mcp_tools:
+        agent = CodeAgent(tools=[*mcp_tools.tools], model=model)
+        for event in agent.run(task, stream=True, max_steps=5):
+            if isinstance(event, ActionStep):
+                result = f"\n======Step {event.step_number}======\n{event.code_action}"
+                # yield result
+            if isinstance(event, FinalAnswerStep):
+                result = f"\n======Final======\n{event.output}"
+                # yield result
+    return result
 
 # get_content('https://money.udn.com/money/story/5612/8832289?from=edn_search_result')  # Example usage to fetch content from a specific URL
 
@@ -105,7 +133,6 @@ def get_content(url):
 # using the gradio to create two tab 
 # 1. search news
 # 2. get content from url
-import gradio as gr
 def main():
     with gr.Blocks() as demo:
         gr.Markdown("# News Search and Content Fetcher")
@@ -121,11 +148,15 @@ def main():
         with gr.Tab("Get Content from URL"):
             url_input = gr.Textbox(label="URL", placeholder="Enter URL to fetch content")
             content_output = gr.JSON(label="Content Output")
-            
             url_input.submit(get_content, inputs=url_input, outputs=content_output)
-    demo.launch(mcp_server=True, server_name="0.0.0.0",allowed_paths=["/"])
 
-if __name__ == "__main__":
-    main()
+        with gr.Tab("News Agent"):
+            agent_input = gr.Textbox(label="Task", placeholder="Enter the task")
+            run_button = gr.Button("Run")
+            result_output = gr.Textbox(label="Result", lines=10)
+            run_button.click(newsAgent, inputs=agent_input, outputs=result_output)
 
+    demo.launch(mcp_server=True, server_name="0.0.0.0",allowed_paths=["/"], share=True)
 
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -7,4 +7,6 @@ requires-python = ">=3.13"
 dependencies = [
     "bs4>=0.0.2",
     "gradio[mcp]>=5.33.1",
+    "litellm>=1.74.1",
+    "smolagents[litellm,mcp]>=1.20.0",
 ]

requirements.txt

@@ -1,2 +1,3 @@
 gradio[mcp]
 bs4
+smolagents[litellm,mcp]