File tree 5 files changed +10
-10
lines changed
src/backend/base/langflow/services/tracing
5 files changed +10
-10
lines changed Original file line number Diff line number Diff line change 20
20
from langflow .schema .data import Data
21
21
from langflow .schema .message import Message
22
22
from langflow .services .tracing .base import BaseTracer
23
+ from langflow .services .tracing .utils import set_env_from_globals
23
24
24
25
if TYPE_CHECKING :
25
26
from collections .abc import Sequence
@@ -70,8 +71,7 @@ def __init__(
70
71
self .session_id = session_id
71
72
72
73
for key in ArizePhoenixTracer .get_required_variable_names ():
73
- if key in global_vars :
74
- os .environ [key ] = global_vars .get (key )
74
+ set_env_from_globals (key , global_vars )
75
75
76
76
try :
77
77
self ._ready = self .setup_arize_phoenix ()
Original file line number Diff line number Diff line change 9
9
from typing_extensions import override
10
10
11
11
from langflow .services .tracing .base import BaseTracer
12
+ from langflow .services .tracing .utils import set_env_from_globals
12
13
13
14
if TYPE_CHECKING :
14
15
from collections .abc import Sequence
@@ -51,8 +52,7 @@ def __init__(
51
52
self .spans : dict = OrderedDict () # spans that are not ended
52
53
53
54
for key in LangFuseTracer .get_required_variable_names ():
54
- if key in global_vars :
55
- os .environ [key ] = global_vars .get (key )
55
+ set_env_from_globals (key , global_vars )
56
56
57
57
config = self ._get_config ()
58
58
self ._ready : bool = self .setup_langfuse (config ) if config else False
Original file line number Diff line number Diff line change 11
11
12
12
from langflow .schema .data import Data
13
13
from langflow .services .tracing .base import BaseTracer
14
+ from langflow .services .tracing .utils import set_env_from_globals
14
15
15
16
if TYPE_CHECKING :
16
17
from collections .abc import Sequence
@@ -34,8 +35,7 @@ def __init__(
34
35
):
35
36
try :
36
37
for key in LangSmithTracer .get_required_variable_names ():
37
- if key in global_vars :
38
- os .environ [key ] = global_vars .get (key )
38
+ set_env_from_globals (key , global_vars )
39
39
40
40
self ._ready = self .setup_langsmith ()
41
41
if not self ._ready :
Original file line number Diff line number Diff line change 9
9
10
10
from langflow .schema .data import Data
11
11
from langflow .services .tracing .base import BaseTracer
12
+ from langflow .services .tracing .utils import set_env_from_globals
12
13
13
14
if TYPE_CHECKING :
14
15
from collections .abc import Sequence
@@ -38,8 +39,7 @@ def __init__(
38
39
self .flow_id = trace_name .split (" - " )[- 1 ]
39
40
40
41
for key in LangWatchTracer .get_required_variable_names ():
41
- if key in global_vars :
42
- os .environ [key ] = global_vars .get (key )
42
+ set_env_from_globals (key , global_vars )
43
43
44
44
try :
45
45
self ._ready : bool = self .setup_langwatch ()
Original file line number Diff line number Diff line change 12
12
from langflow .schema .data import Data
13
13
from langflow .schema .message import Message
14
14
from langflow .services .tracing .base import BaseTracer
15
+ from langflow .services .tracing .utils import set_env_from_globals
15
16
16
17
if TYPE_CHECKING :
17
18
from collections .abc import Sequence
@@ -59,8 +60,7 @@ def __init__(
59
60
self .spans : dict = {}
60
61
61
62
for key in OpikTracer .get_required_variable_names ():
62
- if key in global_vars :
63
- os .environ [key ] = global_vars .get (key )
63
+ set_env_from_globals (key , global_vars )
64
64
65
65
config = self ._get_config ()
66
66
self ._ready : bool = self ._setup_opik (config , trace_id ) if config else False
You can’t perform that action at this time.
0 commit comments