fix timeout issue (see: openai/openai-python#387)

zzh1996 · Apr 27, 2023 · a550b35 · a550b35
1 parent 155764e
commit a550b35
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,2 +1,2 @@
 FROM python:3.11
-RUN pip install python-telegram-bot==20.2 openai==0.27.2
+RUN pip install python-telegram-bot==20.2 git+https://github.com/zzh1996/openai-python.git@fix
diff --git a/main.py b/main.py
@@ -146,7 +146,7 @@ async def completion(chat_history, model, chat_id, msg_id): # chat_history = [us
         messages.append({"role": roles[role_id], "content": msg})
         role_id = 1 - role_id
     logging.info('Request (chat_id=%r, msg_id=%r): %s', chat_id, msg_id, messages)
-    stream = await openai.ChatCompletion.acreate(model=model, messages=messages, stream=True)
+    stream = await openai.ChatCompletion.acreate(model=model, messages=messages, stream=True, request_timeout=15)
     async for response in stream:
         logging.info('Response (chat_id=%r, msg_id=%r): %s', chat_id, msg_id, json.dumps(response, ensure_ascii=False))
         obj = response['choices'][0]