kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-20 13:47:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: image create prefix match bug

Browse Source
This commit is contained in:
zhayujie
2022-12-19 03:06:48 +08:00
parent bf50694658
commit 17b6c63fb9
2 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bot/openai/open_ai_bot.py
View File

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

2
channel/wechat/wechat_channel.py
View File

@@ -52,7 +52,7 @@ class WechatChannel(Channel):
if len(str_list) == 2: if len(str_list) == 2:
content = str_list[1].strip() 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: if img_match_prefix:
content = content.split(img_match_prefix, 1)[1].strip() content = content.split(img_match_prefix, 1)[1].strip()
thead_pool.submit(self._do_send_img, content, from_user_id) thead_pool.submit(self._do_send_img, content, from_user_id)