Skip to content

Commit aacbb5f

Browse files
committed
conditional import
1 parent 5c9194b commit aacbb5f

1 file changed

Lines changed: 40 additions & 5 deletions

File tree

eval_protocol/__init__.py

Lines changed: 40 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,34 @@
3939
from .typed_interface import reward_function
4040
from .quickstart import aha_judge, split_multi_turn_rows
4141
from .pytest import evaluation_test, SingleTurnRolloutProcessor
42-
from .adapters import OpenAIResponsesAdapter, LangfuseAdapter, BraintrustAdapter, LangSmithAdapter
42+
43+
try:
44+
from .adapters import OpenAIResponsesAdapter
45+
46+
_OPENAI_RESPONSES_AVAILABLE = True
47+
except ImportError:
48+
_OPENAI_RESPONSES_AVAILABLE = False
49+
50+
try:
51+
from .adapters import LangfuseAdapter
52+
53+
_LANGFUSE_AVAILABLE = True
54+
except ImportError:
55+
_LANGFUSE_AVAILABLE = False
56+
57+
try:
58+
from .adapters import BraintrustAdapter
59+
60+
_BRAINTRUST_AVAILABLE = True
61+
except ImportError:
62+
_BRAINTRUST_AVAILABLE = False
63+
64+
try:
65+
from .adapters import LangSmithAdapter
66+
67+
_LANGSMITH_AVAILABLE = True
68+
except ImportError:
69+
_LANGSMITH_AVAILABLE = False
4370

4471
warnings.filterwarnings("default", category=DeprecationWarning, module="eval_protocol")
4572

@@ -48,10 +75,6 @@
4875
"split_multi_turn_rows",
4976
"evaluation_test",
5077
"SingleTurnRolloutProcessor",
51-
"OpenAIResponsesAdapter",
52-
"LangfuseAdapter",
53-
"BraintrustAdapter",
54-
"LangSmithAdapter",
5578
# Core interfaces
5679
"Message",
5780
"MetricResult",
@@ -84,6 +107,18 @@
84107
"mcp",
85108
]
86109

110+
if _OPENAI_RESPONSES_AVAILABLE:
111+
__all__.append("OpenAIResponsesAdapter")
112+
113+
if _LANGFUSE_AVAILABLE:
114+
__all__.append("LangfuseAdapter")
115+
116+
if _BRAINTRUST_AVAILABLE:
117+
__all__.append("BraintrustAdapter")
118+
119+
if _LANGSMITH_AVAILABLE:
120+
__all__.append("LangSmithAdapter")
121+
87122
from . import _version
88123

89124
__version__ = _version.get_versions()["version"]

0 commit comments

Comments
 (0)