diff --git a/src/backend/base/langflow/__main__.py b/src/backend/base/langflow/__main__.py
index 34ed75f9b..96f4f1575 100644
--- a/src/backend/base/langflow/__main__.py
+++ b/src/backend/base/langflow/__main__.py
@@ -246,9 +246,17 @@ def get_free_port(port):
 
 
 def print_banner(host, port):
-    from langflow.version import __version__
+    try:
+        from langflow.version import __version__
+
+        version = __version__
+        word = "Langflow"
+    except ImportError:
+        from importlib import metadata
+
+        version = metadata.version("langflow-base")
+        word = "Langflow Base"
 
-    word = "Langflow"
     colors = ["#6e42f5"]
 
     styled_word = ""
@@ -259,7 +267,7 @@ def print_banner(host, port):
 
     # Title with emojis and gradient text
     title = (
-        f"[bold]Welcome to :chains: {styled_word} v{__version__}[/bold]\n"
+        f"[bold]Welcome to :chains: {styled_word} v{version}[/bold]\n"
         f"Access [link=http://{host}:{port}]http://{host}:{port}[/link]"
     )
     info_text = (
diff --git a/src/backend/base/langflow/api/v1/endpoints.py b/src/backend/base/langflow/api/v1/endpoints.py
index cdcf40443..4b032f6cb 100644
--- a/src/backend/base/langflow/api/v1/endpoints.py
+++ b/src/backend/base/langflow/api/v1/endpoints.py
@@ -372,9 +372,17 @@ async def create_upload_file(
 # get endpoint to return version of langflow
 @router.get("/version")
 def get_version():
-    from langflow.version import __version__  # type: ignore
+    try:
+        from langflow.version import __version__
 
-    return {"version": __version__}
+        version = __version__
+        package = "Langflow"
+    except ImportError:
+        from importlib import metadata
+
+        version = metadata.version("langflow-base")
+        package = "Langflow Base"
+    return {"version": version, "package": package}
 
 
 @router.post("/custom_component", status_code=HTTPStatus.OK)