Skip to content

conditional import #185

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
xzrderek wants to merge 1 commit into from
Closed

conditional import #185

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
45 changes: 40 additions & 5 deletions eval_protocol/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,34 @@
from .typed_interface import reward_function
from .quickstart import aha_judge, split_multi_turn_rows
from .pytest import evaluation_test, SingleTurnRolloutProcessor
from .adapters import OpenAIResponsesAdapter, LangfuseAdapter, BraintrustAdapter, LangSmithAdapter

try:
from .adapters import OpenAIResponsesAdapter

_OPENAI_RESPONSES_AVAILABLE = True
except ImportError:
_OPENAI_RESPONSES_AVAILABLE = False

try:
from .adapters import LangfuseAdapter

_LANGFUSE_AVAILABLE = True
except ImportError:
_LANGFUSE_AVAILABLE = False

try:
from .adapters import BraintrustAdapter

_BRAINTRUST_AVAILABLE = True
except ImportError:
_BRAINTRUST_AVAILABLE = False

try:
from .adapters import LangSmithAdapter

_LANGSMITH_AVAILABLE = True
except ImportError:
_LANGSMITH_AVAILABLE = False

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

Expand All @@ -48,10 +75,6 @@
"split_multi_turn_rows",
"evaluation_test",
"SingleTurnRolloutProcessor",
"OpenAIResponsesAdapter",
"LangfuseAdapter",
"BraintrustAdapter",
"LangSmithAdapter",
# Core interfaces
"Message",
"MetricResult",
Expand Down Expand Up @@ -84,6 +107,18 @@
"mcp",
]

if _OPENAI_RESPONSES_AVAILABLE:
__all__.append("OpenAIResponsesAdapter")

if _LANGFUSE_AVAILABLE:
__all__.append("LangfuseAdapter")

if _BRAINTRUST_AVAILABLE:
__all__.append("BraintrustAdapter")

if _LANGSMITH_AVAILABLE:
__all__.append("LangSmithAdapter")

from . import _version

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