From 07a65a20823ff2340f48b9287c8276a79750a625 Mon Sep 17 00:00:00 2001
From: phact <estevezsebastian@gmail.com>
Date: Thu, 10 Jul 2025 23:33:16 -0400
Subject: [PATCH] add search endpoint

---
 src/app.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/src/app.py b/src/app.py
index 3277083d..db23646e 100644
--- a/src/app.py
+++ b/src/app.py
@@ -211,9 +211,42 @@ async def upload_path(request: Request):
     results = await asyncio.gather(*tasks)
     return JSONResponse({"results": results})
 
+async def search(request: Request):
+    payload = await request.json()
+    query = payload.get("query")
+    if not query:
+        return JSONResponse({"error": "Query is required"}, status_code=400)
+
+    # Embed the query
+    resp = client.embeddings.create(model=EMBED_MODEL, input=[query])
+    query_embedding = resp.data[0].embedding
+
+    # Search using vector similarity
+    search_body = {
+        "query": {
+            "nested": {
+                "path": "chunks",
+                "query": {
+                    "knn": {
+                        "chunks.chunk_embedding": {
+                            "vector": query_embedding,
+                            "k": 10
+                        }
+                    }
+                }
+            }
+        },
+        "_source": ["chunks.text", "chunks.page", "filename", "mimetype"],
+        "size": 10
+    }
+
+    results = await es.search(index=INDEX_NAME, body=search_body)
+    return JSONResponse({"results": results["hits"]["hits"]})
+
 app = Starlette(debug=True, routes=[
     Route("/upload",      upload,       methods=["POST"]),
     Route("/upload_path", upload_path,  methods=["POST"]),
+    Route("/search",      search,       methods=["POST"]),
 ])
 
 if __name__ == "__main__":