@@ -132,20 +132,32 @@ function extractRequestAttributes(
132
132
function addPrivateRequestAttributes ( span : Span , params : Record < string , unknown > ) : void {
133
133
if ( 'contents' in params ) {
134
134
const contents = params . contents ;
135
- const truncatedContents = truncateGenAiMessages ( contents as unknown [ ] ) ;
136
- span . setAttributes ( { [ GEN_AI_REQUEST_MESSAGES_ATTRIBUTE ] : JSON . stringify ( truncatedContents ) } ) ;
135
+ if ( Array . isArray ( contents ) ) {
136
+ const truncatedContents = truncateGenAiMessages ( contents ) ;
137
+ span . setAttributes ( { [ GEN_AI_REQUEST_MESSAGES_ATTRIBUTE ] : JSON . stringify ( truncatedContents ) } ) ;
138
+ } else {
139
+ span . setAttributes ( { [ GEN_AI_REQUEST_MESSAGES_ATTRIBUTE ] : JSON . stringify ( contents ) } ) ;
140
+ }
137
141
}
138
142
139
143
if ( 'message' in params ) {
140
144
const message = params . message ;
141
- const truncatedMessage = truncateGenAiMessages ( message as unknown [ ] ) ;
142
- span . setAttributes ( { [ GEN_AI_REQUEST_MESSAGES_ATTRIBUTE ] : JSON . stringify ( truncatedMessage ) } ) ;
145
+ if ( Array . isArray ( message ) ) {
146
+ const truncatedMessage = truncateGenAiMessages ( message ) ;
147
+ span . setAttributes ( { [ GEN_AI_REQUEST_MESSAGES_ATTRIBUTE ] : JSON . stringify ( truncatedMessage ) } ) ;
148
+ } else {
149
+ span . setAttributes ( { [ GEN_AI_REQUEST_MESSAGES_ATTRIBUTE ] : JSON . stringify ( message ) } ) ;
150
+ }
143
151
}
144
152
145
153
if ( 'history' in params ) {
146
154
const history = params . history ;
147
- const truncatedHistory = truncateGenAiMessages ( history as unknown [ ] ) ;
148
- span . setAttributes ( { [ GEN_AI_REQUEST_MESSAGES_ATTRIBUTE ] : JSON . stringify ( truncatedHistory ) } ) ;
155
+ if ( Array . isArray ( history ) ) {
156
+ const truncatedHistory = truncateGenAiMessages ( history ) ;
157
+ span . setAttributes ( { [ GEN_AI_REQUEST_MESSAGES_ATTRIBUTE ] : JSON . stringify ( truncatedHistory ) } ) ;
158
+ } else {
159
+ span . setAttributes ( { [ GEN_AI_REQUEST_MESSAGES_ATTRIBUTE ] : JSON . stringify ( history ) } ) ;
160
+ }
149
161
}
150
162
}
151
163
0 commit comments