Archived
final basiclly
This commit is contained in:
1 parent
7935b01771
commit
8383520a3a
1 file changed
+4
@@ -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