From 8865b232ca68906f042a89fcdffbf3cf24907878 Mon Sep 17 00:00:00 2001
From: ReeInk <hi@ree.ink>
Date: Tue, 2 May 2023 00:12:35 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=EF=BC=9A=E8=AF=BB=E5=8F=96?=
 =?UTF-8?q?=E7=8E=AF=E5=A2=83=E5=8F=98=E9=87=8F=E9=87=8D=E5=AE=9A=E5=90=91?=
 =?UTF-8?q?URL=E6=A0=BC=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 toolbox.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/toolbox.py b/toolbox.py
index 81ef1ed..20a9abb 100644
--- a/toolbox.py
+++ b/toolbox.py
@@ -545,13 +545,11 @@ def read_single_conf_from_env(arg, default_value):
         # 格式为：username1:password1;username2:password2
         for item in os.environ[env_arg].split(";"):
             r.append(tuple(item.split(":")))
-    elif arg == "API_URL_REDIRECT":
-        r = {}
-        # 对于API_URL_REDIRECT的环境变量配置，我们允许用户使用;分隔转发地址
-        # 格式为：url1:redirect1;url2:redirect2
-        for item in os.environ[env_arg].split(";"):
-            k, v = item.split(":")
-            r[k] = v
+    elif arg == "API_URL_REDIRECT": 
+        # 对于API_URL_REDIRECT的环境变量，我们允许用户使用json格式配置多个url重定向
+        # 格式为一个json字符串，例如：{"https://api.openai.com/v1/chat/completions": "https://ai.open.com/api/conversation"}
+        import json
+        r = json.loads(os.environ[env_arg])
     elif isinstance(default_value, bool):
         r = bool(os.environ[env_arg])
     elif isinstance(default_value, int):