include get_latest_version, is_running_outdated_version

Signed-off-by: Huong Nguyen <[email protected]>
kedro-org · Jan 17, 2025 · 3dd6315 · 3dd6315
1 parent 15ab3e8
commit 3dd6315
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 13 deletions.
diff --git a/package/kedro_viz/api/rest/responses/version.py b/package/kedro_viz/api/rest/responses/version.py
@@ -2,7 +2,8 @@
 and utility functions for the `/version` REST endpoint"""
 
 from kedro_viz import __version__
-from pydantic import BaseModel, ConfigDict
+from kedro_viz.integrations.pypi import get_latest_version, is_running_outdated_version
+from pydantic import ConfigDict
 from kedro_viz.api.rest.responses.base import BaseAPIResponse
 
 class VersionAPIResponse(BaseAPIResponse):
@@ -22,15 +23,15 @@ class VersionAPIResponse(BaseAPIResponse):
         json_schema_extra={
             "installed": __version__,
             "isOutdated": False,  
-            "latest": "0.5.0" # how do i check the latest version?
+            "latest": "0.0.0"
         }
     )
 
 def get_version_response():
     """API response for `/api/version`."""
     installed_version = __version__
-    latest_version = "0.5.0"  # how do i check the latest version?
-    is_outdated = installed_version != latest_version 
+    latest_version = get_latest_version()
+    is_outdated = is_running_outdated_version(installed_version, latest_version)
 
     return VersionAPIResponse(
         installed=installed_version,

diff --git a/package/kedro_viz/api/rest/router.py b/package/kedro_viz/api/rest/router.py
@@ -58,7 +58,7 @@ async def get_single_pipeline_data(registered_pipeline_id: str):
     response_model=VersionAPIResponse,
 )
 
-asyn def get_version():
+async def get_version():
     return get_version_response()
 
 

diff --git a/src/components/wrapper/wrapper.js b/src/components/wrapper/wrapper.js
@@ -48,13 +48,6 @@ export const Wrapper = ({ displayGlobalNavigation, theme }) => {
     checkKedroVizVersion();
   }, []);
 
-  // useEffect(() => {
-  //   if (versionData) {
-  //     setIsOutdated(versionData.version.isOutdated);
-  //     setLatestVersion(versionData.version.latest);
-  //   }
-  // }, [versionData]);
-
   return (
     <div
       className={classnames('kedro-pipeline kedro', {

diff --git a/src/utils/index.js b/src/utils/index.js
@@ -249,7 +249,7 @@ export async function deployViz(inputValues) {
 }
 
 export async function getVersion() {
-  const request = await fetch(`api/version`, {
+  const request = await fetch(`${pathRoot}/version`, {
     headers: {
       'Content-Type': 'application/json',
       Accept: 'application/json',