Spaces:
Running
Running
| """Small setup diagnostic for the local Ollama-based GAIA agent.""" | |
| from __future__ import annotations | |
| import os | |
| import sys | |
| import requests | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| BASE_URL = os.getenv("OLLAMA_BASE_URL", "http://localhost:11434").rstrip("/") | |
| MODEL = os.getenv("OLLAMA_MODEL", "qwen3:4b").strip() | |
| def main() -> int: | |
| print("GAIA Unit 4 - Local Setup Check\n") | |
| print(f"Ollama URL: {BASE_URL}") | |
| print(f"Ollama model: {MODEL}") | |
| try: | |
| response = requests.get(f"{BASE_URL}/api/tags", timeout=5) | |
| response.raise_for_status() | |
| except Exception as exc: | |
| print(f"\n[FAIL] Ollama is not reachable: {exc}") | |
| print("Start Ollama, then run this check again.") | |
| return 1 | |
| models = [item.get("name", "") for item in response.json().get("models", [])] | |
| print("\n[OK] Ollama is running.") | |
| if not models: | |
| print("[FAIL] No Ollama models are installed.") | |
| print(f"Run: ollama pull {MODEL}") | |
| return 1 | |
| print("Installed models:") | |
| for installed in models: | |
| print(f" - {installed}") | |
| requested_base = MODEL.split(":", 1)[0] | |
| found = any( | |
| installed == MODEL | |
| or installed.removesuffix(":latest") == MODEL | |
| or installed.split(":", 1)[0] == requested_base | |
| for installed in models | |
| ) | |
| if not found: | |
| print(f"\n[FAIL] Configured model '{MODEL}' was not found.") | |
| print("Either pull it or update OLLAMA_MODEL in .env.") | |
| return 1 | |
| print(f"\n[OK] Configured model '{MODEL}' is available.") | |
| print("[OK] Basic local setup is ready.") | |
| return 0 | |
| if __name__ == "__main__": | |
| sys.exit(main()) | |