🐛 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]
|
[tool.poetry]
|
||||||
name = "langflow"
|
name = "langflow"
|
||||||
version = "0.5.0b4"
|
version = "0.5.0b5"
|
||||||
description = "A Python package with a built-in web application"
|
description = "A Python package with a built-in web application"
|
||||||
authors = ["Logspace <contact@logspace.ai>"]
|
authors = ["Logspace <contact@logspace.ai>"]
|
||||||
maintainers = [
|
maintainers = [
|
||||||
|
|
|
||||||
|
|
@ -156,7 +156,7 @@ async def process_flow(
|
||||||
result=task_result,
|
result=task_result,
|
||||||
task=task_response,
|
task=task_response,
|
||||||
session_id=session_id,
|
session_id=session_id,
|
||||||
backend=str(type(task_service.backend)),
|
backend=task_service.backend_name,
|
||||||
)
|
)
|
||||||
except sa.exc.StatementError as exc:
|
except sa.exc.StatementError as exc:
|
||||||
# StatementError('(builtins.ValueError) badly formed hexadecimal UUID string')
|
# StatementError('(builtins.ValueError) badly formed hexadecimal UUID string')
|
||||||
|
|
|
||||||
|
|
@ -34,6 +34,8 @@ class AnyIOTaskResult:
|
||||||
|
|
||||||
|
|
||||||
class AnyIOBackend(TaskBackend):
|
class AnyIOBackend(TaskBackend):
|
||||||
|
name = "anyio"
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.tasks = {}
|
self.tasks = {}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -5,6 +5,8 @@ from langflow.worker import celery_app
|
||||||
|
|
||||||
|
|
||||||
class CeleryBackend(TaskBackend):
|
class CeleryBackend(TaskBackend):
|
||||||
|
name = "celery"
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.celery_app = celery_app
|
self.celery_app = celery_app
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -35,6 +35,10 @@ class TaskService(Service):
|
||||||
self.backend = self.get_backend()
|
self.backend = self.get_backend()
|
||||||
self.use_celery = USE_CELERY
|
self.use_celery = USE_CELERY
|
||||||
|
|
||||||
|
@property
|
||||||
|
def backend_name(self) -> str:
|
||||||
|
return self.backend.name
|
||||||
|
|
||||||
def get_backend(self) -> TaskBackend:
|
def get_backend(self) -> TaskBackend:
|
||||||
if USE_CELERY:
|
if USE_CELERY:
|
||||||
from langflow.services.task.backends.celery import CeleryBackend
|
from langflow.services.task.backends.celery import CeleryBackend
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue