diff --git a/api/tests/unit_tests/core/app/apps/advanced_chat/test_app_generator.py b/api/tests/unit_tests/core/app/apps/advanced_chat/test_app_generator.py
index f347a5fae7e..3d61f2f19bc 100644
--- a/api/tests/unit_tests/core/app/apps/advanced_chat/test_app_generator.py
+++ b/api/tests/unit_tests/core/app/apps/advanced_chat/test_app_generator.py
@@ -401,7 +401,14 @@ class TestAdvancedChatAppGeneratorInternals:
             workflow_run_id="run-id",
         )
 
-        workflow = SimpleNamespace(id="wf-1", tenant_id="tenant", features={"feature": True}, features_dict={})
+        workflow = SimpleNamespace(
+            id="wf-1",
+            tenant_id="tenant",
+            features={"feature": True},
+            features_dict={},
+            environment_variables=[],
+            conversation_variables=[],
+        )
         conversation = SimpleNamespace(id="conv-1", mode=AppMode.ADVANCED_CHAT, override_model_configs=None)
         message = SimpleNamespace(
             id="msg-1",
@@ -506,7 +513,14 @@ class TestAdvancedChatAppGeneratorInternals:
             workflow_run_id="run-id",
         )
 
-        workflow = SimpleNamespace(id="wf-2", tenant_id="tenant", features={}, features_dict={})
+        workflow = SimpleNamespace(
+            id="wf-2",
+            tenant_id="tenant",
+            features={},
+            features_dict={},
+            environment_variables=[],
+            conversation_variables=[],
+        )
         conversation = SimpleNamespace(id="conv-2", mode=AppMode.ADVANCED_CHAT, override_model_configs=None)
         message = SimpleNamespace(
             id="msg-2",
@@ -981,7 +995,13 @@ class TestAdvancedChatAppGeneratorInternals:
         with pytest.raises(GenerateTaskStoppedError):
             generator._handle_advanced_chat_response(
                 application_generate_entity=application_generate_entity,
-                workflow=WorkflowSnapshot(id="wf", tenant_id="tenant", features_dict={}),
+                workflow=WorkflowSnapshot(
+                    id="wf",
+                    tenant_id="tenant",
+                    features_dict={},
+                    environment_variables=(),
+                    conversation_variables=(),
+                ),
                 queue_manager=SimpleNamespace(),
                 conversation=ConversationSnapshot(id="conv", mode=AppMode.ADVANCED_CHAT),
                 message=MessageSnapshot(
@@ -1029,7 +1049,13 @@ class TestAdvancedChatAppGeneratorInternals:
             with pytest.raises(ValueError, match="other error"):
                 generator._handle_advanced_chat_response(
                     application_generate_entity=application_generate_entity,
-                    workflow=WorkflowSnapshot(id="wf", tenant_id="tenant", features_dict={}),
+                    workflow=WorkflowSnapshot(
+                        id="wf",
+                        tenant_id="tenant",
+                        features_dict={},
+                        environment_variables=(),
+                        conversation_variables=(),
+                    ),
                     queue_manager=SimpleNamespace(),
                     conversation=ConversationSnapshot(id="conv", mode=AppMode.ADVANCED_CHAT),
                     message=MessageSnapshot(
diff --git a/api/tests/unit_tests/core/app/apps/advanced_chat/test_generate_task_pipeline.py b/api/tests/unit_tests/core/app/apps/advanced_chat/test_generate_task_pipeline.py
index 9f936bb8035..b319887b8d3 100644
--- a/api/tests/unit_tests/core/app/apps/advanced_chat/test_generate_task_pipeline.py
+++ b/api/tests/unit_tests/core/app/apps/advanced_chat/test_generate_task_pipeline.py
@@ -170,7 +170,13 @@ def test_resume_appends_chunks_to_paused_answer() -> None:
     user = EndUser()
     user.id = "user-1"
     user.session_id = "session-1"
-    workflow = pipeline_module.WorkflowSnapshot(id="workflow-1", tenant_id="tenant-1", features_dict={})
+    workflow = pipeline_module.WorkflowSnapshot(
+        id="workflow-1",
+        tenant_id="tenant-1",
+        features_dict={},
+        environment_variables=(),
+        conversation_variables=(),
+    )
 
     pipeline = pipeline_module.AdvancedChatAppGenerateTaskPipeline(
         application_generate_entity=application_generate_entity,
diff --git a/api/tests/unit_tests/core/app/apps/advanced_chat/test_generate_task_pipeline_core.py b/api/tests/unit_tests/core/app/apps/advanced_chat/test_generate_task_pipeline_core.py
index 1f975c74fb1..e301cee5d0b 100644
--- a/api/tests/unit_tests/core/app/apps/advanced_chat/test_generate_task_pipeline_core.py
+++ b/api/tests/unit_tests/core/app/apps/advanced_chat/test_generate_task_pipeline_core.py
@@ -92,7 +92,13 @@ def _make_pipeline():
         answer="",
     )
     conversation = ConversationSnapshot(id="conv-id", mode=AppMode.ADVANCED_CHAT)
-    workflow = WorkflowSnapshot(id="workflow-id", tenant_id="tenant", features_dict={})
+    workflow = WorkflowSnapshot(
+        id="workflow-id",
+        tenant_id="tenant",
+        features_dict={},
+        environment_variables=(),
+        conversation_variables=(),
+    )
     user = EndUser(tenant_id="tenant", type="session", name="tester", session_id="session")
 
     pipeline = AdvancedChatAppGenerateTaskPipeline(