fix: request timeout when openai completion (#1265)

2023-10-01 16:00:23 +08:00
parent 5bffa1d918
commit 8606d80c66
2 changed files with 10 additions and 1 deletions
--- a/api/services/completion_service.py
+++ b/api/services/completion_service.py
@@ -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: