Update app.py

app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, Request, Query
 from fastapi.templating import Jinja2Templates
 from fastapi import File, UploadFile
 from fastapi.responses import FileResponse
+from fastapi.responses import Response
 
 from pydantic import BaseModel
 from sentence_transformers import SentenceTransformer
@@ -43,7 +44,7 @@ def embed_strings(request: EmbedRequest):
 def search_string(request: SearchRequest):
     embedding = model.encode([request.text])
     distances, indices = index.search(np.array(embedding), request.n)
-    found_documents = index.reconstruct_n(indices[0], request.n)
+    found_documents = index.reconstruct_n(indices[0], int(request.n))
     return {
         "distances": distances[0].tolist(),
         "indices": indices[0].tolist(),