Skip to content

Commit

Permalink
Added snapshots for unit tests.
Browse files Browse the repository at this point in the history
  • Loading branch information
3coins committed Oct 2, 2024
1 parent 31881b2 commit 3c9ea84
Showing 1 changed file with 80 additions and 0 deletions.
80 changes: 80 additions & 0 deletions libs/aws/tests/unit_tests/__snapshots__/test_standard.ambr
Original file line number Diff line number Diff line change
@@ -0,0 +1,80 @@
# serializer version: 1
# name: TestBedrockAsConverseStandard.test_serdes[serialized]
dict({
'id': list([
'langchain',
'chat_models',
'bedrock',
'ChatBedrock',
]),
'kwargs': dict({
'beta_use_converse_api': True,
'guardrails': dict({
'guardrailIdentifier': None,
'guardrailVersion': None,
'trace': None,
}),
'model_id': 'anthropic.claude-3-sonnet-20240229-v1:0',
'model_kwargs': dict({
'max_tokens': 100,
'stop': list([
]),
'temperature': 0,
}),
'provider_stop_reason_key_map': dict({
'ai21': 'finishReason',
'amazon': 'completionReason',
'anthropic': 'stop_reason',
'cohere': 'finish_reason',
'mistral': 'stop_reason',
}),
'provider_stop_sequence_key_name_map': dict({
'ai21': 'stop_sequences',
'amazon': 'stopSequences',
'anthropic': 'stop_sequences',
'cohere': 'stop_sequences',
'mistral': 'stop_sequences',
}),
'region_name': 'us-east-1',
}),
'lc': 1,
'name': 'ChatBedrock',
'type': 'constructor',
})
# ---
# name: TestBedrockStandard.test_serdes[serialized]
dict({
'id': list([
'langchain',
'chat_models',
'bedrock',
'ChatBedrock',
]),
'kwargs': dict({
'guardrails': dict({
'guardrailIdentifier': None,
'guardrailVersion': None,
'trace': None,
}),
'model_id': 'anthropic.claude-3-sonnet-20240229-v1:0',
'provider_stop_reason_key_map': dict({
'ai21': 'finishReason',
'amazon': 'completionReason',
'anthropic': 'stop_reason',
'cohere': 'finish_reason',
'mistral': 'stop_reason',
}),
'provider_stop_sequence_key_name_map': dict({
'ai21': 'stop_sequences',
'amazon': 'stopSequences',
'anthropic': 'stop_sequences',
'cohere': 'stop_sequences',
'mistral': 'stop_sequences',
}),
'region_name': 'us-east-1',
}),
'lc': 1,
'name': 'ChatBedrock',
'type': 'constructor',
})
# ---

0 comments on commit 3c9ea84

Please sign in to comment.