Added integration tests.

libs/aws/pyproject.toml

@@ -84,6 +84,9 @@ addopts = "--snapshot-warn-unused --strict-markers --strict-config --durations=5
 # Registering custom markers.
 # https://docs.pytest.org/en/7.1.x/example/markers.html#registering-markers
 markers = [
+  "requires: mark tests as requiring a specific library",
+  "asyncio: mark tests as requiring asyncio",
   "compile: mark placeholder test used to compile integration tests without running them",
+  "scheduled: mark tests to run in scheduled testing",
 ]
 asyncio_mode = "auto"

libs/aws/tests/integration_tests/llms/test_sagemaker_endpoint.py

@@ -0,0 +1,21 @@
+from typing import Dict
+
+from langchain_aws.llms import SagemakerEndpoint
+from langchain_aws.llms.sagemaker_endpoint import LLMContentHandler
+
+
+class ContentHandler(LLMContentHandler):
+    def transform_input(self, prompt: str, model_kwargs: Dict) -> bytes:
+        return b""
+
+    def transform_output(self, output: bytes) -> str:
+        return ""
+
+
+def test_sagemaker_endpoint_name_param() -> None:
+    llm = SagemakerEndpoint(
+        endpoint_name="foo",
+        content_handler=ContentHandler(),
+        region_name="us-west-2",
+    )
+    assert llm.endpoint_name == "foo"

libs/aws/tests/integration_tests/test_compile.py

@@ -0,0 +1,7 @@
+import pytest
+
+
+@pytest.mark.compile
+def test_placeholder() -> None:
+    """Used for compiling integration tests without running any real tests."""
+    pass