| |
| import json |
| import sys |
| from pathlib import Path |
|
|
| ROOT = Path("/home/harness_user_1/dgx_ai_factory") |
| ROUTER_JSON = ROOT / "router_templates/operational_router_v1.json" |
|
|
| def load_router(): |
| return json.loads(ROUTER_JSON.read_text(encoding="utf-8")) |
|
|
| def route(prompt: str): |
| router = load_router() |
| low = prompt.lower() |
| best = None |
| best_hits = 0 |
|
|
| for r in router.get("routes", []): |
| hits = sum(1 for kw in r.get("keywords_any", []) if kw.lower() in low) |
| if hits > best_hits: |
| best = r |
| best_hits = hits |
|
|
| if best and best_hits >= 1: |
| return { |
| "routed": True, |
| "route_id": best.get("id"), |
| "category": best.get("category"), |
| "hits": best_hits, |
| "answer": best.get("answer"), |
| } |
|
|
| return { |
| "routed": False, |
| "route_id": None, |
| "category": "model", |
| "hits": 0, |
| "answer": router.get("fallback", {}).get("answer", "ROUTE_TO_CHAMPION_MODEL"), |
| } |
|
|
| def main(): |
| if len(sys.argv) < 2: |
| print("usage: python scripts/120_operational_router.py '질문'") |
| node-7.example.invalid SystemExit(2) |
|
|
| prompt = " ".join(sys.argv[1:]) |
| result = route(prompt) |
|
|
| print("==== ROUTER RESULT ====") |
| print("routed:", result["routed"]) |
| print("route_id:", result["route_id"]) |
| print("category:", result["category"]) |
| print("hits:", result["hits"]) |
| print() |
| print(result["answer"]) |
|
|
| if __name__ == "__main__": |
| main() |
|
|
