hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-10 11:26:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: db session not commit before long llm call running (#1251)

Browse Source
This commit is contained in:
takatost
2023-09-27 21:40:26 +08:00
committed by GitHub
parent d38eac959b
commit a31466d34e
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/model_providers/models/llm/base.py
View File

@@ -1,6 +1,7 @@
import json import json
import os import os
import re import re
import time
from abc import abstractmethod from abc import abstractmethod
from typing import List, Optional, Any, Union, Tuple from typing import List, Optional, Any, Union, Tuple
import decimal import decimal
@@ -20,6 +21,8 @@ from core.prompt.prompt_template import JinjaPromptTemplate
from core.third_party.langchain.llms.fake import FakeLLM from core.third_party.langchain.llms.fake import FakeLLM
import logging import logging
from extensions.ext_database import db
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -129,6 +132,8 @@ class BaseLLM(BaseProviderModel):
if self.deduct_quota: if self.deduct_quota:
self.model_provider.check_quota_over_limit() self.model_provider.check_quota_over_limit()
db.session.commit()
if not callbacks: if not callbacks:
callbacks = self.callbacks callbacks = self.callbacks
else: else: