From 629d9b78fa5561608925c179e8a8b271d4fe6a55 Mon Sep 17 00:00:00 2001 From: mvhensbergen <46875782+mvhensbergen@users.noreply.github.com> Date: Thu, 12 Oct 2023 03:44:47 +0200 Subject: [PATCH] Make example work during pydantic transition (#11498) **Description:** Make the example extraction code on https://python.langchain.com/docs/use_cases/extraction work again by importing the langchain.pydantic_v1 lib instead of the v2. **Issue:** Solves issue https://github.com/langchain-ai/langchain/issues/11468 Co-authored-by: Martin van Hensbergen Co-authored-by: Bagatur --- docs/docs/use_cases/extraction.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/docs/use_cases/extraction.ipynb b/docs/docs/use_cases/extraction.ipynb index 356445c676..fa3695cab2 100644 --- a/docs/docs/use_cases/extraction.ipynb +++ b/docs/docs/use_cases/extraction.ipynb @@ -379,8 +379,8 @@ } ], "source": [ - "from typing import Optional, List\n", - "from pydantic import BaseModel, Field\n", + "from typing import Optional\n", + "from langchain.pydantic_v1 import BaseModel\n", "from langchain.chains import create_extraction_chain_pydantic\n", "\n", "# Pydantic data class\n",