github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-14 17:07:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

pinecone[patch]: integration test new namespace (#17121)

Browse Source
This commit is contained in:
Erick Friis 2024-02-06 11:56:00 -08:00 committed by GitHub
parent 3ccffa5dcc
commit 15bd1154a7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/partners/pinecone/tests/integration_tests/test_vectorstores.py
View File

@ -100,15 +100,17 @@ class TestPinecone:
texts = [needs] + texts texts = [needs] + texts
metadatas = [{"page": i} for i in range(len(texts))] metadatas = [{"page": i} for i in range(len(texts))]
namespace = f"{NAMESPACE_NAME}-md"
docsearch = Pinecone.from_texts( docsearch = Pinecone.from_texts(
texts, texts,
embedding_openai, embedding_openai,
index_name=INDEX_NAME, index_name=INDEX_NAME,
metadatas=metadatas, metadatas=metadatas,
namespace=NAMESPACE_NAME, namespace=namespace,
) )
time.sleep(DEFAULT_SLEEP) # prevent race condition time.sleep(DEFAULT_SLEEP) # prevent race condition
output = docsearch.similarity_search(needs, k=1, namespace=NAMESPACE_NAME) output = docsearch.similarity_search(needs, k=1, namespace=namespace)
# TODO: why metadata={"page": 0.0}) instead of {"page": 0}? # TODO: why metadata={"page": 0.0}) instead of {"page": 0}?
assert output == [Document(page_content=needs, metadata={"page": 0.0})] assert output == [Document(page_content=needs, metadata={"page": 0.0})]