From da24690bf70b9f1b424c0529a495818dce701df6 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 15 Jun 2023 18:16:02 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=92=20chore(flows.py):=20add=20support?= =?UTF-8?q?=20for=20removing=20API=20keys=20from=20flow=20data=20if=20save?= =?UTF-8?q?=5Fapi=5Fkeys=20setting=20is=20False=20The=20`remove=5Fapi=5Fke?= =?UTF-8?q?ys`=20function=20is=20now=20called=20on=20the=20`flow=5Fdata`?= =?UTF-8?q?=20dictionary=20if=20the=20`save=5Fapi=5Fkeys`=20setting=20is?= =?UTF-8?q?=20False.=20This=20ensures=20that=20sensitive=20information=20i?= =?UTF-8?q?s=20not=20saved=20in=20the=20database.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/api/v1/flows.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/backend/langflow/api/v1/flows.py b/src/backend/langflow/api/v1/flows.py index b707f5e05..6534d83e6 100644 --- a/src/backend/langflow/api/v1/flows.py +++ b/src/backend/langflow/api/v1/flows.py @@ -1,5 +1,7 @@ from typing import List from uuid import UUID +from langflow import settings +from langflow.api.utils import remove_api_keys from langflow.api.v1.schemas import FlowListCreate, FlowListRead from langflow.database.models.flow import ( Flow, @@ -54,10 +56,13 @@ def update_flow( *, session: Session = Depends(get_session), flow_id: UUID, flow: FlowUpdate ): """Update a flow.""" + db_flow = session.get(Flow, flow_id) if not db_flow: raise HTTPException(status_code=404, detail="Flow not found") flow_data = flow.dict(exclude_unset=True) + if settings.save_api_keys: + flow_data = remove_api_keys(flow_data) for key, value in flow_data.items(): setattr(db_flow, key, value) session.add(db_flow)