update schema for store endpoints

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-19 22:14:45 -03:00
commit 54c2535b95

View file

@ -1,13 +1,12 @@
from typing import List, Optional from typing import List, Optional
from uuid import UUID from uuid import UUID
from langflow.services.auth import utils as auth_utils from langflow.services.auth import utils as auth_utils
from langflow.services.database.models.flow.flow import FlowCreate
from langflow.services.database.models.user.user import User from langflow.services.database.models.user.user import User
from langflow.services.deps import ( from langflow.services.deps import (
get_store_service, get_store_service,
get_settings_service, get_settings_service,
) )
from langflow.services.store.schema import ComponentResponse from langflow.services.store.schema import ComponentResponse, StoreComponentCreate
from fastapi import APIRouter, Depends, HTTPException, Query from fastapi import APIRouter, Depends, HTTPException, Query
from datetime import datetime from datetime import datetime
@ -28,7 +27,7 @@ def get_user_store_api_key(user: User = Depends(auth_utils.get_current_active_us
@router.post("/", response_model=ComponentResponse) @router.post("/", response_model=ComponentResponse)
def create_component( def create_component(
component: FlowCreate, component: StoreComponentCreate,
store_service: StoreService = Depends(get_store_service), store_service: StoreService = Depends(get_store_service),
settings_service=Depends(get_settings_service), settings_service=Depends(get_settings_service),
store_api_Key: str = Depends(get_user_store_api_key), store_api_Key: str = Depends(get_user_store_api_key),
@ -69,7 +68,8 @@ def list_components(
): ):
try: try:
decrypted = auth_utils.decrypt_api_key(store_api_Key, settings_service) decrypted = auth_utils.decrypt_api_key(store_api_Key, settings_service)
return store_service.list_components(decrypted, page, limit) result = store_service.list_components(decrypted, page, limit)
return result
except Exception as exc: except Exception as exc:
raise HTTPException(status_code=400, detail=str(exc)) raise HTTPException(status_code=400, detail=str(exc))