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

🔧 fix(store.py): import missing module 'langflow.api' to resolve NameError

Browse source 
🔧 fix(store.py): add missing metadata comment to clarify the purpose of the code block
✨ feat(store.py): add 'has_api_key' field to the response of 'check_if_store_is_enabled' endpoint to indicate if the user has an API key
🔧 fix(store.py): add missing metadata using the Component.data in the 'create_component' endpoint to properly handle component metadata
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-27 11:37:54 -03:00
commit 5716ca21dc
1 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,5 +1,6 @@
from typing import List, Optional from typing import List, Optional
from uuid import UUID from uuid import UUID
from langflow import api
from langflow.services.auth import utils as auth_utils from langflow.services.auth import utils as auth_utils
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 (
@ -49,8 +50,12 @@ def get_optional_user_store_api_key(
@router.get("/") @router.get("/")
def check_if_store_is_enabled( def check_if_store_is_enabled(
settings_service=Depends(get_settings_service), settings_service=Depends(get_settings_service),
api_key=Depends(get_optional_user_store_api_key),
): ):
return {"enabled": settings_service.settings.STORE} return {
"enabled": settings_service.settings.STORE,
"has_api_key": api_key is not None,
}
@router.post("/components/", response_model=ComponentResponse, status_code=201) @router.post("/components/", response_model=ComponentResponse, status_code=201)
@ -60,6 +65,9 @@ def create_component(
store_api_Key: str = Depends(get_user_store_api_key), store_api_Key: str = Depends(get_user_store_api_key),
): ):
try: try:
# We need to add the metadata using the Component.data
#
return store_service.upload(store_api_Key, component) return store_service.upload(store_api_Key, component)
except Exception as exc: except Exception as exc:
raise HTTPException(status_code=400, detail=str(exc)) raise HTTPException(status_code=400, detail=str(exc))