Added langchain-aws package. (#382)

Co-authored-by: Andrii Ieroshenko <[email protected]>
aws · May 31, 2024 · 6825fbe · 6825fbe
1 parent 1d40c55
commit 6825fbe
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 0 deletions.
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
@@ -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
@@ -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
@@ -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
@@ -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(
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		conda-forge::langchain-aws[version='>=0.1.1,<0.2.0']