diff --git a/main.py b/main.py index 481ee55..d500ca2 100644 --- 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: