From 3c9ea84eef4cc7b04a21770b1dac7d498787e14f Mon Sep 17 00:00:00 2001 From: Piyush Jain Date: Wed, 2 Oct 2024 16:34:38 -0700 Subject: [PATCH] Added snapshots for unit tests. --- .../__snapshots__/test_standard.ambr | 80 +++++++++++++++++++ 1 file changed, 80 insertions(+) create mode 100644 libs/aws/tests/unit_tests/__snapshots__/test_standard.ambr diff --git a/libs/aws/tests/unit_tests/__snapshots__/test_standard.ambr b/libs/aws/tests/unit_tests/__snapshots__/test_standard.ambr new file mode 100644 index 00000000..6f98f04b --- /dev/null +++ b/libs/aws/tests/unit_tests/__snapshots__/test_standard.ambr @@ -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', + }) +# ---