fix

Author: JqRrt

cozeloop/internal/trace/noop_span.py

@@ -116,6 +116,9 @@ def set_service_name(self, service_name: str) -> None:
     def set_log_id(self, log_id: str) -> None:
         pass
 
+    def set_system_tags(self, system_tags: Dict[str, Any]) -> None:
+        pass
+
     def __enter__(self):
         return self
 

cozeloop/internal/trace/span.py

@@ -374,6 +374,15 @@ def set_service_name(self, service_name: str) -> None:
     def set_log_id(self, log_id: str) -> None:
         self.log_id = log_id
 
+    def set_system_tags(self, system_tags: Dict[str, Any]) -> None:
+        if not system_tags:
+            return
+        with self.lock:
+            if self.system_tag_map is None:
+                self.system_tag_map = {}
+            for key, value in system_tags.items():
+                self.system_tag_map[key] = value
+
     def get_rectified_map(self, input_map: Dict[str, Any]) -> (Dict[str, Any], List[str], int):
         validate_map = {}
         cut_off_keys = []

cozeloop/span.py

@@ -180,6 +180,12 @@ def set_log_id(self, log_id: str) -> None:
         Set the custom log id, identify different query.
         """
 
+    @abstractmethod
+    def set_system_tags(self, system_tags: Dict[str, Any]) -> None:
+        """
+        Set system tags. DO NOT use this method unless you know what you are doing.
+        """
+
 
 class Span(CommonSpanSetter, SpanContext):
     """

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cozeloop"
-version = "0.1.8"
+version = "0.1.10"
 description = "coze loop sdk"
 authors = ["JiangQi715 <jiangqi.rrt@bytedance.com>"]
 license = "MIT"