From b968fae19e5af5caa988faadcb563ba50a99e577 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 23 Oct 2023 20:42:45 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A5=20refactor(store.py):=20remove=20u?= =?UTF-8?q?nused=20code=20and=20endpoint=20'fork=5Fcomponent'=20to=20impro?= =?UTF-8?q?ve=20code=20readability=20and=20maintainability?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/api/v1/store.py | 51 ---------------------------- 1 file changed, 51 deletions(-) diff --git a/src/backend/langflow/api/v1/store.py b/src/backend/langflow/api/v1/store.py index 55b53f88f..0560033a3 100644 --- a/src/backend/langflow/api/v1/store.py +++ b/src/backend/langflow/api/v1/store.py @@ -1,7 +1,6 @@ from typing import List, Optional from uuid import UUID from langflow.services.auth import utils as auth_utils -from langflow.services.database.models.flow.flow import FlowCreate, FlowRead from langflow.services.database.models.user.user import User from langflow.services.deps import ( get_session, @@ -89,56 +88,6 @@ def read_component( return component -@router.post("/components/{component_id}/fork", response_model=FlowRead) -def fork_component( - component_id: UUID, - store_service: StoreService = Depends(get_store_service), - store_api_Key: str = Depends(get_user_store_api_key), - settings_service=Depends(get_settings_service), - user: User = Depends(auth_utils.get_current_active_user), - session=Depends(get_session), -): - # If the component is from the store, we need to get it from the store - try: - decrypted = auth_utils.decrypt_api_key(store_api_Key, settings_service) - component = store_service.get(decrypted, component_id) - except Exception as exc: - raise HTTPException(status_code=400, detail=str(exc)) from exc - - if component is None: - raise HTTPException(status_code=400, detail="Component not found") - - # Now we need to get the component id and put it in StoreComponentCreate.parent - parent_id = component.id - new_component = StoreComponentCreate( - name=component.name, - data=component.data, - parent=parent_id, - description=component.description, - tags=component.tags, - ) - - try: - created_component = store_service.upload(decrypted, new_component.dict()) - except Exception as exc: - raise HTTPException(status_code=500, detail=str(exc)) from exc - - if created_component is None: - raise HTTPException(status_code=500, detail="Component not created") - - # Now save it locally - db_component = FlowCreate( - name=created_component.name, - data=created_component.data, - description=created_component.description, - user_id=user.id, - ) - session.add(db_component) - session.commit() - session.refresh(db_component) - return db_component - - @router.get("/search", response_model=List[ComponentResponse]) async def search_endpoint( query: str = Query(...),