mirror of
https://github.com/langflow-ai/langflow.git
synced 2026-07-24 02:05:11 +08:00
f2e3f2ccc5
* Remove models from __init__.py to avoid circular dependency * [autofix.ci] apply automated fixes * [autofix.ci] apply automated fixes (attempt 2/3) * add back init to avoid backwards-compat issues * [autofix.ci] apply automated fixes * ref: standardize imports to use full qualified names Replaces uses of the "public" APIs to use full qualified names to reduce instances of circular dependencies * [autofix.ci] apply automated fixes * [autofix.ci] apply automated fixes (attempt 2/3) * Fix path * update starters" " * remove old files' * [autofix.ci] apply automated fixes * Fix some imports * [autofix.ci] apply automated fixes * ruff * [autofix.ci] apply automated fixes * starter projects? * starter projects? * [autofix.ci] apply automated fixes * [autofix.ci] apply automated fixes --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
107 lines
3.7 KiB
Python
107 lines
3.7 KiB
Python
from uuid import UUID
|
|
|
|
from fastapi import APIRouter, HTTPException
|
|
from sqlalchemy.exc import NoResultFound
|
|
|
|
from langflow.api.utils import CurrentActiveUser, DbSession
|
|
from langflow.services.database.models.variable.model import VariableCreate, VariableRead, VariableUpdate
|
|
from langflow.services.deps import get_variable_service
|
|
from langflow.services.variable.constants import CREDENTIAL_TYPE
|
|
from langflow.services.variable.service import DatabaseVariableService
|
|
|
|
router = APIRouter(prefix="/variables", tags=["Variables"])
|
|
|
|
|
|
@router.post("/", response_model=VariableRead, status_code=201)
|
|
async def create_variable(
|
|
*,
|
|
session: DbSession,
|
|
variable: VariableCreate,
|
|
current_user: CurrentActiveUser,
|
|
):
|
|
"""Create a new variable."""
|
|
variable_service = get_variable_service()
|
|
if not variable.name and not variable.value:
|
|
raise HTTPException(status_code=400, detail="Variable name and value cannot be empty")
|
|
|
|
if not variable.name:
|
|
raise HTTPException(status_code=400, detail="Variable name cannot be empty")
|
|
|
|
if not variable.value:
|
|
raise HTTPException(status_code=400, detail="Variable value cannot be empty")
|
|
|
|
if variable.name in await variable_service.list_variables(user_id=current_user.id, session=session):
|
|
raise HTTPException(status_code=400, detail="Variable name already exists")
|
|
try:
|
|
return await variable_service.create_variable(
|
|
user_id=current_user.id,
|
|
name=variable.name,
|
|
value=variable.value,
|
|
default_fields=variable.default_fields or [],
|
|
type_=variable.type or CREDENTIAL_TYPE,
|
|
session=session,
|
|
)
|
|
except Exception as e:
|
|
if isinstance(e, HTTPException):
|
|
raise
|
|
raise HTTPException(status_code=500, detail=str(e)) from e
|
|
|
|
|
|
@router.get("/", response_model=list[VariableRead], status_code=200)
|
|
async def read_variables(
|
|
*,
|
|
session: DbSession,
|
|
current_user: CurrentActiveUser,
|
|
):
|
|
"""Read all variables."""
|
|
variable_service = get_variable_service()
|
|
if not isinstance(variable_service, DatabaseVariableService):
|
|
msg = "Variable service is not an instance of DatabaseVariableService"
|
|
raise TypeError(msg)
|
|
try:
|
|
return await variable_service.get_all(user_id=current_user.id, session=session)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e)) from e
|
|
|
|
|
|
@router.patch("/{variable_id}", response_model=VariableRead, status_code=200)
|
|
async def update_variable(
|
|
*,
|
|
session: DbSession,
|
|
variable_id: UUID,
|
|
variable: VariableUpdate,
|
|
current_user: CurrentActiveUser,
|
|
):
|
|
"""Update a variable."""
|
|
variable_service = get_variable_service()
|
|
if not isinstance(variable_service, DatabaseVariableService):
|
|
msg = "Variable service is not an instance of DatabaseVariableService"
|
|
raise TypeError(msg)
|
|
try:
|
|
return await variable_service.update_variable_fields(
|
|
user_id=current_user.id,
|
|
variable_id=variable_id,
|
|
variable=variable,
|
|
session=session,
|
|
)
|
|
except NoResultFound as e:
|
|
raise HTTPException(status_code=404, detail="Variable not found") from e
|
|
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e)) from e
|
|
|
|
|
|
@router.delete("/{variable_id}", status_code=204)
|
|
async def delete_variable(
|
|
*,
|
|
session: DbSession,
|
|
variable_id: UUID,
|
|
current_user: CurrentActiveUser,
|
|
) -> None:
|
|
"""Delete a variable."""
|
|
variable_service = get_variable_service()
|
|
try:
|
|
await variable_service.delete_variable_by_id(user_id=current_user.id, variable_id=variable_id, session=session)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e)) from e
|