Spaces:
Sleeping
Sleeping
| import json | |
| import os | |
| import uuid | |
| from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer | |
| from pathlib import Path | |
| from urllib.parse import urlparse | |
| from agent import run_agent_mode, tool_search_kb | |
| HERE = Path(__file__).resolve().parent | |
| WEB_DIR = HERE / "web" | |
| OUT_DIR = HERE / "out" | |
| KB_DIR = HERE / "kb" | |
| def _read_body_json(handler: BaseHTTPRequestHandler): | |
| length = int(handler.headers.get("Content-Length") or "0") | |
| raw = handler.rfile.read(length) if length > 0 else b"" | |
| if not raw: | |
| return {} | |
| try: | |
| return json.loads(raw.decode("utf-8")) | |
| except Exception: | |
| return None | |
| class Handler(BaseHTTPRequestHandler): | |
| server_version = "AgentDemo/1.0" | |
| def log_message(self, format, *args): | |
| return | |
| def _send(self, status: int, content_type: str, body: bytes): | |
| self.send_response(status) | |
| self.send_header("Content-Type", content_type) | |
| self.send_header("Content-Length", str(len(body))) | |
| self.send_header("Cache-Control", "no-store") | |
| self.end_headers() | |
| self.wfile.write(body) | |
| def _send_json(self, status: int, obj): | |
| body = json.dumps(obj, ensure_ascii=False).encode("utf-8") | |
| self._send(status, "application/json; charset=utf-8", body) | |
| def do_GET(self): | |
| u = urlparse(self.path) | |
| path = u.path or "/" | |
| if path == "/": | |
| p = WEB_DIR / "index.html" | |
| self._send(200, "text/html; charset=utf-8", p.read_bytes()) | |
| return | |
| if path == "/assets/app.js": | |
| p = WEB_DIR / "app.js" | |
| self._send(200, "application/javascript; charset=utf-8", p.read_bytes()) | |
| return | |
| if path == "/assets/app.css": | |
| p = WEB_DIR / "app.css" | |
| self._send(200, "text/css; charset=utf-8", p.read_bytes()) | |
| return | |
| if path == "/api/health": | |
| self._send_json(200, {"ok": True, "service": "agent-demo", "version": "1.0"}) | |
| return | |
| if path == "/api/kb/list": | |
| items = [] | |
| for p in sorted(KB_DIR.glob("**/*.md")): | |
| try: | |
| txt = p.read_text(encoding="utf-8").splitlines() | |
| title = "" | |
| for ln in txt[:10]: | |
| ln = ln.strip() | |
| if ln.startswith("#"): | |
| title = ln.lstrip("#").strip() | |
| break | |
| items.append({"path": str(p.relative_to(HERE)), "title": title or p.name}) | |
| except Exception: | |
| continue | |
| self._send_json(200, {"ok": True, "items": items}) | |
| return | |
| self._send(404, "text/plain; charset=utf-8", "not found".encode("utf-8")) | |
| def do_POST(self): | |
| u = urlparse(self.path) | |
| path = u.path or "/" | |
| if path == "/api/kb/search": | |
| payload = _read_body_json(self) | |
| if payload is None: | |
| self._send_json(400, {"ok": False, "error": "请求 JSON 解析失败"}) | |
| return | |
| query = str(payload.get("query") or "").strip() | |
| if not query: | |
| self._send_json(400, {"ok": False, "error": "query 不能为空"}) | |
| return | |
| res = tool_search_kb(query=query, kb_dir=str(KB_DIR), top_k=int(payload.get("top_k") or 6)) | |
| if not res.ok: | |
| self._send_json(500, {"ok": False, "error": res.output}) | |
| return | |
| try: | |
| items = json.loads(res.output) | |
| except Exception: | |
| items = [] | |
| self._send_json(200, {"ok": True, "items": items}) | |
| return | |
| if path != "/api/run": | |
| self._send(404, "text/plain; charset=utf-8", "not found".encode("utf-8")) | |
| return | |
| payload = _read_body_json(self) | |
| if payload is None: | |
| self._send_json(400, {"ok": False, "error": "请求 JSON 解析失败"}) | |
| return | |
| mode = str(payload.get("mode") or "general").strip() | |
| if mode == "lead_followup": | |
| lead = payload.get("lead") or {} | |
| if not isinstance(lead, dict): | |
| self._send_json(400, {"ok": False, "error": "lead 必须是对象"}) | |
| return | |
| company = str(lead.get("company") or "").strip() | |
| pain = str(lead.get("pain_points") or "").strip() | |
| if not company and not pain: | |
| self._send_json(400, {"ok": False, "error": "至少填写“公司/组织”或“主要诉求/痛点”"}) | |
| return | |
| else: | |
| goal = str(payload.get("goal") or "").strip() | |
| if not goal: | |
| self._send_json(400, {"ok": False, "error": "goal 不能为空"}) | |
| return | |
| if len(goal) > 600: | |
| self._send_json(400, {"ok": False, "error": "goal 过长（最多 600 字符）"}) | |
| return | |
| OUT_DIR.mkdir(parents=True, exist_ok=True) | |
| report_name = f"report_{uuid.uuid4().hex[:12]}.md" | |
| out_report = str(OUT_DIR / report_name) | |
| try: | |
| result = run_agent_mode(mode=mode, payload=payload, kb_dir=str(KB_DIR), out_report=out_report, max_rounds=3) | |
| except Exception as e: | |
| self._send_json(500, {"ok": False, "error": f"执行失败: {e}"}) | |
| return | |
| try: | |
| report_text = Path(out_report).read_text(encoding="utf-8") if os.path.exists(out_report) else "" | |
| except Exception: | |
| report_text = "" | |
| self._send_json( | |
| 200, | |
| { | |
| "ok": True, | |
| "mode": result.get("mode") or mode, | |
| "goal": result.get("goal"), | |
| "final_answer": result.get("final_answer"), | |
| "round_logs": result.get("round_logs"), | |
| "structured": result.get("structured"), | |
| "artifact_path": out_report, | |
| "report_text": report_text, | |
| }, | |
| ) | |
| def main(): | |
| host = "0.0.0.0" | |
| if "PORT" in os.environ: | |
| port = int(os.environ["PORT"]) | |
| httpd = ThreadingHTTPServer((host, port), Handler) | |
| print(f"本地演示页已启动： http://127.0.0.1:{port}/") | |
| httpd.serve_forever() | |
| return | |
| in_space = any(k.startswith("SPACE_") for k in os.environ.keys()) | |
| base_port = 7860 if in_space else 8000 | |
| last_err = None | |
| for port in range(base_port, base_port + 20): | |
| try: | |
| httpd = ThreadingHTTPServer((host, port), Handler) | |
| print(f"本地演示页已启动： http://127.0.0.1:{port}/") | |
| httpd.serve_forever() | |
| return | |
| except OSError as e: | |
| last_err = e | |
| continue | |
| raise SystemExit(f"无法绑定端口（从 {base_port} 起尝试 20 个）：{last_err}") | |
| if __name__ == "__main__": | |
| main() | |