fix: request timeout when openai completion (#1265)
This commit is contained in:
parent
5bffa1d918
commit
8606d80c66
2 changed files with 10 additions and 1 deletions
|
|
@ -209,6 +209,8 @@ class CompletionService:
|
|||
db.session.rollback()
|
||||
logging.exception("Unknown Error in completion")
|
||||
PubHandler.pub_error(user, generate_task_id, e)
|
||||
finally:
|
||||
db.session.close()
|
||||
|
||||
@classmethod
|
||||
def countdown_and_close(cls, worker_thread, pubsub, user, generate_task_id) -> threading.Thread:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue