Lil fix

BasedHardware · Sep 22, 2024 · da48add · da48add
1 parent 5a94bc2
commit da48add
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/plugins/example/_multion/router.py b/plugins/example/_multion/router.py
@@ -169,7 +169,14 @@ async def multion_endpoint(memory: Memory, uid: str = Query(...)):
     if not books:
         return EndpointResponse(message='No books were suggested or mentioned.')
 
-    result = await call_multion(books, user_id)
+    try:
+        result = await asyncio.wait_for(call_multion(books, user_id), timeout=120)
+    except asyncio.TimeoutError:
+        print("Timeout error occurred")
+        return EndpointResponse(message="Timeout error occurred.")
+    except Exception as e:
+        print(f"Error calling Multion API: {str(e)}")
+        return EndpointResponse(message=f"Error calling Multion API: {str(e)}")
     if isinstance(result, bytes):
         result = result.decode('utf-8')