From 58b7a3ba16d12051f0c399e5968d26acaee0b230 Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Wed, 4 Oct 2023 23:31:51 -0400 Subject: [PATCH] Rm bedrock anthropic error (#11403) --- libs/langchain/langchain/llms/bedrock.py | 5 +++-- libs/langchain/tests/unit_tests/llms/test_bedrock.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/libs/langchain/langchain/llms/bedrock.py b/libs/langchain/langchain/llms/bedrock.py index 6a0f355b34..7971f6e9c5 100644 --- a/libs/langchain/langchain/llms/bedrock.py +++ b/libs/langchain/langchain/llms/bedrock.py @@ -1,4 +1,5 @@ import json +import warnings from abc import ABC from typing import Any, Dict, Iterator, List, Mapping, Optional @@ -42,12 +43,12 @@ def _human_assistant_format(input_text: str) -> str: if count % 2 == 0: count += 1 else: - raise ValueError(ALTERNATION_ERROR + f" Received {input_text}") + warnings.warn(ALTERNATION_ERROR + f" Received {input_text}") if input_text[i : i + len(ASSISTANT_PROMPT)] == ASSISTANT_PROMPT: if count % 2 == 1: count += 1 else: - raise ValueError(ALTERNATION_ERROR + f" Received {input_text}") + warnings.warn(ALTERNATION_ERROR + f" Received {input_text}") if count % 2 == 1: # Only saw Human, no Assistant input_text = input_text + ASSISTANT_PROMPT # SILENT CORRECTION diff --git a/libs/langchain/tests/unit_tests/llms/test_bedrock.py b/libs/langchain/tests/unit_tests/llms/test_bedrock.py index 985011c277..fe70674719 100644 --- a/libs/langchain/tests/unit_tests/llms/test_bedrock.py +++ b/libs/langchain/tests/unit_tests/llms/test_bedrock.py @@ -245,7 +245,7 @@ Assistant:""", def test__human_assistant_format() -> None: for input_text, expected_output in TEST_CASES.items(): if expected_output == ALTERNATION_ERROR: - with pytest.raises(ValueError): + with pytest.warns(UserWarning, match=ALTERNATION_ERROR): _human_assistant_format(input_text) else: output = _human_assistant_format(input_text)