Update app.py

app.py

@@ -17,6 +17,10 @@ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 class ModifyQueryRequest(BaseModel):
     query_string: str
 
+# Define request models
+class ModifyQueryRequest_v3(BaseModel):
+    query_string_list: [str]
+
 class AnswerQuestionRequest(BaseModel):
     question: str
     context: list
@@ -78,11 +82,11 @@ async def t5answer(request: T5QuestionRequest):
 
 
 # Define API endpoints
-@app.post("/modify_query2", response_model=ModifyQueryResponse)
-async def modify_query2(request: ModifyQueryRequest):
+@app.post("/modify_query_v3", response_model=ModifyQueryResponse)
+async def modify_query2(request: ModifyQueryRequest_v3):
     try:
-        embeddings = model.encode([request.query_string])
-        return ModifyQueryResponse(embeddings=embeddings[0].tolist())
+        embeddings = model.encode(request.query_string_list)
+        return ModifyQueryResponse(embeddings=[emb.tolist() for emb in embeddings])
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))