feat: Dify加入Http代理

Author: HenryXiaoYang

plugins/all_in_one_config.toml

@@ -17,6 +17,11 @@ price = 0 # 用一次扣积分，如果0则不扣
 admin_ignore = true #admin是否忽略扣除
 whitelist_ignore = true #白名单是否忽略扣除
 
+# Http代理设置
+# 格式: http://用户名:密码@代理地址:代理端口
+# 例如：http://127.0.0.1:7890
+http-proxy = ""
+
 [BotStatus]
 enable = true
 command = ["status", "bot", "机器人状态", "状态"]

plugins/dify.py

@@ -43,6 +43,8 @@ def __init__(self):
         self.admin_ignore = plugin_config["admin_ignore"]
         self.whitelist_ignore = plugin_config["whitelist_ignore"]
 
+        self.http_proxy = plugin_config["http-proxy"]
+
         self.db = BotDatabase()
 
     @on_text_message
@@ -174,7 +176,7 @@ async def dify(self, bot: WechatAPIClient, message: dict, query: str, files=None
         url = f"{self.base_url}/chat-messages"
 
         ai_resp = ""
-        async with aiohttp.ClientSession() as session:
+        async with aiohttp.ClientSession(proxy=self.http_proxy) as session:
             async with session.post(url=url, headers=headers, data=payload) as resp:
                 if resp.status == 200:
                     # 读取响应
@@ -238,7 +240,7 @@ async def upload_file(self, user: str, file: bytes):
 
         url = f"{self.base_url}/files/upload"
 
-        async with aiohttp.ClientSession() as session:
+        async with aiohttp.ClientSession(proxy=self.http_proxy) as session:
             async with session.post(url, headers=headers, data=formdata) as resp:
                 resp_json = await resp.json()
 
@@ -267,16 +269,14 @@ async def dify_handle_text(self, bot: WechatAPIClient, message: dict, text: str)
         if text:
             await bot.send_at_message(message["FromWxid"], "\n" + text, [message["SenderWxid"]])
 
-    @staticmethod
-    async def download_file(url: str) -> bytes:
-        async with aiohttp.ClientSession() as session:
+    async def download_file(self, url: str) -> bytes:
+        async with aiohttp.ClientSession(proxy=self.http_proxy) as session:
             async with session.get(url) as resp:
                 return await resp.read()
 
-    @staticmethod
-    async def dify_handle_image(bot: WechatAPIClient, message: dict, image: Union[str, bytes]):
+    async def dify_handle_image(self, bot: WechatAPIClient, message: dict, image: Union[str, bytes]):
         if isinstance(image, str) and image.startswith("http"):
-            async with aiohttp.ClientSession() as session:
+            async with aiohttp.ClientSession(proxy=self.http_proxy) as session:
                 async with session.get(image) as resp:
                     image = bot.byte_to_base64(await resp.read())
         elif isinstance(image, bytes):