github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-04 20:28:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

pinecone[patch]: integration test debug (#17960)

Browse Source
This commit is contained in:
Erick Friis 2024-02-22 09:11:21 -08:00 committed by GitHub
parent 9cf6661dc5
commit e237dcec91
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -32,11 +32,13 @@ class TestPinecone:
if i["name"] == INDEX_NAME: if i["name"] == INDEX_NAME:
client.delete_index(INDEX_NAME) client.delete_index(INDEX_NAME)
break break
if len(index_list) > 0:
time.sleep(DEFAULT_SLEEP) # prevent race with creation
client.create_index( client.create_index(
name=INDEX_NAME, name=INDEX_NAME,
dimension=DIMENSION, dimension=DIMENSION,
metric="cosine", metric="cosine",
spec=PodSpec(environment=os.environ["PINECONE_ENVIRONMENT"]), spec=PodSpec(environment="gcp-starter"),
) )
cls.index = client.Index(INDEX_NAME) cls.index = client.Index(INDEX_NAME)
@ -57,12 +59,12 @@ class TestPinecone:
def setup(self) -> None: def setup(self) -> None:
# delete all the vectors in the index # delete all the vectors in the index
print("called") # noqa: T201 print("called") # noqa: T201
self.index.delete(delete_all=True, namespace=NAMESPACE_NAME) try:
# index_stats = self.index.describe_index_stats() self.index.delete(delete_all=True, namespace=NAMESPACE_NAME)
# for _namespace_name in index_stats["namespaces"].keys(): time.sleep(DEFAULT_SLEEP) # prevent race condition with previous step
# self.index.delete(delete_all=True, namespace=_namespace_name) except Exception:
time.sleep(DEFAULT_SLEEP) # prevent race condition with previous step # if namespace not found
# index_stats = self.index.describe_index_stats pass
@pytest.fixture @pytest.fixture
def embedding_openai(self) -> OpenAIEmbeddings: def embedding_openai(self) -> OpenAIEmbeddings: