🔥 refactor(store.py): remove unused code and endpoint 'fork_component' to improve code readability and maintainability
This commit is contained in:
parent
1033369795
commit
b968fae19e
1 changed files with 0 additions and 51 deletions
|
|
@ -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(...),
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue