From 7abc6435f600b633f24ea0f6e5c9064cac600edc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Omar=20Yesid=20Mari=C3=B1o?= <omarinho@gmail.com>
Date: Tue, 14 May 2024 12:22:27 -0500
Subject: [PATCH] Remove "Truncate" parameter from Bedrock Cohere invoke model
request. (#13442)
---
.../llama_index/embeddings/bedrock/base.py | 4 +---
.../llms/llama-index-llms-bedrock/pyproject.toml | 2 +-
2 files changed, 2 insertions(+), 4 deletions(-)
diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-bedrock/llama_index/embeddings/bedrock/base.py b/llama-index-integrations/embeddings/llama-index-embeddings-bedrock/llama_index/embeddings/bedrock/base.py
index bd55071da2..571fe78b49 100644
--- a/llama-index-integrations/embeddings/llama-index-embeddings-bedrock/llama_index/embeddings/bedrock/base.py
+++ b/llama-index-integrations/embeddings/llama-index-embeddings-bedrock/llama_index/embeddings/bedrock/base.py
@@ -382,8 +382,7 @@ class BedrockEmbedding(BaseEmbedding):
Sample Payload of type dict of following format
{
'texts': ["This is a test document", "This is another document"],
- 'input_type': 'search_document',
- 'truncate': 'NONE'
+ 'input_type': 'search_document'
}
"""
@@ -402,7 +401,6 @@ class BedrockEmbedding(BaseEmbedding):
{
"texts": payload,
"input_type": input_types[input_type],
- "truncate": "NONE",
}
)
else:
diff --git a/llama-index-integrations/llms/llama-index-llms-bedrock/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-bedrock/pyproject.toml
index 6d980b0c48..2abc1cd7e5 100644
--- a/llama-index-integrations/llms/llama-index-llms-bedrock/pyproject.toml
+++ b/llama-index-integrations/llms/llama-index-llms-bedrock/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
license = "MIT"
name = "llama-index-llms-bedrock"
readme = "README.md"
-version = "0.1.7"
+version = "0.1.8"
[tool.poetry.dependencies]
python = ">=3.8.1,<4.0"
--
GitLab