final basiclly

This commit is contained in:
2026-04-19 15:54:47 -05:00
parent 7935b01771
commit 8383520a3a

View File

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