Spaces:
Running on Zero
Running on Zero
| #!/usr/bin/env python3 | |
| """CLI test runner for ArkAssistant. | |
| 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?" | |
| # Sanity check all tools without calling the LLM: | |
| .venv/bin/python test_agent.py --test-tools | |
| The project uses the Hugging Face Inference API and requires HF_TOKEN in .env. | |
| """ | |
| import sys | |
| from tools.get_current_time_in_timezone import get_current_time_in_timezone | |
| from tools.get_weather import get_weather | |
| from tools.search_wikipedia import search_wikipedia | |
| from tools.visit_webpage import VisitWebpageTool | |
| from tools.web_search import DuckDuckGoSearchTool | |
| def _require_result(result: str) -> str: | |
| text = str(result).strip() | |
| if not text: | |
| raise RuntimeError("tool returned an empty result") | |
| if text.lower().startswith("error"): | |
| raise RuntimeError(text) | |
| return text | |
| def run_tools_sanity() -> int: | |
| print("=" * 60) | |
| print("🔬 Running Sanity Check on ArkAssistant Tools (No LLM Calls)") | |
| print("=" * 60) | |
| visit_webpage = VisitWebpageTool() | |
| web_search = DuckDuckGoSearchTool() | |
| failures: list[str] = [] | |
| checks = [ | |
| ("Weather", lambda: get_weather("London")), | |
| ("Timezone", lambda: get_current_time_in_timezone("UTC")), | |
| ("Wikipedia", lambda: search_wikipedia("Alan Turing")), | |
| ("Visit Webpage", lambda: visit_webpage("https://httpbin.org/html")), | |
| ("Web Search", lambda: web_search("smolagents Hugging Face")), | |
| ] | |
| for name, check in checks: | |
| try: | |
| result = _require_result(check()) | |
| preview = result.replace("\n", " ")[:100] | |
| print(f"✅ {name} Tool: {preview}") | |
| except Exception as exc: | |
| failures.append(f"{name}: {exc}") | |
| print(f"❌ {name} Tool Failed: {exc}") | |
| print("=" * 60) | |
| if failures: | |
| print(f"❌ {len(failures)} tool check(s) failed.") | |
| for failure in failures: | |
| print(f" - {failure}") | |
| return 1 | |
| print("✅ All tool sanity checks passed.") | |
| return 0 | |
| def main() -> int: | |
| args = sys.argv[1:] | |
| if "--test-tools" in args: | |
| return run_tools_sanity() | |
| prompt = " ".join(arg for arg in args if not arg.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("🚀 Initializing ArkAssistant Agent...") | |
| try: | |
| from app import agent | |
| print(f"🤖 Running Agent Task: \"{prompt}\"") | |
| print("-" * 60) | |
| response = agent.run(prompt) | |
| print("-" * 60) | |
| print(f"\n🎯 Agent Final Answer:\n{response}") | |
| return 0 | |
| except Exception as exc: | |
| message = str(exc) | |
| print(f"\n❌ Error during agent execution: {message}") | |
| if "402" in message or "Payment Required" in message: | |
| print( | |
| "Hint: Hugging Face Inference has no available credits. " | |
| "Add credits/PRO or configure another cloud provider in .env." | |
| ) | |
| return 1 | |
| if __name__ == "__main__": | |
| sys.exit(main()) | |