浏览代码

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

tags/0.3.24
takatost 2 年前
父节点
当前提交
a31466d34e
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 5 次插入0 次删除
  1. 5
    0
      api/core/model_providers/models/llm/base.py

+ 5
- 0
api/core/model_providers/models/llm/base.py 查看文件

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
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__)




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:

正在加载...
取消
保存