Skip to content

Commit 968a2c4

Browse files
committed
refactor: use set_env_from_globals in tracers
1 parent b9ea359 commit 968a2c4

File tree

5 files changed

+10
-10
lines changed

5 files changed

+10
-10
lines changed

src/backend/base/langflow/services/tracing/arize_phoenix.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
from langflow.schema.data import Data
2121
from langflow.schema.message import Message
2222
from langflow.services.tracing.base import BaseTracer
23+
from langflow.services.tracing.utils import set_env_from_globals
2324

2425
if TYPE_CHECKING:
2526
from collections.abc import Sequence
@@ -70,8 +71,7 @@ def __init__(
7071
self.session_id = session_id
7172

7273
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)
7575

7676
try:
7777
self._ready = self.setup_arize_phoenix()

src/backend/base/langflow/services/tracing/langfuse.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from typing_extensions import override
1010

1111
from langflow.services.tracing.base import BaseTracer
12+
from langflow.services.tracing.utils import set_env_from_globals
1213

1314
if TYPE_CHECKING:
1415
from collections.abc import Sequence
@@ -51,8 +52,7 @@ def __init__(
5152
self.spans: dict = OrderedDict() # spans that are not ended
5253

5354
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)
5656

5757
config = self._get_config()
5858
self._ready: bool = self.setup_langfuse(config) if config else False

src/backend/base/langflow/services/tracing/langsmith.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111

1212
from langflow.schema.data import Data
1313
from langflow.services.tracing.base import BaseTracer
14+
from langflow.services.tracing.utils import set_env_from_globals
1415

1516
if TYPE_CHECKING:
1617
from collections.abc import Sequence
@@ -34,8 +35,7 @@ def __init__(
3435
):
3536
try:
3637
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)
3939

4040
self._ready = self.setup_langsmith()
4141
if not self._ready:

src/backend/base/langflow/services/tracing/langwatch.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99

1010
from langflow.schema.data import Data
1111
from langflow.services.tracing.base import BaseTracer
12+
from langflow.services.tracing.utils import set_env_from_globals
1213

1314
if TYPE_CHECKING:
1415
from collections.abc import Sequence
@@ -38,8 +39,7 @@ def __init__(
3839
self.flow_id = trace_name.split(" - ")[-1]
3940

4041
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)
4343

4444
try:
4545
self._ready: bool = self.setup_langwatch()

src/backend/base/langflow/services/tracing/opik.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
from langflow.schema.data import Data
1313
from langflow.schema.message import Message
1414
from langflow.services.tracing.base import BaseTracer
15+
from langflow.services.tracing.utils import set_env_from_globals
1516

1617
if TYPE_CHECKING:
1718
from collections.abc import Sequence
@@ -59,8 +60,7 @@ def __init__(
5960
self.spans: dict = {}
6061

6162
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)
6464

6565
config = self._get_config()
6666
self._ready: bool = self._setup_opik(config, trace_id) if config else False

0 commit comments

Comments
 (0)