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