diff --git a/packages/artifacts/src/hooks.ts b/packages/artifacts/src/hooks.ts
index 0eab60da..ea295b0b 100644
--- a/packages/artifacts/src/hooks.ts
+++ b/packages/artifacts/src/hooks.ts
@@ -100,6 +100,7 @@ export function useArtifact<
     error: currentArtifact?.error,
     isActive,
     hasData: currentArtifact !== null,
+    setCurrentArtifact,
   };
 }
 
diff --git a/packages/artifacts/src/types.ts b/packages/artifacts/src/types.ts
index 1bf09594..f6766f68 100644
--- a/packages/artifacts/src/types.ts
+++ b/packages/artifacts/src/types.ts
@@ -55,6 +55,7 @@ export interface UseArtifactReturn<T = unknown> {
   error?: string;
   isActive: boolean;
   hasData: boolean;
+  setCurrentArtifact: (artifact: ArtifactData<T> | null) => void;
 }
 
 export interface UseArtifactsOptions {