diff --git a/build_artifacts/v1/v1.9/v1.9.0/cpu.additional_packages_env.in b/build_artifacts/v1/v1.9/v1.9.0/cpu.additional_packages_env.in new file mode 100644 index 00000000..37039a80 --- /dev/null +++ b/build_artifacts/v1/v1.9/v1.9.0/cpu.additional_packages_env.in @@ -0,0 +1 @@ +conda-forge::langchain-aws[version='>=0.1.1,<0.2.0'] diff --git a/build_artifacts/v1/v1.9/v1.9.0/gpu.additional_packages_env.in b/build_artifacts/v1/v1.9/v1.9.0/gpu.additional_packages_env.in new file mode 100644 index 00000000..37039a80 --- /dev/null +++ b/build_artifacts/v1/v1.9/v1.9.0/gpu.additional_packages_env.in @@ -0,0 +1 @@ +conda-forge::langchain-aws[version='>=0.1.1,<0.2.0'] diff --git a/test/test_artifacts/v1/langchain-aws.test.Dockerfile b/test/test_artifacts/v1/langchain-aws.test.Dockerfile new file mode 100644 index 00000000..73353bb0 --- /dev/null +++ b/test/test_artifacts/v1/langchain-aws.test.Dockerfile @@ -0,0 +1,13 @@ +ARG SAGEMAKER_DISTRIBUTION_IMAGE +FROM $SAGEMAKER_DISTRIBUTION_IMAGE + +ARG MAMBA_DOCKERFILE_ACTIVATE=1 + +CMD ["python", "-c", "import langchain_aws"] +CMD ["python", "-c", "from langchain_aws import BedrockLLM"] +CMD ["python", "-c", "from langchain_aws import ChatBedrock"] +CMD ["python", "-c", "from langchain_aws import SagemakerEndpoint"] +CMD ["python", "-c", "from langchain_aws import AmazonKendraRetriever"] +CMD ["python", "-c", "from langchain_aws import AmazonKnowledgeBasesRetriever"] +CMD ["python", "-c", "from langchain_aws import NeptuneAnalyticsGraph"] +CMD ["python", "-c", "from langchain_aws import NeptuneGraph"] diff --git a/test/test_dockerfile_based_harness.py b/test/test_dockerfile_based_harness.py index 78008f3f..56253b2f 100644 --- a/test/test_dockerfile_based_harness.py +++ b/test/test_dockerfile_based_harness.py @@ -43,6 +43,7 @@ ("amazon-sagemaker-sql-magic.test.Dockerfile", ["amazon-sagemaker-sql-magic"]), ("amazon_sagemaker_sql_editor.test.Dockerfile", ["amazon_sagemaker_sql_editor"]), ("serve.test.Dockerfile", ["serve-langchain"]), + ("langchain-aws.test.Dockerfile", ["langchain-aws"]), ], ) def test_dockerfiles_for_cpu( @@ -81,6 +82,7 @@ def test_dockerfiles_for_cpu( ("amazon-sagemaker-sql-magic.test.Dockerfile", ["amazon-sagemaker-sql-magic"]), ("amazon_sagemaker_sql_editor.test.Dockerfile", ["amazon_sagemaker_sql_editor"]), ("serve.test.Dockerfile", ["serve-langchain"]), + ("langchain-aws.test.Dockerfile", ["langchain-aws"]), ], ) def test_dockerfiles_for_gpu(