fix(nbs): Pydantic version issue

See aws/sagemaker-distribution#436
athewsey · Jul 9, 2024 · 8ec2c3c · 8ec2c3c
1 parent 6883bae
commit 8ec2c3c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/RAG (Bedrock and Ragas).ipynb b/RAG (Bedrock and Ragas).ipynb
@@ -7,7 +7,7 @@
    "source": [
     "# Evaluating Knowledge Bases for Amazon Bedrock with Ragas\n",
     "\n",
-    "> *This notebook has been tested in the Python 3 kernel of SageMaker Studio JupyterLab (Distribution v1.7)*\n",
+    "> *This notebook has been tested in the Python 3 kernel of SageMaker Studio JupyterLab (Distribution v1.9)*\n",
     "\n",
     "In this notebook, we'll explore how open-source library [Ragas](https://docs.ragas.io/en/latest/) can be applied to evaluate the quality of [Retrieval-Augmented Generation (RAG)](https://aws.amazon.com/what-is/retrieval-augmented-generation/) flows managed by [Amazon Bedrock Knowledge Bases](https://docs.aws.amazon.com/bedrock/latest/userguide/knowledge-base.html)."
    ]
@@ -51,7 +51,7 @@
    "outputs": [],
    "source": [
     "# We also force Pydantic version to avoid: https://github.com/explodinggradients/ragas/issues/867\n",
-    "%pip install \"langchain-aws>=0.1,<0.2\" \"ragas==0.1.8\" \"pydantic>=2,<3\""
+    "%pip install \"langchain-aws>=0.1,<0.2\" \"ragas==0.1.8\" \"pydantic>=2.8,<3\""
    ]
   },
   {

diff --git a/conversational-tests/Conversational Tests.ipynb b/conversational-tests/Conversational Tests.ipynb
@@ -7,7 +7,7 @@
    "source": [
     "# Conversational tests with `agent-evaluation`\n",
     "\n",
-    "> *This notebook has been tested in the Python 3 kernel of SageMaker Studio JupyterLab (Distribution v1.7)*\n",
+    "> *This notebook has been tested in the Python 3 kernel of SageMaker Studio JupyterLab (Distribution v1.9)*\n",
     "\n",
     "In this notebook, we'll show how you can use AWS Labs' open-source [`agent-evaluation` framework](https://awslabs.github.io/agent-evaluation/) to validate integrated agent systems perform as expected over multi-turn conversations - with expectations defined in natural language."
    ]
@@ -31,7 +31,8 @@
    },
    "outputs": [],
    "source": [
-    "%pip install agent-evaluation"
+    "# Force Pydantic for https://github.com/aws/sagemaker-distribution/issues/436\n",
+    "%pip install agent-evaluation \"pydantic>=2.8,<2.9\""
    ]
   },
   {