Pārlūkot izejas kodu

fix: image model check

zhayujie 1 gadu atpakaļ
vecāks
revīzija
fa3370df5b
1 mainītis faili ar 3 papildinājumiem un 0 dzēšanām
  1. 3 0
      common/linkai_client.py

+ 3 - 0
common/linkai_client.py

@@ -60,6 +60,9 @@ class ChatClient(LinkAIClient):
             if pconf("linkai")["midjourney"]:
                 pconf("linkai")["midjourney"]["enabled"] = True
                 pconf("linkai")["midjourney"]["use_image_create_prefix"] = True
+        elif config.get("text_to_image") and config.get("text_to_image") in ["dall-e-2", "dall-e-3"]:
+            if pconf("linkai")["midjourney"]:
+                pconf("linkai")["midjourney"]["use_image_create_prefix"] = False
 
 
 def start(channel):