From fc5d3e4e9c13804f415101c17838a73945d3aefc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E8=B6=85?= <lichao@ahxz.vip>
Date: Sun, 16 Apr 2023 22:28:10 +0800
Subject: [PATCH] feat: Make the size parameter of the resulting picture
 configurable

---
 bot/openai/open_ai_image.py | 2 +-
 config.py                   | 2 ++
 docker/build.latest.sh      | 6 +++++-
 3 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/bot/openai/open_ai_image.py b/bot/openai/open_ai_image.py
index 2fae243..c3e18f5 100644
--- a/bot/openai/open_ai_image.py
+++ b/bot/openai/open_ai_image.py
@@ -20,7 +20,7 @@ class OpenAIImage(object):
             response = openai.Image.create(
                 prompt=query,    #图片描述
                 n=1,             #每次生成图片的数量
-                size="256x256"   #图片大小,可选有 256x256, 512x512, 1024x1024
+                size=conf().get('image_create_size', "256x256")   #图片大小,可选有 256x256, 512x512, 1024x1024
             )
             image_url = response['data'][0]['url']
             logger.info("[OPEN_AI] image_url={}".format(image_url))
diff --git a/config.py b/config.py
index 7f4270d..6d42a9d 100644
--- a/config.py
+++ b/config.py
@@ -32,6 +32,8 @@ available_setting = {
     "image_create_prefix": ["画", "看", "找"],  # 开启图片回复的前缀
     "concurrency_in_session": 1, # 同一会话最多有多少条消息在处理中，大于1可能乱序
 
+    "image_create_size": "256x256", #图片大小,可选有 256x256, 512x512, 1024x1024
+
     # chatgpt会话参数
     "expires_in_seconds": 3600,  # 无操作会话的过期时间
     "character_desc": "你是ChatGPT, 一个由OpenAI训练的大型语言模型, 你旨在回答并解决人们的任何问题，并且可以使用多种语言与人交流。",  # 人格描述
diff --git a/docker/build.latest.sh b/docker/build.latest.sh
index 0f06f82..92c3564 100644
--- a/docker/build.latest.sh
+++ b/docker/build.latest.sh
@@ -1,4 +1,8 @@
 #!/bin/bash
 
+unset KUBECONFIG
+
 cd .. && docker build -f docker/Dockerfile.latest \
-             -t zhayujie/chatgpt-on-wechat .
\ No newline at end of file
+             -t zhayujie/chatgpt-on-wechat .
+
+docker tag zhayujie/chatgpt-on-wechat zhayujie/chatgpt-on-wechat:$(date +%y%m%d)
\ No newline at end of file