From 79c46319ddd7959ca868772982aad25bcdbef1c2 Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Wed, 4 Sep 2024 16:29:14 -0700 Subject: [PATCH] couchbase[patch]: rm pydantic usage (#26068) --- libs/partners/couchbase/tests/utils.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/libs/partners/couchbase/tests/utils.py b/libs/partners/couchbase/tests/utils.py index 29fde66eaeb..d4c30a59b12 100644 --- a/libs/partners/couchbase/tests/utils.py +++ b/libs/partners/couchbase/tests/utils.py @@ -5,7 +5,6 @@ from typing import Any, Dict, List, Mapping, Optional, cast from langchain_core.callbacks import CallbackManagerForLLMRun from langchain_core.embeddings import Embeddings from langchain_core.language_models.llms import LLM -from langchain_core.pydantic_v1 import validator class FakeEmbeddings(Embeddings): @@ -63,16 +62,6 @@ 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: - raise ValueError( - "queries is required when sequential_response is set to True" - ) - return queries - def get_num_tokens(self, text: str) -> int: """Return number of tokens.""" return len(text.split())