From 997639e325a1084083998af3792880a077d8225b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 4 Oct 2023 13:40:51 -0300 Subject: [PATCH 1/2] =?UTF-8?q?=F0=9F=90=9B=20fix(endpoints.py):=20replace?= =?UTF-8?q?=20backend=20type=20string=20with=20backend=20name=20to=20impro?= =?UTF-8?q?ve=20readability=20and=20semantics=20=E2=9C=A8=20feat(anyio.py)?= =?UTF-8?q?:=20add=20name=20attribute=20to=20AnyIOBackend=20class=20to=20i?= =?UTF-8?q?dentify=20the=20backend=20by=20name=20=E2=9C=A8=20feat(celery.p?= =?UTF-8?q?y):=20add=20name=20attribute=20to=20CeleryBackend=20class=20to?= =?UTF-8?q?=20identify=20the=20backend=20by=20name=20=E2=9C=A8=20feat(mana?= =?UTF-8?q?ger.py):=20add=20backend=5Fname=20property=20to=20TaskService?= =?UTF-8?q?=20class=20to=20get=20the=20name=20of=20the=20backend?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/api/v1/endpoints.py | 2 +- src/backend/langflow/services/task/backends/anyio.py | 2 ++ src/backend/langflow/services/task/backends/celery.py | 2 ++ src/backend/langflow/services/task/manager.py | 4 ++++ 4 files changed, 9 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/api/v1/endpoints.py b/src/backend/langflow/api/v1/endpoints.py index 864467f9b..4e68193d8 100644 --- a/src/backend/langflow/api/v1/endpoints.py +++ b/src/backend/langflow/api/v1/endpoints.py @@ -156,7 +156,7 @@ async def process_flow( result=task_result, task=task_response, session_id=session_id, - backend=str(type(task_service.backend)), + backend=task_service.backend_name, ) except sa.exc.StatementError as exc: # StatementError('(builtins.ValueError) badly formed hexadecimal UUID string') diff --git a/src/backend/langflow/services/task/backends/anyio.py b/src/backend/langflow/services/task/backends/anyio.py index ca91eca19..49c9c0e4f 100644 --- a/src/backend/langflow/services/task/backends/anyio.py +++ b/src/backend/langflow/services/task/backends/anyio.py @@ -34,6 +34,8 @@ class AnyIOTaskResult: class AnyIOBackend(TaskBackend): + name = "anyio" + def __init__(self): self.tasks = {} diff --git a/src/backend/langflow/services/task/backends/celery.py b/src/backend/langflow/services/task/backends/celery.py index 5b0a035ef..eae985f3a 100644 --- a/src/backend/langflow/services/task/backends/celery.py +++ b/src/backend/langflow/services/task/backends/celery.py @@ -5,6 +5,8 @@ from langflow.worker import celery_app class CeleryBackend(TaskBackend): + name = "celery" + def __init__(self): self.celery_app = celery_app diff --git a/src/backend/langflow/services/task/manager.py b/src/backend/langflow/services/task/manager.py index 422b34faa..807505c3a 100644 --- a/src/backend/langflow/services/task/manager.py +++ b/src/backend/langflow/services/task/manager.py @@ -35,6 +35,10 @@ class TaskService(Service): self.backend = self.get_backend() self.use_celery = USE_CELERY + @property + def backend_name(self) -> str: + return self.backend.name + def get_backend(self) -> TaskBackend: if USE_CELERY: from langflow.services.task.backends.celery import CeleryBackend From 6276e5c233a716f33e91c1cec79faf1eeaa6d774 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 4 Oct 2023 13:42:20 -0300 Subject: [PATCH 2/2] =?UTF-8?q?=F0=9F=94=96=20chore(pyproject.toml):=20bum?= =?UTF-8?q?p=20version=20from=200.5.0b4=20to=200.5.0b5=20for=20package=20r?= =?UTF-8?q?elease?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index cc00ed6ed..9a9bfcf77 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "langflow" -version = "0.5.0b4" +version = "0.5.0b5" description = "A Python package with a built-in web application" authors = ["Logspace "] maintainers = [