Procházet zdrojové kódy

fix: quota update error on azure openai (#172)

John Wang před 1 rokem
rodič
revize
056898bf21
1 změnil soubory, kde provedl 3 přidání a 0 odebrání
  1. 3 0
      api/core/conversation_message_task.py

+ 3 - 0
api/core/conversation_message_task.py

@@ -56,6 +56,9 @@ class ConversationMessageTask:
         )
 
     def init(self):
+        provider_name = LLMBuilder.get_default_provider(self.app.tenant_id)
+        self.model_dict['provider'] = provider_name
+
         override_model_configs = None
         if self.is_override:
             override_model_configs = {