couchbase[patch]: rm pydantic usage (#26068)

This commit is contained in:
Bagatur
2024-09-04 16:29:14 -07:00
committed by GitHub
parent c5d4dfefc0
commit 79c46319dd

View File

@@ -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())