Fix tests (#308)

pinecone-io · Feb 26, 2024 · 050869d · 050869d
1 parent 8dfa859
commit 050869d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/canopy/knowledge_base/knowledge_base.py b/src/canopy/knowledge_base/knowledge_base.py
@@ -666,7 +666,8 @@ def from_config(cls,
     @lru_cache(maxsize=1)
     def _is_serverless_env(self):
         description = self._pinecone_client.describe_index(self.index_name)
-        return "serverless" in description["spec"]
+        return ("serverless" in description["spec"] or
+                description["spec"].get("pod", {}).get("environment") == "gcp-starter")
 
     async def aquery(self,
                      queries: List[Query],

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -4,6 +4,7 @@
 TEST_CREATE_INDEX_PARAMS = [
     {"spec": {"serverless": {"cloud": "aws", "region": "us-west-2"}}},
     {"spec": {"pod": {"environment": "eu-west1-gcp", "pod_type": "p1.x1"}}},
+    {"spec": {"pod": {"environment": "gcp-starter", "pod_type": "p1.x1"}}},
 ]