Skip to content
Snippets Groups Projects
Unverified Commit 7abc6435 authored by Omar Yesid Mariño's avatar Omar Yesid Mariño Committed by GitHub
Browse files

Remove "Truncate" parameter from Bedrock Cohere invoke model request. (#13442)

parent 7e9d1c68
No related branches found
No related tags found
No related merge requests found
...@@ -382,8 +382,7 @@ class BedrockEmbedding(BaseEmbedding): ...@@ -382,8 +382,7 @@ class BedrockEmbedding(BaseEmbedding):
Sample Payload of type dict of following format Sample Payload of type dict of following format
{ {
'texts': ["This is a test document", "This is another document"], 'texts': ["This is a test document", "This is another document"],
'input_type': 'search_document', 'input_type': 'search_document'
'truncate': 'NONE'
} }
""" """
...@@ -402,7 +401,6 @@ class BedrockEmbedding(BaseEmbedding): ...@@ -402,7 +401,6 @@ class BedrockEmbedding(BaseEmbedding):
{ {
"texts": payload, "texts": payload,
"input_type": input_types[input_type], "input_type": input_types[input_type],
"truncate": "NONE",
} }
) )
else: else:
......
...@@ -27,7 +27,7 @@ exclude = ["**/BUILD"] ...@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
license = "MIT" license = "MIT"
name = "llama-index-llms-bedrock" name = "llama-index-llms-bedrock"
readme = "README.md" readme = "README.md"
version = "0.1.7" version = "0.1.8"
[tool.poetry.dependencies] [tool.poetry.dependencies]
python = ">=3.8.1,<4.0" python = ">=3.8.1,<4.0"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment