Update main.py

main.py

@@ -185,20 +185,15 @@ def chat(
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error processing chat query: {str(e)}")
 
-    # Optionally update the user's chat_histories with the new messages here
-
     return StreamingResponse(stream_generator, media_type="text/event-stream")
 
 # ----------------------- Remaining Endpoints -----------------------
 @app.post("/add_document")
 async def add_document(
-    file: Optional[Any] = File(None),
+    file: Optional[UploadFile] = File(None),  # Changed type to UploadFile
     wiki_query: Optional[str] = Query(None),
     wiki_url: Optional[str] = Query(None)
 ):
-    if not isinstance(file, UploadFile):
-        file = None
-
     if file is None and wiki_query is None and wiki_url is None:
         raise HTTPException(status_code=400, detail="No document input provided (file, wiki_query, or wiki_url).")