diff --git a/src/backend/langflow/__init__.py b/src/backend/langflow/__init__.py index 17b1d940c..9b80c2ea7 100644 --- a/src/backend/langflow/__init__.py +++ b/src/backend/langflow/__init__.py @@ -1,4 +1,12 @@ +from importlib import metadata from langflow.cache import cache_manager from langflow.processing.process import load_flow_from_json +try: + __version__ = metadata.version(__package__) +except metadata.PackageNotFoundError: + # Case where package metadata is not available. + __version__ = "" +del metadata # optional, avoids polluting the results of dir(__package__) + __all__ = ["load_flow_from_json", "cache_manager"] diff --git a/src/backend/langflow/api/v1/endpoints.py b/src/backend/langflow/api/v1/endpoints.py index 1e9b0deb1..0427eb291 100644 --- a/src/backend/langflow/api/v1/endpoints.py +++ b/src/backend/langflow/api/v1/endpoints.py @@ -41,4 +41,6 @@ async def get_load(predict_request: PredictRequest): # get endpoint to return version of langflow @router.get("/version") def get_version(): - return {"version": version("langflow")} + from langflow import __version__ + + return {"version": __version__}