From bfa08716094211882e6f8156fb6347db4e91db56 Mon Sep 17 00:00:00 2001 From: NAPTlME Date: Fri, 17 May 2024 16:37:41 -0500 Subject: [PATCH] Ran into an issue where the `test_chat_bedrock_streaming_generation_info` test would occasionally fail (returned "Hello! How are you doing?" one time) --- libs/aws/tests/integration_tests/chat_models/test_bedrock.py | 1 + 1 file changed, 1 insertion(+) diff --git a/libs/aws/tests/integration_tests/chat_models/test_bedrock.py b/libs/aws/tests/integration_tests/chat_models/test_bedrock.py index 0ec4d633..6d1fb57a 100644 --- a/libs/aws/tests/integration_tests/chat_models/test_bedrock.py +++ b/libs/aws/tests/integration_tests/chat_models/test_bedrock.py @@ -111,6 +111,7 @@ def on_llm_end( chat = ChatBedrock( # type: ignore[call-arg] model_id="anthropic.claude-v2", callbacks=[callback], + model_kwargs={"temperature": 0}, ) list(chat.stream("hi")) generation = callback.saved_things["generation"]