From 995ef8a7fc02711886614098864a2df36a0a2f0f Mon Sep 17 00:00:00 2001
From: Bagatur <22008038+baskaryan@users.noreply.github.com>
Date: Thu, 17 Aug 2023 01:55:46 -0700
Subject: [PATCH] unpin pydantic (#9356)

---
 .github/workflows/langchain_ci.yml                       | 2 +-
 libs/experimental/tests/unit_tests/fake_llm.py           | 2 +-
 libs/langchain/langchain/chains/openai_functions/base.py | 4 ++--
 libs/langchain/poetry.lock                               | 2 +-
 libs/langchain/pyproject.toml                            | 2 +-
 5 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/langchain_ci.yml b/.github/workflows/langchain_ci.yml
index bb5b5fcd477..9dfbd8c3d6e 100644
--- a/.github/workflows/langchain_ci.yml
+++ b/.github/workflows/langchain_ci.yml
@@ -24,5 +24,5 @@ jobs:
       ./.github/workflows/_test.yml
     with:
       working-directory: libs/langchain
-      test_type: '["core", "extended"]'
+      test_type: '["core", "extended", "core-pydantic-2"]'
     secrets: inherit
\ No newline at end of file
diff --git a/libs/experimental/tests/unit_tests/fake_llm.py b/libs/experimental/tests/unit_tests/fake_llm.py
index 7da86861ea0..4e1a663056a 100644
--- a/libs/experimental/tests/unit_tests/fake_llm.py
+++ b/libs/experimental/tests/unit_tests/fake_llm.py
@@ -3,7 +3,7 @@ from typing import Any, Dict, List, Mapping, Optional, cast
 
 from langchain.callbacks.manager import CallbackManagerForLLMRun
 from langchain.llms.base import LLM
-from pydantic import validator
+from pydantic_v1 import validator
 
 
 class FakeLLM(LLM):
diff --git a/libs/langchain/langchain/chains/openai_functions/base.py b/libs/langchain/langchain/chains/openai_functions/base.py
index db322041435..f83dddd2544 100644
--- a/libs/langchain/langchain/chains/openai_functions/base.py
+++ b/libs/langchain/langchain/chains/openai_functions/base.py
@@ -230,7 +230,7 @@ def create_openai_fn_chain(
                 from langchain.chat_models import ChatOpenAI
                 from langchain.prompts import ChatPromptTemplate
 
-                from pydantic import BaseModel, Field
+                from pydantic_v1 import BaseModel, Field
 
 
                 class RecordPerson(BaseModel):
@@ -316,7 +316,7 @@ def create_structured_output_chain(
                 from langchain.chat_models import ChatOpenAI
                 from langchain.prompts import ChatPromptTemplate
 
-                from pydantic import BaseModel, Field
+                from pydantic_v1 import BaseModel, Field
 
                 class Dog(BaseModel):
                     \"\"\"Identifying information about a dog.\"\"\"
diff --git a/libs/langchain/poetry.lock b/libs/langchain/poetry.lock
index 3a1023763da..bb10e8ee272 100644
--- a/libs/langchain/poetry.lock
+++ b/libs/langchain/poetry.lock
@@ -10487,4 +10487,4 @@ text-helpers = ["chardet"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.8.1,<4.0"
-content-hash = "0c6f0b48849161928ab5df69f818779d6f5c58760cf967448a8785c938a23b57"
+content-hash = "594d1f6ea7a3e00f0ab6c74cab8b75245d112a84635af440df7ab1242d464140"
diff --git a/libs/langchain/pyproject.toml b/libs/langchain/pyproject.toml
index 6641184a5fc..8c775792fd9 100644
--- a/libs/langchain/pyproject.toml
+++ b/libs/langchain/pyproject.toml
@@ -12,7 +12,7 @@ langchain-server = "langchain.server:main"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"
-pydantic = "^1"
+pydantic = ">=1,<3"
 SQLAlchemy = ">=1.4,<3"
 requests = "^2"
 PyYAML = ">=5.3"