Add authorization check for store_api_Key

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-17 09:20:37 -03:00
commit e3b0941df9

View file

@ -4,6 +4,8 @@ from uuid import UUID
import httpx import httpx
from httpx import HTTPError, HTTPStatusError from httpx import HTTPError, HTTPStatusError
from loguru import logger
from langflow.services.base import Service from langflow.services.base import Service
from langflow.services.store.schema import ( from langflow.services.store.schema import (
CreateComponentResponse, CreateComponentResponse,
@ -13,7 +15,6 @@ from langflow.services.store.schema import (
StoreComponentCreate, StoreComponentCreate,
) )
from langflow.services.store.utils import process_tags_for_post, update_components_with_user_data from langflow.services.store.utils import process_tags_for_post, update_components_with_user_data
from loguru import logger
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.services.settings.service import SettingsService from langflow.services.settings.service import SettingsService
@ -469,4 +470,6 @@ class StoreService(Service):
except Exception: except Exception:
# If we get an error here, it means the user is not authorized # If we get an error here, it means the user is not authorized
authorized = False authorized = False
elif store_api_Key and not result:
authorized = True
return ListComponentResponseModel(results=result, authorized=authorized, count=comp_count) return ListComponentResponseModel(results=result, authorized=authorized, count=comp_count)