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:
parent
0f5810928d
commit
10eaa77399
2 changed files with 14 additions and 11 deletions
|
|
@ -3,6 +3,7 @@ from http import HTTPStatus
|
|||
from typing import TYPE_CHECKING, Annotated, List, Optional, Union
|
||||
from uuid import UUID
|
||||
|
||||
from langflow.utils.version import get_version_info
|
||||
import sqlalchemy as sa
|
||||
from fastapi import APIRouter, BackgroundTasks, Body, Depends, HTTPException, Request, UploadFile, status
|
||||
from loguru import logger
|
||||
|
|
@ -445,17 +446,7 @@ async def create_upload_file(
|
|||
# get endpoint to return version of langflow
|
||||
@router.get("/version")
|
||||
def get_version():
|
||||
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}
|
||||
return get_version_info()
|
||||
|
||||
|
||||
@router.post("/custom_component", status_code=HTTPStatus.OK, response_model=CustomComponentResponse)
|
||||
|
|
|
|||
12
src/backend/base/langflow/utils/version.py
Normal file
12
src/backend/base/langflow/utils/version.py
Normal 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}
|
||||
Loading…
Add table
Add a link
Reference in a new issue