diff --git a/RAG (Bedrock and Ragas).ipynb b/RAG (Bedrock and Ragas).ipynb index 68c3ecc..5657f10 100644 --- 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 index 73398e6..682b07f 100644 --- 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\"" ] }, {