diff --git a/e2b_adk/__init__.py b/e2b_adk/__init__.py index 8d7c720..7d1f352 100644 --- a/e2b_adk/__init__.py +++ b/e2b_adk/__init__.py @@ -2,6 +2,8 @@ from __future__ import annotations +from importlib.metadata import version + from e2b import ConnectionConfig from .plugin import E2BPlugin @@ -17,10 +19,12 @@ # E2B's integration hook is process-wide and must be set before the SDK builds # a connection config. Importing the modules above has no runtime side effects; # plugin construction can only happen after package initialization completes. -ConnectionConfig.set_integration("e2b-adk/0.1.0") +__version__ = version("e2b-adk") +ConnectionConfig.set_integration(f"e2b-adk/{__version__}") __all__ = [ "E2BPlugin", + "__version__", "ListFiles", "ReadFile", "RunCode", diff --git a/tests/test_plugin.py b/tests/test_plugin.py index ffb6a3f..4dc8340 100644 --- a/tests/test_plugin.py +++ b/tests/test_plugin.py @@ -15,7 +15,7 @@ from e2b_code_interpreter import AsyncSandbox from pytest import LogCaptureFixture -from e2b_adk import E2BPlugin +from e2b_adk import E2BPlugin, __version__ from e2b_adk._sandbox import SandboxManager @@ -23,7 +23,7 @@ def test_e2b_integration_attribution() -> None: config = ConnectionConfig() user_agent_products = config.headers["User-Agent"].split() - assert "e2b-adk/0.1.0" in user_agent_products + assert f"e2b-adk/{__version__}" in user_agent_products def test_no_sandbox_on_init(patched_create: AsyncMock) -> None: