Fix unit tests

Author: jhamon

pinecone/__init__.py

@@ -21,7 +21,11 @@
 }
 
 _db_data_lazy_imports = {
-    "Vector": ("pinecone.db_data.models", "Vector"),
+    "Vector": ("pinecone.db_data.dataclasses", "Vector"),
+    "SparseValues": ("pinecone.db_data.dataclasses", "SparseValues"),
+    "SearchQuery": ("pinecone.db_data.dataclasses", "SearchQuery"),
+    "SearchQueryVector": ("pinecone.db_data.dataclasses", "SearchQueryVector"),
+    "SearchRerank": ("pinecone.db_data.dataclasses", "SearchRerank"),
     "FetchResponse": ("pinecone.db_data.models", "FetchResponse"),
     "DeleteRequest": ("pinecone.db_data.models", "DeleteRequest"),
     "DescribeIndexStatsRequest": ("pinecone.db_data.models", "DescribeIndexStatsRequest"),
@@ -31,12 +35,8 @@
     "SingleQueryResults": ("pinecone.db_data.models", "SingleQueryResults"),
     "QueryRequest": ("pinecone.db_data.models", "QueryRequest"),
     "QueryResponse": ("pinecone.db_data.models", "QueryResponse"),
-    "SearchQuery": ("pinecone.db_data.dataclasses", "SearchQuery"),
-    "SearchQueryVector": ("pinecone.db_data.dataclasses", "SearchQueryVector"),
-    "SearchRerank": ("pinecone.db_data.dataclasses", "SearchRerank"),
     "UpsertResponse": ("pinecone.db_data.models", "UpsertResponse"),
     "UpdateRequest": ("pinecone.db_data.models", "UpdateRequest"),
-    "SparseValues": ("pinecone.db_data.models", "SparseValues"),
     "ImportErrorMode": ("pinecone.core.openapi.db_data.model", "ImportErrorMode"),
     "VectorDictionaryMissingKeysError": (
         "pinecone.db_data.errors",

pyproject.toml

@@ -155,7 +155,7 @@ docstring-code-line-length = "dynamic"
 [tool.ruff.lint.per-file-ignores]
 # F403 Allow star imports
 # F401 allow imported but unused
-"__init__.py" = ["F401", "F403"]
+"__init__.py" = ["F401", "F403", "F405"]
 
 # E402 Allow module level import not at top of file so
 #      tqdm warnings can be disabled ahead of loading any code

tests/unit/test_config.py

@@ -103,7 +103,11 @@ def test_config_pool_threads(self):
         pc = Pinecone(
             api_key="test-api-key", host="test-controller-host.pinecone.io", pool_threads=10
         )
-        assert pc.index_api.api_client.pool_threads == 10
+        # DBControl object is created lazily, so we need to access this property
+        # to trigger the setup so we can inspect the config
+        assert pc.db is not None
+
+        assert pc.db.index_api.api_client.pool_threads == 10
         idx = pc.Index(host="my-index-host.pinecone.io", name="my-index-name")
         assert idx._vector_api.api_client.pool_threads == 10
 
@@ -146,5 +150,9 @@ def test_proxy_config(self):
         assert pc.openapi_config.proxy == "http://localhost:8080"
         assert pc.openapi_config.ssl_ca_cert == "path/to/cert-bundle.pem"
 
-        assert pc.index_api.api_client.configuration.proxy == "http://localhost:8080"
-        assert pc.index_api.api_client.configuration.ssl_ca_cert == "path/to/cert-bundle.pem"
+        # DBControl object is created lazily, so we need to access this property
+        # to trigger the setup so we can inspect the config
+        assert pc.db is not None
+
+        assert pc.db.index_api.api_client.configuration.proxy == "http://localhost:8080"
+        assert pc.db.index_api.api_client.configuration.ssl_ca_cert == "path/to/cert-bundle.pem"

tests/unit/test_index_initialization.py

@@ -51,5 +51,6 @@ def test_overwrite_useragent(self):
     def test_set_source_tag(self):
         pc = Pinecone(api_key="123-456-789", source_tag="test_source_tag")
         assert (
-            re.search(r"source_tag=test_source_tag", pc.index_api.api_client.user_agent) is not None
+            re.search(r"source_tag=test_source_tag", pc.db.index_api.api_client.user_agent)
+            is not None
         )

tests/unit/utils/test_docs_links.py

@@ -1,11 +1,17 @@
 import pytest
 import requests
 from pinecone.utils import docslinks
+from pinecone import __version__
 
 urls = list(docslinks.values())
 
 
 @pytest.mark.parametrize("url", urls)
 def test_valid_links(url):
-    response = requests.get(url)
-    assert response.status_code == 200, f"Docs link is invalid: {url}"
+    if isinstance(url, str):
+        response = requests.get(url)
+        assert response.status_code == 200, f"Docs link is invalid: {url}"
+    else:
+        versioned_url = url(__version__)
+        response = requests.get(versioned_url)
+        assert response.status_code == 200, f"Docs link is invalid: {versioned_url}"