update

Author: JqRrt

cozeloop/integration/langchain/trace_callback.py

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

cozeloop/internal/trace/span.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: MIT
 
 import logging
+import os
 from abc import ABC
 from typing import Dict, Any, List, Optional
 from datetime import datetime
@@ -368,8 +369,10 @@ def set_start_time_first_resp(self, start_time_first_resp: int):
 
     def set_runtime(self, runtime: Runtime) -> None:
         r = runtime
-        if r.scene is None or r.scene == "":
-            r.scene = V_SCENE_CUSTOM
+        r.scene = V_SCENE_CUSTOM
+        scene = os.getenv('COZELOOP_SCENE') # record scene from env
+        if scene:
+            r.scene = scene
         with self.lock:
             if self.system_tag_map is None:
                 self.system_tag_map = {}