| 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: |