File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -36,6 +36,19 @@ export interface PreparedProviderAttachment {
3636 contentType : 'image' | 'document' | 'audio' | 'video'
3737}
3838
39+ type ProviderMessageInput = {
40+ role : string
41+ content ?: string | null
42+ files ?: UserFile [ ]
43+ }
44+
45+ type ProviderFormattedMessage = {
46+ role : string
47+ content ?: string | null | Array < Record < string , unknown > >
48+ files ?: UserFile [ ]
49+ [ key : string ] : unknown
50+ }
51+
3952const AGENT_ATTACHMENT_MAX_BYTES = 10 * 1024 * 1024
4053const PDF_MIME_TYPE = 'application/pdf'
4154
@@ -532,12 +545,12 @@ export function buildBedrockMessageContent(
532545}
533546
534547export function formatMessagesForProvider (
535- messages : Array < { role : string ; content ?: string | null ; files ?: UserFile [ ] } > ,
548+ messages : ProviderMessageInput [ ] ,
536549 providerId : ProviderId | string
537- ) {
550+ ) : ProviderFormattedMessage [ ] {
538551 return messages . map ( ( message ) => {
539552 if ( ! message . files ?. length || ( message . role !== 'user' && message . role !== 'assistant' ) ) {
540- return message
553+ return message as ProviderFormattedMessage
541554 }
542555
543556 const provider = getAttachmentProvider ( providerId )
You can’t perform that action at this time.
0 commit comments