Skip to content

Commit

Permalink
Ran into an issue where the `test_chat_bedrock_streaming_generation_i…
Browse files Browse the repository at this point in the history
…nfo` test would occasionally fail (returned "Hello! How are you doing?" one time)
  • Loading branch information
NAPTlME committed May 17, 2024
1 parent 5eace06 commit bfa0871
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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"]
Expand Down

0 comments on commit bfa0871

Please sign in to comment.