| 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() |
|
|
