add timeout

app.py

@@ -48,6 +48,7 @@ def get_agent_model(model_name: str, oauth_token: gr.OAuthToken = None):
     model = InferenceClientModel( 
         model_id=model_name, 
         token=token,
+        timeout=1000,
     )
     return model
 

src/upgrade_advisor/chat/chat.py

@@ -31,7 +31,7 @@ async def query(payload, token=None):
         "Authorization": f"Bearer {token}",
     }
     try:
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=300)
+        response = requests.post(API_URL, headers=headers, json=payload, timeout=1000)
     except Exception as e:
         logger.error(f"Error during API request: {e}")
         raise e