reduce upload_file_map completely

Author: howieleung

src/api/main.py

@@ -77,33 +77,6 @@ async def lifespan(app: fastapi.FastAPI):
         app.state.ai_client = ai_client
         app.state.agent = agent
 
-        
-        file_map: Dict[str, Dict[str, str]] = {}
-
-        logger.info(f"Creating UPLOADED_FILE_MAP")
-        folder_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'files'))
-        if agent.tool_resources and agent.tool_resources.file_search:
-            for vector_store_id in agent.tool_resources.file_search.vector_store_ids:
-                after: Optional[str] = None
-                has_more = True
-                while has_more:
-                    files = await ai_client.agents.list_vector_store_files(vector_store_id, after=after)
-                    has_more = files.has_more
-                    after = files.last_id
-                    for file in files.data:
-                        openAI_file = await ai_client.agents.get_file(file.id)
-                        logger.info(f"Retrieved file, file ID: {openAI_file.filename}")
-                        file_name = openAI_file.filename
-                        file_path = os.path.join(folder_path, file_name)
-
-                        if not os.path.exists(file_path):
-                            logger.warning(f"File path does not exist: {file_path}")
-                            continue 
-                                       
-                        # Store both file id and the file path using the file name as key.
-                        file_map[file_name] = {"id": file.id, "path": file_path}
-        app.state.upload_file_map = file_map
-        logger.info(f"Set UPLOADED_FILE_MAP {file_map}")                    
         yield
 
     except Exception as e:

src/api/routes.py

@@ -272,15 +272,14 @@ async def fetch_document(request: Request):
     if not file_name:
         raise HTTPException(status_code=400, detail="file_name is required")
 
-    files = request.app.state.upload_file_map
+    folder_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'files'))
 
-    logger.info(f"File requested: {file_name}. Current file keys: {list(files.keys())}")
+    file_path = os.path.join(folder_path, file_name)
 
-    if file_name not in files:
+    if not os.path.exists(file_path):
         raise HTTPException(status_code=404, detail="File not found")
 
     try:
-        file_path = files[file_name]["path"]
         data = await asyncio.to_thread(read_file, file_path)
         return PlainTextResponse(data)
     except Exception as e: