added timeout / seems to be working

2025-05-01 03:09:35 +01:00
parent e48b5b3d4e
commit cf1e389df1
1 changed files with 2 additions and 1 deletions
--- a/main.py
+++ b/main.py
@@ -80,7 +80,8 @@ async def message_callback(room: MatrixRoom, event: RoomMessageText):
                    {"role": "system", "content": "You are a helpful assistant."},
                    {"role": "user",   "content": question}
                ],
-                max_tokens=150
+                max_tokens=150,
+                timeout=300
            )
            answer = response.choices[0].message.content.strip()
        except Exception: