Get version using restAPI

.github/workflows/lint.yml

@@ -38,8 +38,5 @@ jobs:
           - name: Run security scan
             run: make security-scan
 
-          - name: Verify GraphQL schema is up to date
-            run: make schema-check
-
           - name: Run Python formatters and linters
             run: make format-check lint-check          

package/kedro_viz/api/apps.py

@@ -18,7 +18,6 @@
 from kedro_viz.api.rest.responses.utils import EnhancedORJSONResponse
 from kedro_viz.integrations.kedro import telemetry as kedro_telemetry
 
-from .graphql.router import router as graphql_router
 from .rest.router import router as rest_router
 
 _HTML_DIR = Path(__file__).parent.parent.absolute() / "html"
@@ -63,7 +62,6 @@ def create_api_app_from_project(
     """
     app = _create_base_api_app()
     app.include_router(rest_router)
-    app.include_router(graphql_router)
 
     # Check for html directory existence.
     if Path(_HTML_DIR).is_dir():

package/kedro_viz/api/rest/responses/version.py

@@ -0,0 +1,43 @@
+"""`kedro_viz.api.rest.responses.version` contains response classes
+and utility functions for the `/version` REST endpoint"""
+
+from pydantic import ConfigDict
+
+from kedro_viz import __version__
+from kedro_viz.api.rest.responses.base import BaseAPIResponse
+from kedro_viz.integrations.pypi import get_latest_version, is_running_outdated_version
+
+
+class VersionAPIResponse(BaseAPIResponse):
+    """
+    VersionAPIResponse is a subclass of BaseAPIResponse that represents the response structure for version API.
+
+    Attributes:
+        installed (str): The installed version of the Kedro Viz package.
+        is_outdated (bool): Whether the installed version is outdated.
+        latest (str): The latest available version of the Kedro Viz package.
+    """
+
+    installed: str
+    is_outdated: bool
+    latest: str
+    model_config = ConfigDict(
+        json_schema_extra={
+            "installed": __version__,
+            "is_outdated": False,
+            "latest": "0.0.0",
+        }
+    )
+
+
+def get_version_response():
+    """API response for `/api/version`."""
+    installed_version = str(__version__)
+    latest_version = str(get_latest_version())
+    is_outdated = is_running_outdated_version(installed_version, latest_version)
+
+    return VersionAPIResponse(
+        installed=installed_version,
+        is_outdated=is_outdated,
+        latest=latest_version,
+    )

package/kedro_viz/api/rest/router.py

@@ -19,6 +19,10 @@
     GraphAPIResponse,
     get_pipeline_response,
 )
+from kedro_viz.api.rest.responses.version import (
+    VersionAPIResponse,
+    get_version_response,
+)
 
 logger = logging.getLogger(__name__)
 
@@ -50,6 +54,14 @@ async def get_single_pipeline_data(registered_pipeline_id: str):
     return get_pipeline_response(registered_pipeline_id)
 
 
+@router.get(
+    "/version",
+    response_model=VersionAPIResponse,
+)
+async def get_version():
+    return get_version_response()
+
+
 @router.post("/deploy")
 async def deploy_kedro_viz(input_values: DeployerConfiguration):
     from kedro_viz.integrations.deployment.deployer_factory import DeployerFactory

package/tests/test_api/test_rest/test_router.py

@@ -90,3 +90,17 @@ def test_metadata(
     mock_get_metadata_response.assert_called_once()
     assert response.status_code == expected_status_code
     assert response.json() == expected_response
+
+
+def test_version(client):
+    response = client.get("/api/version")
+    assert response.status_code == 200
+
+    json_response = response.json()
+    assert "installed" in json_response
+    assert "is_outdated" in json_response
+    assert "latest" in json_response
+
+    assert isinstance(json_response["installed"], str)
+    assert isinstance(json_response["is_outdated"], bool)
+    assert isinstance(json_response["latest"], str)

src/components/wrapper/wrapper.js

@@ -7,6 +7,7 @@ import { useApolloQuery } from '../../apollo/utils';
 import { client } from '../../apollo/config';
 import { GraphQLProvider } from '../provider/provider';
 import { GET_VERSIONS } from '../../apollo/queries';
+import { getVersion } from '../../utils';
 
 import FeatureHints from '../feature-hints';
 import GlobalToolbar from '../global-toolbar';
@@ -29,11 +30,22 @@ export const Wrapper = ({ displayGlobalNavigation, theme }) => {
   const [latestVersion, setLatestVersion] = useState(null);
 
   useEffect(() => {
-    if (versionData) {
-      setIsOutdated(versionData.version.isOutdated);
-      setLatestVersion(versionData.version.latest);
+    async function checkKedroVizVersion() {
+      try {
+        const request = await getVersion();
+        const response = await request.json();
+
+        if (request.ok) {
+          setIsOutdated(response.is_outdated);
+          setLatestVersion(response.latest);
+        }
+      } catch (error) {
+        console.error('Error fetching Kedro-Viz version:', error);
+      }
     }
-  }, [versionData]);
+
+    checkKedroVizVersion();
+  }, []);
 
   return (
     <div

src/utils/index.js

@@ -248,6 +248,16 @@ export async function deployViz(inputValues) {
   return request;
 }
 
+export async function getVersion() {
+  const request = await fetch(`${pathRoot}/version`, {
+    headers: {
+      'Content-Type': 'application/json',
+      Accept: 'application/json',
+    },
+  });
+  return request;
+}
+
 const nodeTypeMapObj = {
   nodes: 'task',
   task: 'nodes',