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

updates to colliding endpoints and search

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-19 22:52:38 -03:00
commit 4c6dc748a0
2 changed files with 14 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -25,7 +25,13 @@ def get_user_store_api_key(user: User = Depends(auth_utils.get_current_active_us
return user.store_api_key return user.store_api_key
@router.post("/", response_model=ComponentResponse) def get_optional_user_store_api_key(
user: User = Depends(auth_utils.get_current_active_user),
):
return user.store_api_key
@router.post("/components/", response_model=ComponentResponse)
def create_component( def create_component(
component: StoreComponentCreate, component: StoreComponentCreate,
store_service: StoreService = Depends(get_store_service), store_service: StoreService = Depends(get_store_service),
@ -39,7 +45,7 @@ def create_component(
raise HTTPException(status_code=400, detail=str(exc)) raise HTTPException(status_code=400, detail=str(exc))
@router.get("/{component_id}", response_model=ComponentResponse) @router.get("/components/{component_id}", response_model=ComponentResponse)
def read_component( def read_component(
component_id: UUID, component_id: UUID,
store_service: StoreService = Depends(get_store_service), store_service: StoreService = Depends(get_store_service),
@ -58,7 +64,7 @@ def read_component(
return component return component
@router.get("/", response_model=List[ComponentResponse]) @router.get("/components/", response_model=List[ComponentResponse])
def list_components( def list_components(
page: int = 1, page: int = 1,
limit: int = 10, limit: int = 10,
@ -76,7 +82,6 @@ def list_components(
@router.get("/search", response_model=List[ComponentResponse]) @router.get("/search", response_model=List[ComponentResponse])
async def search_endpoint( async def search_endpoint(
api_key: Optional[str] = Query(None),
query: str = Query(...), query: str = Query(...),
page: int = Query(1), page: int = Query(1),
limit: int = Query(10), limit: int = Query(10),
@ -88,10 +93,11 @@ async def search_endpoint(
sort: Optional[List[str]] = Query(None), sort: Optional[List[str]] = Query(None),
fields: Optional[List[str]] = Query(None), fields: Optional[List[str]] = Query(None),
store_service: "StoreService" = Depends(get_store_service), store_service: "StoreService" = Depends(get_store_service),
store_api_Key: str = Depends(get_optional_user_store_api_key),
): ):
try: try:
return await store_service.search( return store_service.search(
api_key=api_key, api_key=store_api_Key,
query=query, query=query,
page=page, page=page,
limit=limit, limit=limit,
@ -104,6 +110,4 @@ async def search_endpoint(
fields=fields, fields=fields,
) )
except Exception as exc: except Exception as exc:
raise HTTPException( raise HTTPException(status_code=500, detail=str(exc))
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(exc)
)

2
src/backend/langflow/services/store/service.py
View file

@ -51,7 +51,7 @@ class StoreService(Service):
fields: Optional[List[str]] = None, fields: Optional[List[str]] = None,
) -> List[ComponentResponse]: ) -> List[ComponentResponse]:
params = { params = {
"filter[name][_like]": query, "search": query,
"page": page, "page": page,
"limit": limit, "limit": limit,
"sort": sort_by, "sort": sort_by,