llc set coze2.0 runtime info

Author: JqRrt

CHANGLOG.md

@@ -1,4 +1,8 @@
-## [0.1.20] - 2025-11-10
+## [0.1.21] - 2025-12-23
+### Added
+- langchain callback set coze2.0 runtime info
+
+## [0.1.20] - 2025-12-08
 ### Added
 - langchain callback support langchain V1
 - langchain callback support set tag and name

cozeloop/integration/langchain/trace_callback.py

@@ -3,6 +3,7 @@
 
 from __future__ import annotations
 import json
+import os
 import time
 import traceback
 from typing import List, Dict, Union, Any, Optional, Callable, Protocol
@@ -324,8 +325,12 @@ def _new_flow_span(self, node_name: str, span_type: str, **kwargs: Any) -> Span:
         flow_span = _trace_callback_client.start_span(span_name, span_type, child_of=parent_span)
         run_id = str(kwargs['run_id'])
         self.run_map[run_id] = Run(run_id, flow_span, span_type)
-        # set default tags
-        flow_span.set_runtime(RuntimeInfo())
+        # set runtime
+        runtime_info = RuntimeInfo()
+        scene = os.getenv('COZELOOP_SCENE')
+        if scene:
+            runtime_info.scene = scene
+        flow_span.set_runtime(runtime_info)
         # set extra tags
         flow_span.set_tags(self._tags) # global tags
         try:

cozeloop/internal/trace/span.py

@@ -368,7 +368,8 @@ def set_start_time_first_resp(self, start_time_first_resp: int):
 
     def set_runtime(self, runtime: Runtime) -> None:
         r = runtime
-        r.scene = V_SCENE_CUSTOM
+        if r.scene is None or r.scene == "":
+            r.scene = V_SCENE_CUSTOM
         with self.lock:
             if self.system_tag_map is None:
                 self.system_tag_map = {}

cozeloop/internal/version.py

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

pyproject.toml

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