fix

Author: BotellaA

src/opengeodeweb_viewer/vtk_protocol.py

@@ -49,12 +49,8 @@ def get_data(self, data_id: str) -> dict[str, str | list[str] | None]:
 
                 return {
                     "id": data.id,
-                    "native_file_name": data.native_file_name,
                     "viewable_file_name": data.viewable_file_name,
-                    "geode_object": data.geode_object,
-                    "light_viewable": data.light_viewable,
-                    "input_file": data.input_file,
-                    "additional_files": data.additional_files,
+                    "viewer_object": data.viewer_object,
                 }
             except Exception as e:
                 print(f"Error fetching data {data_id}: {e}")

src/tests/conftest.py

@@ -208,14 +208,11 @@ def configure_test_environment() -> Generator[None, None, None]:
 
 @pytest.fixture
 def dataset_factory() -> Callable[..., str]:
-    def create_dataset(
-        *, id: str, viewable_file_name: str, viewer_object: str | None = None
-    ) -> str:
+    def create_dataset(*, id: str, viewable_file_name: str) -> str:
         session = get_session()
-        if viewer_object is None:
-            viewer_object = (
-                "model" if viewable_file_name.lower().endswith(".vtm") else "mesh"
-            )
+        viewer_object = (
+            "model" if viewable_file_name.lower().endswith(".vtm") else "mesh"
+        )
 
         row = session.get(Data, id)
         if row is None: