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 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)
|
||||||
|
|
|
||||||
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