github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-27 14:26:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Bagatur
2024-09-04 16:29:14 -07:00
committed by GitHub
parent c5d4dfefc0
commit 79c46319dd
1 changed files with 0 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
libs/partners/couchbase/tests/utils.py
View File

@@ -5,7 +5,6 @@ from typing import Any, Dict, List, Mapping, Optional, cast
from langchain_core.callbacks import CallbackManagerForLLMRun from langchain_core.callbacks import CallbackManagerForLLMRun
from langchain_core.embeddings import Embeddings from langchain_core.embeddings import Embeddings
from langchain_core.language_models.llms import LLM from langchain_core.language_models.llms import LLM
from langchain_core.pydantic_v1 import validator
class FakeEmbeddings(Embeddings): class FakeEmbeddings(Embeddings):
@@ -63,16 +62,6 @@ class FakeLLM(LLM):
sequential_responses: Optional[bool] = False sequential_responses: Optional[bool] = False
response_index: int = 0 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: def get_num_tokens(self, text: str) -> int:
"""Return number of tokens.""" """Return number of tokens."""
return len(text.split()) return len(text.split())