chore(tasks): Add trace headers injection and update payload filtering

Author: LucienShui

src/tasks/dto/task.dto.ts

@@ -41,7 +41,7 @@ export class TaskMetaDto {
     if (task.payload) {
       obj.attrs = {};
       for (const [key, value] of Object.entries(task.payload)) {
-        if (key !== 'otel_headers') {
+        if (key !== 'trace_headers') {
           obj.attrs[key] = value;
         }
       }

src/tasks/wizard-task.service.ts

@@ -7,16 +7,24 @@ import {
   Message,
   OpenAIMessageRole,
 } from 'omniboxd/messages/entities/message.entity';
+import { context, propagation } from '@opentelemetry/api';
 
 @Injectable()
 export class WizardTaskService {
   constructor(
     @InjectRepository(Task) public taskRepository: Repository<Task>,
   ) {}
 
+  injectTraceHeaders(task: Partial<Task>) {
+    const traceHeaders: Record<string, string> = {};
+    propagation.inject(context.active(), traceHeaders);
+    task.payload = { ...(task.payload || {}), trace_headers: traceHeaders };
+    return task;
+  }
+
   async create(data: Partial<Task>, repo?: Repository<Task>) {
     const repository = repo || this.taskRepository;
-    const task = repository.create(data);
+    const task = repository.create(this.injectTraceHeaders(data));
     return await repository.save(task);
   }