support annotation (#7)

* support annotation

Author: JqRrt

cozeloop/__init__.py

@@ -13,6 +13,7 @@
 from .client import Client
 from ._client import (
     new_client,
+    set_default_client,
     workspace_id,
     close,
     get_prompt,

cozeloop/_client.py

@@ -258,6 +258,18 @@ def flush(self):
         self._trace_provider.flush()
 
 
+def set_default_client(client: Client):
+    global _default_client
+    with _client_lock:
+        if _default_client:
+            temp_client = _default_client
+            _default_client = client
+            if temp_client != client:
+                temp_client.close()
+        else:
+            _default_client = client
+
+
 def get_default_client() -> Client:
     global _default_client
     if _default_client is None:

cozeloop/decorator/__init__.py

@@ -0,0 +1,7 @@
+# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
+# SPDX-License-Identifier: MIT
+
+from cozeloop.decorator.decorator import CozeLoopDecorator
+
+coze_loop_decorator= CozeLoopDecorator()
+observe = coze_loop_decorator.observe