multiple: pydantic 2 compatibility, v0.3 (#26443)

Signed-off-by: ChengZi <chen.zhang@zilliz.com>
Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com>
Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
Co-authored-by: Dan O'Donovan <dan.odonovan@gmail.com>
Co-authored-by: Tom Daniel Grande <tomdgrande@gmail.com>
Co-authored-by: Grande <Tom.Daniel.Grande@statsbygg.no>
Co-authored-by: Bagatur <baskaryan@gmail.com>
Co-authored-by: ccurme <chester.curme@gmail.com>
Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>
Co-authored-by: Tomaz Bratanic <bratanic.tomaz@gmail.com>
Co-authored-by: ZhangShenao <15201440436@163.com>
Co-authored-by: Friso H. Kingma <fhkingma@gmail.com>
Co-authored-by: ChengZi <chen.zhang@zilliz.com>
Co-authored-by: Nuno Campos <nuno@langchain.dev>
Co-authored-by: Morgante Pell <morgantep@google.com>

libs/experimental/tests/unit_tests/fake_llm.py

@@ -4,8 +4,7 @@ from typing import Any, Dict, List, Mapping, Optional, cast
 
 from langchain_core.callbacks.manager import CallbackManagerForLLMRun
 from langchain_core.language_models import LLM
-
-from langchain_experimental.pydantic_v1 import validator
+from pydantic import model_validator
 
 
 class FakeLLM(LLM):
@@ -15,15 +14,14 @@ class FakeLLM(LLM):
     sequential_responses: Optional[bool] = False
     response_index: int = 0
 
-    @validator("queries", always=True)
-    def check_queries_required(
-        cls, queries: Optional[Mapping], values: Mapping[str, Any]
-    ) -> Optional[Mapping]:
-        if values.get("sequential_response") and not queries:
+    @model_validator(mode="before")
+    @classmethod
+    def check_queries_required(cls, values: dict) -> dict:
+        if values.get("sequential_response") and not values.get("queries"):
             raise ValueError(
                 "queries is required when sequential_response is set to True"
             )
-        return queries
+        return values
 
     def get_num_tokens(self, text: str) -> int:
         """Return number of tokens."""

libs/experimental/tests/unit_tests/test_ollama_functions.py

@@ -3,7 +3,7 @@ from typing import Any
 from unittest.mock import patch
 
 from langchain_core.prompts import ChatPromptTemplate
-from langchain_core.pydantic_v1 import BaseModel
+from pydantic import BaseModel
 
 from langchain_experimental.llms.ollama_functions import OllamaFunctions
 

libs/experimental/tests/unit_tests/test_sql.py

@@ -1,6 +1,6 @@
 from langchain.memory import ConversationBufferMemory
 from langchain.output_parsers.list import CommaSeparatedListOutputParser
-from langchain.sql_database import SQLDatabase
+from langchain_community.utilities import SQLDatabase
 from langchain_core.prompts import PromptTemplate
 
 from langchain_experimental.sql.base import SQLDatabaseChain, SQLDatabaseSequentialChain