refactor: Update get_version endpoint to use get_version_info utility function

The get_version endpoint in the API router has been updated to use the get_version_info utility function from langflow.utils.version. This change improves code organization and maintainability by centralizing the version retrieval logic.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-23 00:33:35 -03:00
commit 10eaa77399
2 changed files with 14 additions and 11 deletions

View file

@ -3,6 +3,7 @@ from http import HTTPStatus
from typing import TYPE_CHECKING, Annotated, List, Optional, Union from typing import TYPE_CHECKING, Annotated, List, Optional, Union
from uuid import UUID from uuid import UUID
from langflow.utils.version import get_version_info
import sqlalchemy as sa import sqlalchemy as sa
from fastapi import APIRouter, BackgroundTasks, Body, Depends, HTTPException, Request, UploadFile, status from fastapi import APIRouter, BackgroundTasks, Body, Depends, HTTPException, Request, UploadFile, status
from loguru import logger from loguru import logger
@ -445,17 +446,7 @@ async def create_upload_file(
# get endpoint to return version of langflow # get endpoint to return version of langflow
@router.get("/version") @router.get("/version")
def get_version(): def get_version():
try: return get_version_info()
from langflow.version import __version__ # type: ignore
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, response_model=CustomComponentResponse) @router.post("/custom_component", status_code=HTTPStatus.OK, response_model=CustomComponentResponse)

View file

@ -0,0 +1,12 @@
def get_version_info():
try:
from langflow.version import __version__ # type: ignore
version = __version__
package = "Langflow"
except ImportError:
from importlib import metadata
version = metadata.version("langflow-base")
package = "Langflow Base"
return {"version": version, "package": package}