minor bug fix

app.py

@@ -3,6 +3,7 @@ import helper
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
+from typing import List
 import os
 from dotenv import load_dotenv
 
@@ -13,7 +14,7 @@ app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
     allow_credentials=True,
-    allow_methods=["GET", "POST"],
+    allow_methods=["*"],
     allow_headers=["*"]
 )
 
@@ -26,6 +27,7 @@ class RequestData(BaseModel):
 
 class ResponseData(BaseModel):
     summary: str
+    ids: List[str]
 
 @app.post("/generateLiteratureSurvey/", response_model=ResponseData)
 async def generate_literature_survey(request_data: RequestData):
@@ -40,6 +42,12 @@ async def root():
       return {"status": 0}
     return {"status": 1}
 
+@app.get("/test")
+async def root():
+    if llms == None :
+      return {"status": 0}
+    return {"status": 1}
+
 def summarize(query, llms) :
    keywords = helper.extract_keywords(llms['feature_extractor'], query)
    papers = helper.search_papers(llms['arxiv_agent'], keywords)