update

Author: JqRrt

cozeloop/decorator/decorator.py

@@ -344,10 +344,11 @@ def sync_wrapper(*args: Any, **kwargs: Any):
                     if len(args) > 0 and is_class_func(func):
                         extra = {"_inner_class_self": args[0]}
                         args = args[1:]
-                    inp = {
-                        "args": args,
-                        "kwargs": kwargs
-                    }
+                    inp = {}
+                    if len(args) > 0:
+                        inp['args'] = args
+                    if len(kwargs) > 0:
+                        inp['kwargs'] = kwargs
                     res = RunnableLambda(_param_wrapped_func).invoke(input=inp, config=config, **extra)
                     if hasattr(res, "__iter__"):
                         return res
@@ -369,10 +370,11 @@ async def async_wrapper(*args: Any, **kwargs: Any):
                     if len(args) > 0 and is_class_func(func):
                         extra = {"_inner_class_self": args[0]}
                         args = args[1:]
-                    inp = {
-                        "args": args,
-                        "kwargs": kwargs
-                    }
+                    inp = {}
+                    if len(args) > 0:
+                        inp['args'] = args
+                    if len(kwargs) > 0:
+                        inp['kwargs'] = kwargs
                     res = await RunnableLambda(_param_wrapped_func_async).ainvoke(input=inp, config=config, **extra)
                     if hasattr(res, "__aiter__"):
                         return res
@@ -398,10 +400,11 @@ def gen_wrapper(*args: Any, **kwargs: Any):
                     if len(args) > 0 and is_class_func(func):
                         extra = {"_inner_class_self": args[0]}
                         args = args[1:]
-                    inp = {
-                        "args": args,
-                        "kwargs": kwargs
-                    }
+                    inp = {}
+                    if len(args) > 0:
+                        inp['args'] = args
+                    if len(kwargs) > 0:
+                        inp['kwargs'] = kwargs
                     gen = RunnableLambda(_param_wrapped_func).invoke(input=inp, config=config, *extra)
                     try:
                         for item in gen:
@@ -420,10 +423,11 @@ async def async_gen_wrapper(*args: Any, **kwargs: Any):
                     if len(args) > 0 and is_class_func(func):
                         extra = {"_inner_class_self": args[0]}
                         args = args[1:]
-                    inp = {
-                        "args": args,
-                        "kwargs": kwargs
-                    }
+                    inp = {}
+                    if len(args) > 0:
+                        inp['args'] = args
+                    if len(kwargs) > 0:
+                        inp['kwargs'] = kwargs
                     gen = RunnableLambda(_param_wrapped_func_async).invoke(input=inp, config=config, **extra)
                     items = []
                     try: