diff --git a/.changeset/strange-teachers-whisper.md b/.changeset/strange-teachers-whisper.md
new file mode 100644
index 000000000..212caf4fb
--- /dev/null
+++ b/.changeset/strange-teachers-whisper.md
@@ -0,0 +1,5 @@
+---
+'@e2b/python-sdk': patch
+---
+
+Export Artifact class
diff --git a/packages/python-sdk/e2b/__init__.py b/packages/python-sdk/e2b/__init__.py
index 62766bd06..356107dc0 100644
--- a/packages/python-sdk/e2b/__init__.py
+++ b/packages/python-sdk/e2b/__init__.py
@@ -24,4 +24,5 @@
 
 from .templates import (
     DataAnalysis,
+    Artifact,
 )
diff --git a/packages/python-sdk/e2b/templates/__init__.py b/packages/python-sdk/e2b/templates/__init__.py
index c3ef08453..ee83ab1d7 100644
--- a/packages/python-sdk/e2b/templates/__init__.py
+++ b/packages/python-sdk/e2b/templates/__init__.py
@@ -1 +1 @@
-from .data_analysis import DataAnalysis
+from .data_analysis import DataAnalysis, Artifact