File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -41,12 +41,12 @@ class LLMObsSpanWriter extends BaseWriter {
4141 }
4242
4343 makePayload ( events ) {
44- return {
44+ return events . map ( event => ( {
4545 '_dd.stage' : 'raw' ,
4646 '_dd.tracer_version' : tracerVersion ,
4747 event_type : this . _eventType ,
48- spans : events
49- }
48+ spans : [ event ]
49+ } ) )
5050 }
5151
5252 _truncateSpanEvent ( event ) {
Original file line number Diff line number Diff line change @@ -38,7 +38,7 @@ const testCases = [
3838 file : 'index' ,
3939 setup : ( agent , results = { } ) => {
4040 const llmobsRes = agent . assertLlmObsPayloadReceived ( ( { payload } ) => {
41- results . llmobsSpans = payload . spans
41+ results . llmobsSpans = payload . flatMap ( item => item . spans )
4242 } )
4343
4444 const apmRes = agent . assertMessageReceived ( ( { payload } ) => {
Original file line number Diff line number Diff line change @@ -92,8 +92,8 @@ describe('LLMObsSpanWriter', () => {
9292
9393 const payload = writer . makePayload ( events )
9494
95- expect ( payload [ '_dd.stage' ] ) . to . equal ( 'raw' )
96- expect ( payload . event_type ) . to . equal ( 'span' )
97- expect ( payload . spans ) . to . deep . equal ( events )
95+ expect ( payload [ 0 ] [ '_dd.stage' ] ) . to . equal ( 'raw' )
96+ expect ( payload [ 0 ] . event_type ) . to . equal ( 'span' )
97+ expect ( payload [ 0 ] . spans ) . to . deep . equal ( events )
9898 } )
9999} )
You can’t perform that action at this time.
0 commit comments