diff --git a/mcp_server.py b/mcp_server.py index 3f112bd..a98c274 100644 --- a/mcp_server.py +++ b/mcp_server.py @@ -18,6 +18,7 @@ import uvicorn from starlette.middleware.cors import CORSMiddleware from starlette.middleware.base import BaseHTTPMiddleware from starlette.requests import Request +import httpx from mcp.server.transport_security import TransportSecuritySettings, TransportSecurityMiddleware TransportSecurityMiddleware.__init__ = lambda self, settings=None: setattr( @@ -39,6 +40,8 @@ with open(project_root / "data" / "processed" / "parent_lookup.json") as f: # ── Config ───────────────────────────────────────────────────────────────── TOP_K = 10 + + # ── Embed ────────────────────────────────────────────────────────────────── def embed_query(query: str) -> np.ndarray: return model.encode( @@ -160,6 +163,7 @@ def search_textbook(query: str) -> str: return header + passages + footer + # ── Run ──────────────────────────────────────────────────────────────────── if __name__ == "__main__": app = mcp.streamable_http_app()