Просмотр исходного кода

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

John Wang 1 год назад
Родитель
Сommit
056898bf21
1 измененных файлов с 3 добавлено и 0 удалено
  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 = {