🐛 fix(endpoints.py): replace backend type string with backend name to improve readability and semantics (#1010)
This commit is contained in:
commit
6c45b6393f
5 changed files with 10 additions and 2 deletions
|
|
@ -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 = [
|
||||
|
|
|
|||
|
|
@ -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')
|
||||
|
|
|
|||
|
|
@ -34,6 +34,8 @@ class AnyIOTaskResult:
|
|||
|
||||
|
||||
class AnyIOBackend(TaskBackend):
|
||||
name = "anyio"
|
||||
|
||||
def __init__(self):
|
||||
self.tasks = {}
|
||||
|
||||
|
|
|
|||
|
|
@ -5,6 +5,8 @@ from langflow.worker import celery_app
|
|||
|
||||
|
||||
class CeleryBackend(TaskBackend):
|
||||
name = "celery"
|
||||
|
||||
def __init__(self):
|
||||
self.celery_app = celery_app
|
||||
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue