Преглед изворни кода

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

John Wang пре 1 година
родитељ
комит
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 = {