final basiclly
This commit is contained in:
@@ -18,6 +18,7 @@ import uvicorn
|
|||||||
from starlette.middleware.cors import CORSMiddleware
|
from starlette.middleware.cors import CORSMiddleware
|
||||||
from starlette.middleware.base import BaseHTTPMiddleware
|
from starlette.middleware.base import BaseHTTPMiddleware
|
||||||
from starlette.requests import Request
|
from starlette.requests import Request
|
||||||
|
import httpx
|
||||||
from mcp.server.transport_security import TransportSecuritySettings, TransportSecurityMiddleware
|
from mcp.server.transport_security import TransportSecuritySettings, TransportSecurityMiddleware
|
||||||
|
|
||||||
TransportSecurityMiddleware.__init__ = lambda self, settings=None: setattr(
|
TransportSecurityMiddleware.__init__ = lambda self, settings=None: setattr(
|
||||||
@@ -39,6 +40,8 @@ with open(project_root / "data" / "processed" / "parent_lookup.json") as f:
|
|||||||
# ── Config ─────────────────────────────────────────────────────────────────
|
# ── Config ─────────────────────────────────────────────────────────────────
|
||||||
TOP_K = 10
|
TOP_K = 10
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# ── Embed ──────────────────────────────────────────────────────────────────
|
# ── Embed ──────────────────────────────────────────────────────────────────
|
||||||
def embed_query(query: str) -> np.ndarray:
|
def embed_query(query: str) -> np.ndarray:
|
||||||
return model.encode(
|
return model.encode(
|
||||||
@@ -160,6 +163,7 @@ def search_textbook(query: str) -> str:
|
|||||||
|
|
||||||
return header + passages + footer
|
return header + passages + footer
|
||||||
|
|
||||||
|
|
||||||
# ── Run ────────────────────────────────────────────────────────────────────
|
# ── Run ────────────────────────────────────────────────────────────────────
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
app = mcp.streamable_http_app()
|
app = mcp.streamable_http_app()
|
||||||
|
|||||||
Reference in New Issue
Block a user