diff --git a/tests/data/BasicChatwithPromptandHistory.json b/tests/unit/data/BasicChatwithPromptandHistory.json
similarity index 100%
rename from tests/data/BasicChatwithPromptandHistory.json
rename to tests/unit/data/BasicChatwithPromptandHistory.json
diff --git a/tests/data/ChatInputTest.json b/tests/unit/data/ChatInputTest.json
similarity index 100%
rename from tests/data/ChatInputTest.json
rename to tests/unit/data/ChatInputTest.json
diff --git a/tests/data/Openapi.json b/tests/unit/data/Openapi.json
similarity index 100%
rename from tests/data/Openapi.json
rename to tests/unit/data/Openapi.json
diff --git a/tests/data/TwoOutputsTest.json b/tests/unit/data/TwoOutputsTest.json
similarity index 100%
rename from tests/data/TwoOutputsTest.json
rename to tests/unit/data/TwoOutputsTest.json
diff --git a/tests/data/Vector_store.json b/tests/unit/data/Vector_store.json
similarity index 100%
rename from tests/data/Vector_store.json
rename to tests/unit/data/Vector_store.json
diff --git a/tests/data/WebhookTest.json b/tests/unit/data/WebhookTest.json
similarity index 100%
rename from tests/data/WebhookTest.json
rename to tests/unit/data/WebhookTest.json
diff --git a/tests/data/basic_example.json b/tests/unit/data/basic_example.json
similarity index 100%
rename from tests/data/basic_example.json
rename to tests/unit/data/basic_example.json
diff --git a/tests/data/complex_example.json b/tests/unit/data/complex_example.json
similarity index 100%
rename from tests/data/complex_example.json
rename to tests/unit/data/complex_example.json
diff --git a/tests/data/component.py b/tests/unit/data/component.py
similarity index 100%
rename from tests/data/component.py
rename to tests/unit/data/component.py
diff --git a/tests/data/component_multiple_outputs.py b/tests/unit/data/component_multiple_outputs.py
similarity index 100%
rename from tests/data/component_multiple_outputs.py
rename to tests/unit/data/component_multiple_outputs.py
diff --git a/tests/data/component_nested_call.py b/tests/unit/data/component_nested_call.py
similarity index 100%
rename from tests/data/component_nested_call.py
rename to tests/unit/data/component_nested_call.py
diff --git a/tests/data/component_with_templatefield.py b/tests/unit/data/component_with_templatefield.py
similarity index 100%
rename from tests/data/component_with_templatefield.py
rename to tests/unit/data/component_with_templatefield.py
diff --git a/tests/data/grouped_chat.json b/tests/unit/data/grouped_chat.json
similarity index 100%
rename from tests/data/grouped_chat.json
rename to tests/unit/data/grouped_chat.json
diff --git a/tests/data/one_group_chat.json b/tests/unit/data/one_group_chat.json
similarity index 100%
rename from tests/data/one_group_chat.json
rename to tests/unit/data/one_group_chat.json
diff --git a/tests/data/vector_store_grouped.json b/tests/unit/data/vector_store_grouped.json
similarity index 100%
rename from tests/data/vector_store_grouped.json
rename to tests/unit/data/vector_store_grouped.json