Fix decorator bug. (#41)

Change-Id: I8cd52316709595f503a5da7477e9a1710f30a05a

Author: taoyifan89

cozeloop/decorator/decorator.py

@@ -101,8 +101,7 @@ def sync_wrapper(*args: Any, **kwargs: Any):
                     span.set_tags(tags)
                     span.finish()
 
-                    if res:
-                        return res
+                    return res
 
             @wraps(func)
             async def async_wrapper(*args: Any, **kwargs: Any):
@@ -140,8 +139,7 @@ async def async_wrapper(*args: Any, **kwargs: Any):
                     span.set_tags(tags)
                     span.finish()
 
-                    if res:
-                        return res
+                    return res
 
             @wraps(func)
             def gen_wrapper(*args: Any, **kwargs: Any):
@@ -249,7 +247,7 @@ def sync_stream_wrapper(*args: Any, **kwargs: Any):
                     span.set_input(input)
                     span.set_tags(tags)
 
-                    if not hasattr(res, "__iter__") and res:
+                    if not hasattr(res, "__iter__"):
                         return res
 
             @wraps(func)
@@ -288,7 +286,7 @@ async def async_stream_wrapper(*args: Any, **kwargs: Any):
                     span.set_input(input)
                     span.set_tags(tags)
 
-                    if not hasattr(res, "__aiter__") and res:
+                    if not hasattr(res, "__aiter__"):
                         return res
 
             if is_async_gen_func(func):
@@ -449,7 +447,7 @@ async def __async_streamer__(
                     self.__span.set_start_time_first_resp(time.time_ns() // 1_000)
                     self.__is_set_start_time_first_token = True
                 yield s
-        except StopIteration:
+        except (StopIteration, StopAsyncIteration):
             pass
 
 

cozeloop/internal/version.py

@@ -1,4 +1,4 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
-VERSION = 'v0.1.21'
+VERSION = 'v0.1.22'