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