🐛 fix(endpoints.py): replace backend type string with backend name to improve readability and semantics (#1010)

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-04 13:42:42 -03:00 committed by GitHub
commit 6c45b6393f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 10 additions and 2 deletions

View file

@ -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 <contact@logspace.ai>"]
maintainers = [

View file

@ -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')

View file

@ -34,6 +34,8 @@ class AnyIOTaskResult:
class AnyIOBackend(TaskBackend):
name = "anyio"
def __init__(self):
self.tasks = {}

View file

@ -5,6 +5,8 @@ from langflow.worker import celery_app
class CeleryBackend(TaskBackend):
name = "celery"
def __init__(self):
self.celery_app = celery_app

View file

@ -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