Update app.py

app.py

@@ -87,12 +87,8 @@ async def startup_event():
     global agent
     try:
         logger.info("Initializing TxAgent...")
-        # Get Hugging Face token from environment variable
-        hf_token = os.getenv("HUGGINGFACE_TOKEN")
-        model_kwargs = {"token": hf_token} if hf_token else {}
-        
         agent = TxAgent(
-            model_name="mims-harvard/TxAgent-T1-Llama-3.1-8B",  # Corrected model name
+            model_name="mims-harvard/TxAgent-T1-Llama-3.1-8B",
             rag_model_name="mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B",
             tool_files_dict={},
             enable_finish=True,
@@ -100,23 +96,17 @@ async def startup_event():
             force_finish=True,
             enable_checker=True,
             step_rag_num=4,
-            seed=100,
-            **model_kwargs
+            seed=100
         )
         agent.init_model()
         logger.info("TxAgent initialized successfully")
     except Exception as e:
         logger.error(f"Failed to initialize agent: {str(e)}")
-        # Allow app to start, but endpoints will return errors if agent is None
-        agent = None
+        raise RuntimeError(f"Failed to initialize agent: {str(e)}")
 
 @app.post("/chat")
 async def chat_endpoint(request: ChatRequest):
     """Handle chat conversations with formatting options"""
-    if agent is None:
-        logger.error("TxAgent not initialized")
-        raise HTTPException(status_code=503, detail="TxAgent service unavailable")
-    
     try:
         logger.info(f"Chat request received (format: {request.format})")
         raw_response = agent.chat(
@@ -146,10 +136,6 @@ async def chat_endpoint(request: ChatRequest):
 @app.post("/upload")
 async def upload_file(file: UploadFile = File(...)):
     """Handle file uploads and process with TxAgent"""
-    if agent is None:
-        logger.error("TxAgent not initialized")
-        raise HTTPException(status_code=503, detail="TxAgent service unavailable")
-    
     try:
         logger.info(f"File upload received: {file.filename}")
         content = ""
@@ -192,7 +178,7 @@ async def upload_file(file: UploadFile = File(...)):
 async def service_status():
     """Check service status"""
     return {
-        "status": "running" if agent else "failed",
+        "status": "running",
         "version": "2.0.0",
         "model": agent.model_name if agent else "not loaded",
         "formats_available": ["raw", "clean", "structured", "html"],