Deploy ArkAssistant smolagent (Hugging Face Agents Course Unit 1)

.agents/skills/deploy-space/SKILL.md

@@ -0,0 +1,41 @@
+---
+name: deploy-space
+description: Validates and deploys the ArkAssistant smolagent workspace to Hugging Face Spaces. Use when asked to test, validate, check, or deploy this project to Hugging Face Spaces.
+---
+
+# Deploy Space Skill
+
+This skill guides the automated validation and deployment process for the ArkAssistant Hugging Face Space repository (`Ark4u/First_agent_template`).
+
+## Pre-Deployment Verification Checklist
+
+Always run these verification steps prior to deployment:
+
+1. **Tool Sanity Verification**:
+   Ensure all external APIs and tools respond correctly without error:
+   ```bash
+   .venv/bin/python test_agent.py --test-tools
+   ```
+
+2. **Syntax Compilation Check**:
+   Confirm that all Python source files compile cleanly:
+   ```bash
+   .venv/bin/python -m py_compile app.py Gradio_UI.py deploy_space.py test_agent.py tools/*.py
+   ```
+
+3. **Dependency Check**:
+   Ensure all imported packages in `tools/` and `app.py` (e.g., `markdownify`, `smolagents`, `ddgs`, `litellm`) are declared in `requirements.txt`.
+
+4. **Ignore Patterns**:
+   Verify that sensitive files (`.env`), cache folders (`__pycache__`, `.gradio`), and the virtual environment (`.venv`) are listed in `deploy_space.py`'s `ignore_patterns`.
+
+## Deployment Execution
+
+Run the deployment script:
+```bash
+.venv/bin/python deploy_space.py
+```
+
+The script will prompt for confirmation or use the `HF_TOKEN` from the environment.
+Once uploaded, monitor the Space build logs at:
+https://huggingface.co/spaces/Ark4u/First_agent_template

AGENTS.md

@@ -1,32 +1,55 @@
-# OmniAssistant - Hugging Face Agents Course Project
+# ArkAssistant - Hugging Face Agents Course Project
 
-This workspace contains the complete implementation for the **Hugging Face Agents Course (Unit 1)** project.
+This workspace contains the complete implementation for the **Hugging Face Agents Course (Unit 1 & Unit 2)** project, featuring a Hierarchical Multi-Agent System built with `smolagents`.
+
+## System Architecture
+
+- **`ArkAssistant_Manager`** (`app.py`): Top-level orchestrator agent that parses complex user inquiries and coordinates specialist sub-agents.
+  - **`web_researcher`** (`research_agent`): Specialist agent equipped with `web_search`, `search_wikipedia`, and `visit_webpage` for deep web querying and documentation browsing.
+  - **`local_info_specialist`** (`local_info_agent`): Specialist agent equipped with `get_weather` and `get_current_time_in_timezone`.
+  - **`final_answer` & `image_generation_tool`**: Direct manager tools to conclude inquiries or generate visual images.
 
 ## Project Structure
 
-- `app.py`: Main entry point for the agent. Runs a `CodeAgent` with Gradio UI.
+- `app.py`: Main entry point and multi-agent pipeline with Gradio UI.
+- `test_agent.py`: Fast CLI test runner for direct queries and tool verification without launching a browser.
+- `Gradio_UI.py`: Gradio interface with ZeroGPU compatibility and step-by-step agent visualization.
 - `prompts.yaml`: System prompts and agent templates.
 - `requirements.txt`: Package dependencies for the agent and Hugging Face Space.
 - `README.md`: Space card metadata for Gradio on Hugging Face Spaces.
 - `deploy_space.py`: Automated deploy script to upload the workspace to `Ark4u/First_agent_template` on Hugging Face.
-- `tools/`: Modular tools provided to the agent:
-  - `web_search.py` (DuckDuckGo search)
-  - `get_weather.py` (Live weather via wttr.in)
-  - `search_wikipedia.py` (Wikipedia REST API)
-  - `get_current_time_in_timezone.py` (IANA timezone local time)
-  - `final_answer.py` (Final answer tool)
+- `tools/`: Modular tools:
+  - `web_search.py`: DuckDuckGo web search.
+  - `visit_webpage.py`: Webpage scraper and markdown converter (`VisitWebpageTool`).
+  - `search_wikipedia.py`: Wikipedia REST API lookups.
+  - `get_weather.py`: Live weather via wttr.in.
+  - `get_current_time_in_timezone.py`: IANA timezone local time converter.
+  - `final_answer.py`: Final response delivery tool.
 
 ## Environment & Commands
 
 - **Local Virtual Environment**: `.venv/bin/python`
-- **Run Gradio UI Locally**:
+
+- **Sanity Check All Tools (Fast, No LLM)**:
   ```bash
-  .venv/bin/python app.py
+  .venv/bin/python test_agent.py --test-tools
   ```
