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

pull/17954/head^2
Erick Friis 7 months ago committed by GitHub
parent 9cf6661dc5
commit e237dcec91
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -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:

Loading…
Cancel
Save