From 8383520a3afe3be83336cdf64e3e556bab61b317 Mon Sep 17 00:00:00 2001 From: KeshavAnandCode Date: Sun, 19 Apr 2026 15:54:47 -0500 Subject: [PATCH] final basiclly --- mcp_server.py | 4 ++++ 1 file changed, 4 insertions(+) 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()