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