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

pinecone[patch]: Upgrade @root_validators to be consistent with pydantic 2 (#25453)

Browse Source 
Upgrade root validators for pydantic 2 migration
This commit is contained in:
Eugene Yurtsev
2024-08-15 15:45:14 -04:00
committed by GitHub
parent b297af5482
commit 34da8be60b
2 changed files with 29 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
libs/partners/pinecone/tests/unit_tests/test_embeddings.py
View File

@@ -7,10 +7,22 @@ MODEL_NAME = "multilingual-e5-large"
def test_default_config() -> None:
e = PineconeEmbeddings(pinecone_api_key=API_KEY, model=MODEL_NAME)
e = PineconeEmbeddings(
pinecone_api_key=API_KEY, # type: ignore[call-arg]
model=MODEL_NAME,
)
assert e.batch_size == 96
def test_default_config_with_api_key() -> None:
e = PineconeEmbeddings(api_key=API_KEY, model=MODEL_NAME)
assert e.batch_size == 96
def test_custom_config() -> None:
e = PineconeEmbeddings(pinecone_api_key=API_KEY, model=MODEL_NAME, batch_size=128)
e = PineconeEmbeddings(
pinecone_api_key=API_KEY, # type: ignore[call-arg]
model=MODEL_NAME,
batch_size=128,
)
assert e.batch_size == 128