-- **Run with Local Ollama (Offline)**:
+
+- **Run CLI Agent Query**:
   ```bash
-  USE_OLLAMA=true .venv/bin/python app.py
+  .venv/bin/python test_agent.py "What is the capital of Japan and what is the weather there?"
   ```
+
+- **Run CLI Agent Query with Local Ollama (Offline)**:
+  ```bash
+  USE_OLLAMA=true .venv/bin/python test_agent.py "What is the time in UTC?"
+  ```
+
+- **Run Gradio Web UI Locally**:
+  ```bash
+  .venv/bin/python app.py
+  ```
+
 - **Deploy to Hugging Face Spaces**:
   ```bash
   .venv/bin/python deploy_space.py

Gradio_UI.py

@@ -37,6 +37,11 @@ except Exception:
     spaces = _MockSpaces()
 
 
+@spaces.GPU
+def _zero_gpu_init():
+    return "ready"
+
+
 def pull_messages_from_step(
     step_log: MemoryStep,
 ):
@@ -201,7 +206,6 @@ class GradioUI:
             if not os.path.exists(file_upload_folder):
                 os.mkdir(file_upload_folder)
 
-    @spaces.GPU(duration=120)
     def interact_with_agent(self, prompt, messages):
         import gradio as gr
 
@@ -302,6 +306,10 @@ class GradioUI:
                 [stored_messages, text_input],
             ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
 
+            # ZeroGPU initialization trigger
+            dummy_btn = gr.Button(visible=False)
+            dummy_btn.click(_zero_gpu_init)
+
         is_space = bool(os.getenv("SPACE_ID"))
         launch_kwargs = {
             "debug": True,

README.md

@@ -1,5 +1,5 @@
 ---
-title: OmniAssistant Smolagent
+title: ArkAssistant Smolagent
 emoji: 🤖
 colorFrom: indigo
 colorTo: purple
@@ -15,9 +15,9 @@ tags:
 - agent-course
 ---
 
-# 🤖 OmniAssistant - Hugging Face Agents Course (Unit 1)
+# 🤖 ArkAssistant - Hugging Face Agents Course (Unit 1)
 
-OmniAssistant is a multi-functional AI agent built with [`smolagents`](https://github.com/huggingface/smolagents) for Unit 1 of the [Hugging Face Agents Course](https://hf.co/learn/agents-course).
+ArkAssistant is a multi-functional AI agent built with [`smolagents`](https://github.com/huggingface/smolagents) for Unit 1 of the [Hugging Face Agents Course](https://hf.co/learn/agents-course).
 
 ## 🛠️ Capabilities & Tools
 

app.py

@@ -8,6 +8,7 @@ from dotenv import load_dotenv
 
 from smolagents import CodeAgent, DuckDuckGoSearchTool, load_tool, tool
 from tools.final_answer import FinalAnswerTool
+from tools.visit_webpage import VisitWebpageTool
 from Gradio_UI import GradioUI
 
 load_dotenv()
@@ -40,7 +41,7 @@ def get_weather(location: str) -> str:
         from urllib.parse import quote
         encoded_loc = quote(location)
         url = f"https://wttr.in/{encoded_loc}?format=%C+%t+(feels+like+%f)+wind:+%w"
-        headers = {"User-Agent": "HF-Agents-Course-OmniAssistant/1.0"}
+        headers = {"User-Agent": "HF-Agents-Course-ArkAssistant/1.0"}
         resp = requests.get(url, headers=headers, timeout=10)
         if resp.status_code == 200 and resp.text:
             return f"Weather for {location}: {resp.text.strip()}"
@@ -59,7 +60,7 @@ def search_wikipedia(query: str) -> str:
         from urllib.parse import quote
         encoded_query = quote(query)
         url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{encoded_query}"
-        headers = {"User-Agent": "HF-Agents-Course-OmniAssistant/1.0"}
+        headers = {"User-Agent": "HF-Agents-Course-ArkAssistant/1.0"}
         resp = requests.get(url, headers=headers, timeout=10)
         if resp.status_code == 200:
             data = resp.json()
@@ -79,6 +80,7 @@ def search_wikipedia(query: str) -> str:
 
 final_answer = FinalAnswerTool()
 web_search = DuckDuckGoSearchTool()
+visit_webpage = VisitWebpageTool()
 
 # Image generation tool from Hugging Face Hub
 image_generation_tool = None
@@ -131,11 +133,11 @@ with open(prompts_file, "r") as stream:
 # Specialist 1: Web & Wikipedia Researcher
 research_agent = CodeAgent(
     model=model,
-    tools=[web_search, search_wikipedia],
-    max_steps=4,
+    tools=[web_search, search_wikipedia, visit_webpage],
+    max_steps=5,
     verbosity_level=1,
     name="web_researcher",
-    description="A specialist researcher that searches DuckDuckGo and queries Wikipedia to gather accurate facts and summaries.",
+    description="A specialist researcher that searches DuckDuckGo, queries Wikipedia, and reads webpage content to gather accurate facts and summaries.",
     prompt_templates=prompt_templates
 )
 
@@ -162,7 +164,7 @@ manager_agent = CodeAgent(
     max_steps=8,
     verbosity_level=1,
     planning_interval=None,
-    name="OmniAssistant_Manager",
+    name="ArkAssistant_Manager",
     description="A senior manager agent that breaks down user tasks and delegates them to specialized team members (web researcher and local info specialist).",
     prompt_templates=prompt_templates
 )

test_agent.py

@@ -0,0 +1,94 @@
+#!/usr/bin/env python3
+"""
+CLI Test Runner for ArkAssistant smolagent
+Usage:
+    # Run a test query against the manager agent:
+    .venv/bin/python test_agent.py "What is the capital of France and what is the current time there?"
+
+    # Run with local Ollama offline model:
+    USE_OLLAMA=true .venv/bin/python test_agent.py "What is the weather in London?"
+
+    # Sanity check all tools without calling the LLM:
+    .venv/bin/python test_agent.py --test-tools
+"""
+
+import os
+import sys
+
+def run_tools_sanity():
+    print("=" * 60)
+    print("🔬 Running Sanity Check on ArkAssistant Tools (No LLM Calls)")
+    print("=" * 60)
+    
+    # 1. Weather
+    try:
+        from app import get_weather
+        w = get_weather("London")
+        print(f"✅ Weather Tool: {w[:80]}...")
+    except Exception as e:
+        print(f"❌ Weather Tool Failed: {e}")
+
+    # 2. Timezone
+    try:
+        from app import get_current_time_in_timezone
+        t = get_current_time_in_timezone("UTC")
+        print(f"✅ Timezone Tool: {t}")
+    except Exception as e:
+        print(f"❌ Timezone Tool Failed: {e}")
+
+    # 3. Wikipedia
+    try:
+        from app import search_wikipedia
+        wiki = search_wikipedia("Alan Turing")
+        print(f"✅ Wikipedia Tool: {wiki[:80]}...")
+    except Exception as e:
+        print(f"❌ Wikipedia Tool Failed: {e}")
+
+    # 4. Visit Webpage
+    try:
+        from app import visit_webpage
+        res = visit_webpage("https://httpbin.org/html")
+        print(f"✅ Visit Webpage Tool: Extracted {len(res)} characters.")
+    except Exception as e:
+        print(f"❌ Visit Webpage Tool Failed: {e}")
+
+    # 5. DuckDuckGo Web Search
+    try:
+        from app import web_search
+        sr = web_search("smolagents Hugging Face")
+        print(f"✅ Web Search Tool: {sr[:80]}...")
+    except Exception as e:
+        print(f"❌ Web Search Tool Failed: {e}")
+
+    print("=" * 60)
+    print("All tool sanity checks complete.")
+    print("=" * 60)
+
+def main():
+    args = sys.argv[1:]
+    
+    if "--test-tools" in args:
+        run_tools_sanity()
+        return
+
+    # Check for prompt argument
+    prompt = " ".join([a for a in args if not a.startswith("--")])
+    if not prompt:
+        prompt = "What is the current time in UTC and what is the weather in Paris?"
+        print(f"No prompt provided. Using default test prompt:\n  '{prompt}'\n")
+
+    print(f"🚀 Initializing ArkAssistant Agent...")
+    from app import agent
+
+    print(f"🤖 Running Agent Task: \"{prompt}\"")
+    print("-" * 60)
+    try:
+        response = agent.run(prompt)
+        print("-" * 60)
+        print(f"\n🎯 Agent Final Answer:\n{response}")
+    except Exception as e:
+        print(f"\n❌ Error during agent execution: {e}")
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main()

tools/get_weather.py

@@ -17,7 +17,7 @@ class SimpleTool(Tool):
             from urllib.parse import quote
             encoded_loc = quote(location)
             url = f"https://wttr.in/{encoded_loc}?format=%C+%t+(feels+like+%f)+wind:+%w"
-            headers = {"User-Agent": "HF-Agents-Course-OmniAssistant/1.0"}
+            headers = {"User-Agent": "HF-Agents-Course-ArkAssistant/1.0"}
             resp = requests.get(url, headers=headers, timeout=10)
             if resp.status_code == 200 and resp.text:
                 return f"Weather for {location}: {resp.text.strip()}"

tools/search_wikipedia.py

@@ -17,7 +17,7 @@ class SimpleTool(Tool):
             from urllib.parse import quote
             encoded_query = quote(query)
             url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{encoded_query}"
-            headers = {"User-Agent": "HF-Agents-Course-OmniAssistant/1.0"}
+            headers = {"User-Agent": "HF-Agents-Course-ArkAssistant/1.0"}
             resp = requests.get(url, headers=headers, timeout=10)
             if resp.status_code == 200:
                 data = resp.json()

tools/visit_webpage.py

@@ -43,4 +43,5 @@ class VisitWebpageTool(Tool):
             return f"An unexpected error occurred: {str(e)}"
 
     def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
         self.is_initialized = False