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

fix: image create prefix match bug

zhayujie 3 лет назад
Родитель
Сommit
17b6c63fb9
2 измененных файлов с 1 добавлено и 3 удалено
  1. 0 2
      bot/openai/open_ai_bot.py
  2. 1 1
      channel/wechat/wechat_channel.py

+ 0 - 2
bot/openai/open_ai_bot.py

@@ -53,7 +53,6 @@ class OpenAIBot(Bot):
         return image_url
 
     def edit_img(self, query, src_img):
-        openai.api_key = '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'
         try:
             response = openai.Image.create_edit(
                 image=open(src_img, 'rb'),
@@ -70,7 +69,6 @@ class OpenAIBot(Bot):
         return image_url
 
     def migration_img(self, query, src_img):
-        openai.api_key = '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'
 
         try:
             response = openai.Image.create_variation(

+ 1 - 1
channel/wechat/wechat_channel.py

@@ -52,7 +52,7 @@ class WechatChannel(Channel):
             if len(str_list) == 2:
                 content = str_list[1].strip()
 
-            img_match_prefix = self.check_prefix(content, ["画图"])
+            img_match_prefix = self.check_prefix(content, conf().get('image_create_prefix'))
             if img_match_prefix:
                 content = content.split(img_match_prefix, 1)[1].strip()
                 thead_pool.submit(self._do_send_img, content, from_user_id)