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

updates imports to use settings_manager

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-05 23:20:35 -03:00
commit 6ca7308e3c
25 changed files with 182 additions and 269 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,6 +1,5 @@
from typing import List
from uuid import UUID
from langflow.settings import settings
from langflow.api.utils import remove_api_keys
from langflow.api.v1.schemas import FlowListCreate, FlowListRead
from langflow.services.database.models.flow import (
@ -10,7 +9,8 @@ from langflow.services.database.models.flow import (
FlowReadWithStyle,
FlowUpdate,
)
from langflow.services.database.base import get_session
from langflow.services.utils import get_session
from langflow.services.utils import get_settings_manager
from sqlmodel import Session, select
from fastapi import APIRouter, Depends, HTTPException
from fastapi.encoders import jsonable_encoder
@ -61,7 +61,8 @@ def update_flow(
if not db_flow:
raise HTTPException(status_code=404, detail="Flow not found")
flow_data = flow.dict(exclude_unset=True)
if settings.REMOVE_API_KEYS:
settings_manager = get_settings_manager()
if settings_manager.settings.REMOVE_API_KEYS:
flow_data = remove_api_keys(flow_data)
for key, value in flow_data.items():
setattr(db_flow, key, value)