Skip to content

fix: derive integration attribution from package version #2

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

Merged
OndrejDrapalik merged 1 commit into from
Jul 16, 2026
Merged

fix: derive integration attribution from package version #2

Show file tree
Hide file tree
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
6 changes: 5 additions & 1 deletion e2b_adk/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@

from __future__ import annotations

from importlib.metadata import version

from e2b import ConnectionConfig

from .plugin import E2BPlugin
Expand All @@ -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",
Expand Down
4 changes: 2 additions & 2 deletions tests/test_plugin.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,15 +15,15 @@
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


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:
Expand Down