参数输入bug修复

2023-04-05 00:07:08 +08:00
parent 3648648b3d
commit 7a254c150f
2 changed files with 3 additions and 3 deletions
--- a/toolbox.py
+++ b/toolbox.py
@@ -6,10 +6,10 @@ def ArgsGeneralWrapper(f):
    """
        装饰器函数，用于重组输入参数，改变输入参数的顺序与结构。
    """
-    def decorated(txt, txt2, top_p, temperature, chatbot, history, system_prompt, *args, **kwargs):
+    def decorated(txt, txt2, *args, **kwargs):
        txt_passon = txt
        if txt == "" and txt2 != "": txt_passon = txt2
-        yield from f(txt_passon, top_p, temperature, chatbot, history, system_prompt, *args, **kwargs)
+        yield from f(txt_passon, *args, **kwargs)

    return decorated