支持自定义基础功能区

2023-10-11 00:14:56 +08:00
parent 500a0cbd16
commit 6f383c1dc8
3 changed files with 67 additions and 6 deletions
--- a/toolbox.py
+++ b/toolbox.py
@@ -624,7 +624,15 @@ def load_chat_cookies():
    if is_any_api_key(AZURE_API_KEY):
        if is_any_api_key(API_KEY): API_KEY = API_KEY + ',' + AZURE_API_KEY
        else: API_KEY = AZURE_API_KEY
-    return {'api_key': API_KEY, 'llm_model': LLM_MODEL}
+    customize_fn_overwrite_ = {}
+    for k in range(1):
+        customize_fn_overwrite_.update({  
+            f"自定义按钮{k+1}":{
+                "Prefix":   r"请在自定义菜单中定义提示词前缀.",
+                "Suffix":   r"请在自定义菜单中定义提示词后缀",
+            }
+        })
+    return {'api_key': API_KEY, 'llm_model': LLM_MODEL, 'customize_fn_overwrite': customize_fn_overwrite_}

 def is_openai_api_key(key):
    CUSTOM_API_KEY_PATTERN, = get_conf('CUSTOM_API_KEY_PATTERN')