diff --git a/src/adapters/bedrock/chat.ts b/src/adapters/bedrock/chat.ts index 65235cd8..c3e99ef5 100644 --- a/src/adapters/bedrock/chat.ts +++ b/src/adapters/bedrock/chat.ts @@ -217,8 +217,7 @@ export class BedrockChatLLM extends ChatLLM { contentType: "application/json", accept: "application/json", body: JSON.stringify({ - texts: input.map((msgs) => msgs.map((msg) => msg.text)), - input_type: "search_document", + texts: input.flat().map((msg) => msg.text), ...options?.body, }), }); diff --git a/tests/e2e/adapters/bedrock/chat.test.ts b/tests/e2e/adapters/bedrock/chat.test.ts index e7696a6f..88fe53b7 100644 --- a/tests/e2e/adapters/bedrock/chat.test.ts +++ b/tests/e2e/adapters/bedrock/chat.test.ts @@ -1,5 +1,5 @@ /** - * Copyright 2024 IBM Corp. + * Copyright 2025 IBM Corp. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License.