File tree Expand file tree Collapse file tree 1 file changed +36
-2
lines changed Expand file tree Collapse file tree 1 file changed +36
-2
lines changed Original file line number Diff line number Diff line change 1
- import { BEDROCK , documentMimeTypes , imagesMimeTypes } from '../../globals' ;
1
+ import {
2
+ BEDROCK ,
3
+ documentMimeTypes ,
4
+ fileExtensionMimeTypeMap ,
5
+ imagesMimeTypes ,
6
+ } from '../../globals' ;
2
7
import {
3
8
Message ,
4
9
Params ,
@@ -176,6 +181,32 @@ const getMessageContent = (message: Message) => {
176
181
} ,
177
182
} ) ;
178
183
}
184
+ } else if ( item . type === 'file' ) {
185
+ const mimeType = item . file ?. mime_type || fileExtensionMimeTypeMap . pdf ;
186
+ const fileFormat = mimeType . split ( '/' ) [ 1 ] ;
187
+ if ( item . file ?. file_url ) {
188
+ out . push ( {
189
+ document : {
190
+ format : fileFormat ,
191
+ name : crypto . randomUUID ( ) ,
192
+ source : {
193
+ s3Location : {
194
+ uri : item . file . file_url ,
195
+ } ,
196
+ } ,
197
+ } ,
198
+ } ) ;
199
+ } else if ( item . file ?. file_data ) {
200
+ out . push ( {
201
+ document : {
202
+ format : fileFormat ,
203
+ name : crypto . randomUUID ( ) ,
204
+ source : {
205
+ bytes : item . file . file_data ,
206
+ } ,
207
+ } ,
208
+ } ) ;
209
+ }
179
210
}
180
211
181
212
if ( item . cache_control ) {
@@ -377,7 +408,10 @@ type BedrockContentItem = {
377
408
format : string ;
378
409
name : string ;
379
410
source : {
380
- bytes : string ;
411
+ bytes ?: string ;
412
+ s3Location ?: {
413
+ uri : string ;
414
+ } ;
381
415
} ;
382
416
} ;
383
417
cachePoint ?: {
You can’t perform that action at this time.
0 commit comments