github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-11-05 10:45:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: Fix the stop sequence key name for Mistral in Bedrock ()

Browse Source 
Fixing the wrong stop sequence key name that causes an error on AWS
Bedrock.
You can check the MistralAI bedrock parameters
[here](https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters-mistral.html)
This change fixes this
[issue](https://github.com/langchain-ai/langchain/issues/20095)
This commit is contained in:
Matheus Henrique Raymundo
2024-04-21 21:06:06 -03:00
committed by GitHub
parent 1c7b3c75a7
commit bb69819267
1 changed files with 1 additions and 1 deletions

2
libs/community/langchain_community/llms/bedrock.py

@@ -339,7 +339,7 @@ class BedrockBase(BaseModel, ABC):
"amazon": "stopSequences", "amazon": "stopSequences",
"ai21": "stop_sequences", "ai21": "stop_sequences",
"cohere": "stop_sequences", "cohere": "stop_sequences",
"mistral": "stop_sequences", "mistral": "stop",
} }
guardrails: Optional[Mapping[str, Any]] = { guardrails: Optional[Mapping[str, Any]] = {