diff --git a/packages/dd-trace/src/llmobs/writers/spans/base.js b/packages/dd-trace/src/llmobs/writers/spans/base.js index b8575d1a010..0cf15a7c968 100644 --- a/packages/dd-trace/src/llmobs/writers/spans/base.js +++ b/packages/dd-trace/src/llmobs/writers/spans/base.js @@ -41,12 +41,12 @@ class LLMObsSpanWriter extends BaseWriter { } makePayload (events) { - return { + return events.map(event => ({ '_dd.stage': 'raw', '_dd.tracer_version': tracerVersion, event_type: this._eventType, - spans: events - } + spans: [event] + })) } _truncateSpanEvent (event) { diff --git a/packages/dd-trace/test/llmobs/sdk/typescript/index.spec.js b/packages/dd-trace/test/llmobs/sdk/typescript/index.spec.js index 111123b1362..f8a9c33cf41 100644 --- a/packages/dd-trace/test/llmobs/sdk/typescript/index.spec.js +++ b/packages/dd-trace/test/llmobs/sdk/typescript/index.spec.js @@ -38,7 +38,7 @@ const testCases = [ file: 'index', setup: (agent, results = {}) => { const llmobsRes = agent.assertLlmObsPayloadReceived(({ payload }) => { - results.llmobsSpans = payload.spans + results.llmobsSpans = payload.flatMap(item => item.spans) }) const apmRes = agent.assertMessageReceived(({ payload }) => { diff --git a/packages/dd-trace/test/llmobs/writers/spans/base.spec.js b/packages/dd-trace/test/llmobs/writers/spans/base.spec.js index 1c9965cd9c2..6bb98d9a07b 100644 --- a/packages/dd-trace/test/llmobs/writers/spans/base.spec.js +++ b/packages/dd-trace/test/llmobs/writers/spans/base.spec.js @@ -92,8 +92,8 @@ describe('LLMObsSpanWriter', () => { const payload = writer.makePayload(events) - expect(payload['_dd.stage']).to.equal('raw') - expect(payload.event_type).to.equal('span') - expect(payload.spans).to.deep.equal(events) + expect(payload[0]['_dd.stage']).to.equal('raw') + expect(payload[0].event_type).to.equal('span') + expect(payload[0].spans).to.deep.equal(events) }) })