feat(adapters): update embedding support for Bedrock

Signed-off-by: Tomas Dvorak <[email protected]>
i-am-bee · Jan 2, 2025 · 9b84c16 · 9b84c16
1 parent ceedb7d
commit 9b84c16
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/adapters/bedrock/chat.ts b/src/adapters/bedrock/chat.ts
@@ -217,8 +217,7 @@ export class BedrockChatLLM extends ChatLLM<ChatBedrockOutput> {
       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
@@ -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.