diff --git a/agent.py b/agent.py
index e3e044c..eaed377 100644
--- a/agent.py
+++ b/agent.py
@@ -33,6 +33,8 @@ from a2a_pack import (
     State,
     UploadedFile,
     FileUpload,
+    WorkspaceAccess,
+    WorkspaceMode,
 )