jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add support for auto login in read_flows and read_flow

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-06 11:45:27 -03:00
commit d540b0b5c0
1 changed files with 21 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/backend/langflow/api/v1/flows.py
View file

@ -20,6 +20,7 @@ from langflow.services.database.models.flow import (
) )
from langflow.services.database.models.user.model import User from langflow.services.database.models.user.model import User
from langflow.services.deps import get_session, get_settings_service from langflow.services.deps import get_session, get_settings_service
from langflow.services.settings.service import SettingsService
# build router # build router
router = APIRouter(prefix="/flows", tags=["Flows"]) router = APIRouter(prefix="/flows", tags=["Flows"])
@ -50,10 +51,16 @@ def read_flows(
*, *,
current_user: User = Depends(get_current_active_user), current_user: User = Depends(get_current_active_user),
session: Session = Depends(get_session), session: Session = Depends(get_session),
settings_service: "SettingsService" = Depends(get_settings_service),
): ):
"""Read all flows.""" """Read all flows."""
try: try:
auth_settings = settings_service.auth_settings
if auth_settings.AUTO_LOGIN:
flows = session.exec(select(Flow).where(Flow.user_id == None)).all()
else:
flows = current_user.flows flows = current_user.flows
flows = validate_is_component(flows) flows = validate_is_component(flows)
# with the session get the flows that DO NOT have a user_id # with the session get the flows that DO NOT have a user_id
try: try:
@ -76,11 +83,17 @@ def read_flow(
session: Session = Depends(get_session), session: Session = Depends(get_session),
flow_id: UUID, flow_id: UUID,
current_user: User = Depends(get_current_active_user), current_user: User = Depends(get_current_active_user),
settings_service: "SettingsService" = Depends(get_settings_service),
): ):
"""Read a flow.""" """Read a flow."""
auth_settings = settings_service.auth_settings
if auth_settings.AUTO_LOGIN:
user_id = None
else:
user_id = current_user.id
if user_flow := ( if user_flow := (
session.exec( session.exec(
select(Flow).where(Flow.id == flow_id, Flow.user_id == current_user.id) select(Flow).where(Flow.id == flow_id, Flow.user_id == user_id)
).first() ).first()
): ):
return user_flow return user_flow
@ -99,7 +112,12 @@ def update_flow(
): ):
"""Update a flow.""" """Update a flow."""
db_flow = read_flow(session=session, flow_id=flow_id, current_user=current_user) db_flow = read_flow(
session=session,
flow_id=flow_id,
current_user=current_user,
settings_service=settings_service,
)
if not db_flow: if not db_flow:
raise HTTPException(status_code=404, detail="Flow not found") raise HTTPException(status_code=404, detail="Flow not found")
flow_data = flow.model_dump(exclude_unset=True) flow_data = flow.model_dump(exclude_unset=True)