fmt

langchain-ai · Jun 20, 2024 · b557b7f · b557b7f
1 parent 55710e4
commit b557b7f
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/libs/aws/tests/unit_tests/test_standard.py b/libs/aws/tests/unit_tests/test_standard.py
@@ -28,3 +28,31 @@ def standard_chat_model_params(self) -> dict:
     @pytest.mark.xfail(reason="Not implemented.")
     def test_standard_params(self, model: BaseChatModel) -> None:
         super().test_standard_params(model)
+
+
+class TestBedrockAsConverseStandard(ChatModelUnitTests):
+    @property
+    def chat_model_class(self) -> Type[BaseChatModel]:
+        return ChatBedrock
+
+    @property
+    def chat_model_params(self) -> dict:
+        return {
+            "model_id": "anthropic.claude-3-sonnet-20240229-v1:0",
+            "region_name": "us-east-1",
+            "beta_use_converse_api": True,
+        }
+
+    @property
+    def standard_chat_model_params(self) -> dict:
+        return {
+            "model_kwargs": {
+                "temperature": 0,
+                "max_tokens": 100,
+                "stop": [],
+            }
+        }
+
+    @pytest.mark.xfail(reason="Not implemented.")
+    def test_standard_params(self, model: BaseChatModel) -> None:
+        super().test_standard_params(model)