final basiclly
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user