Initial codes commit

service/search.py

@@ -3,10 +3,11 @@ from models.embedder import get_embedding
 from modules.retriever import retrieve_ids
 from modules.corpus import fetch_contexts_by_ids
 from models.reranker import rerank
+from fastapi import Request
 
-def search(query: str) -> list[dict]:
-    embedding = get_embedding(query)
+def search(request: Request, query: str) -> list[dict]:
+    embedding = get_embedding(request, query)
     ids = retrieve_ids(embedding)
     contexts = fetch_contexts_by_ids(ids)
-    reranked = rerank(query, contexts)
+    reranked = rerank(request, query, contexts)
     return reranked