fix(ops_tracing): enhance error handle in celery tasks. (#10401)

2024-11-08 14:43:47 +08:00
parent 033ab5490b
commit aa3da0e24c
4 changed files with 47 additions and 12 deletions
--- a/api/core/ops/entities/trace_entity.py
+++ b/api/core/ops/entities/trace_entity.py
@@ -23,6 +23,11 @@ class BaseTraceInfo(BaseModel):
            return v
        return ""

+    class Config:
+        json_encoders = {
+            datetime: lambda v: v.isoformat(),
+        }
+

 class WorkflowTraceInfo(BaseTraceInfo):
    workflow_data: Any
@@ -100,6 +105,12 @@ class GenerateNameTraceInfo(BaseTraceInfo):
    tenant_id: str


+class TaskData(BaseModel):
+    app_id: str
+    trace_info_type: str
+    trace_info: Any
+
+
 trace_info_info_map = {
    "WorkflowTraceInfo": WorkflowTraceInfo,
    "MessageTraceInfo": MessageTraceInfo,