Refactor store service to include user search by

username
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-01 21:58:09 -03:00
commit 79df6d883a

View file

@ -7,19 +7,16 @@ from httpx import HTTPError, HTTPStatusError
from loguru import logger from loguru import logger
from langflow.services.base import Service from langflow.services.base import Service
from langflow.services.store.exceptions import APIKeyError, FilterError, ForbiddenError from langflow.services.store.exceptions import (APIKeyError, FilterError,
from langflow.services.store.schema import ( ForbiddenError)
CreateComponentResponse, from langflow.services.store.schema import (CreateComponentResponse,
DownloadComponentResponse, DownloadComponentResponse,
ListComponentResponse, ListComponentResponse,
ListComponentResponseModel, ListComponentResponseModel,
StoreComponentCreate, StoreComponentCreate)
) from langflow.services.store.utils import (process_component_data,
from langflow.services.store.utils import ( process_tags_for_post,
process_component_data, update_components_with_user_data)
process_tags_for_post,
update_components_with_user_data,
)
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.services.settings.service import SettingsService from langflow.services.settings.service import SettingsService
@ -167,6 +164,7 @@ class StoreService(Service):
conditions["_or"].append({"name": {"_icontains": query}}) conditions["_or"].append({"name": {"_icontains": query}})
conditions["_or"].append({"description": {"_icontains": query}}) conditions["_or"].append({"description": {"_icontains": query}})
conditions["_or"].append({"tags": {"tags_id": {"name": {"_icontains": query}}}}) conditions["_or"].append({"tags": {"tags_id": {"name": {"_icontains": query}}}})
conditions["_or"].append({"user_created": {"username": {"_icontains": query}}})
return conditions return conditions
def build_filter_conditions( def build_filter_conditions(