update

Author: JqRrt

cozeloop/integration/langchain/trace_callback.py

@@ -200,7 +200,6 @@ def on_agent_finish(self, finish: AgentFinish, **kwargs: Any) -> Any:
         return
 
     def _end_flow_span(self, span: Span):
-        span.set_tags(self._tags)
         span.finish()
 
     def _get_model_tags(self, response: LLMResult, **kwargs: Any) -> Dict[str, Any]:
@@ -328,8 +327,9 @@ def _new_flow_span(self, node_name: str, span_type: str, **kwargs: Any) -> Span:
         # set default tags
         flow_span.set_runtime(RuntimeInfo())
         # set extra tags
+        flow_span.set_tags(self._tags) # global tags
         try:
-            if self.tags_fn:
+            if self.tags_fn: # add tags fn
                 tags = self.tags_fn(node_name)
                 if isinstance(tags, dict):
                     flow_span.set_tags(tags)