Run tests on prod

jhamon · jhamon · commit 706351ef6973 · 2025-05-20T12:33:40.000-04:00
diff --git a/.github/actions/cleanup-all/action.yml b/.github/actions/cleanup-all/action.yml
@@ -12,7 +12,7 @@ inputs:
   PINECONE_ADDITIONAL_HEADERS:
     description: 'Additional headers to send with the request'
     required: false
-    default: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+    default: '{"sdk-test-suite": "pinecone-python-client"}'
 
 runs:
   using: 'composite'
diff --git a/.github/actions/create-index-legacy/action.yml b/.github/actions/create-index-legacy/action.yml
@@ -26,7 +26,7 @@ inputs:
   PINECONE_ADDITIONAL_HEADERS:
     description: 'Additional headers to send with the request'
     required: false
-    default: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+    default: '{"sdk-test-suite": "pinecone-python-client"}'
 
 runs:
   using: 'composite'
diff --git a/.github/actions/create-index/action.yml b/.github/actions/create-index/action.yml
@@ -28,7 +28,7 @@ inputs:
   PINECONE_ADDITIONAL_HEADERS:
     description: 'Additional headers to send with the request'
     required: false
-    default: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+    default: '{"sdk-test-suite": "pinecone-python-client"}'
 
 
 outputs:
diff --git a/.github/actions/delete-index/action.yml b/.github/actions/delete-index/action.yml
@@ -11,7 +11,7 @@ inputs:
   PINECONE_ADDITIONAL_HEADERS:
     description: 'Additional headers to send with the request'
     required: false
-    default: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+    default: '{"sdk-test-suite": "pinecone-python-client"}'
 
 
 runs:
diff --git a/.github/actions/test-data-asyncio/action.yaml b/.github/actions/test-data-asyncio/action.yaml
@@ -18,7 +18,7 @@ inputs:
   PINECONE_ADDITIONAL_HEADERS:
     description: 'Additional headers to send with the request'
     required: false
-    default: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+    default: '{"sdk-test-suite": "pinecone-python-client"}'
   python_version:
     description: 'The version of Python to use'
     required: false
diff --git a/.github/actions/test-data-plane/action.yaml b/.github/actions/test-data-plane/action.yaml
@@ -21,7 +21,7 @@ inputs:
   PINECONE_ADDITIONAL_HEADERS:
     description: 'Additional headers to send with the request'
     required: false
-    default: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+    default: '{"sdk-test-suite": "pinecone-python-client"}'
   python_version:
     description: 'The version of Python to use'
     required: false
diff --git a/.github/actions/test-dependency-asyncio-rest/action.yaml b/.github/actions/test-dependency-asyncio-rest/action.yaml
@@ -8,7 +8,7 @@ inputs:
   PINECONE_ADDITIONAL_HEADERS:
     description: 'Additional headers to send with the request'
     required: false
-    default: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+    default: '{"sdk-test-suite": "pinecone-python-client"}'
   index_name:
     description: 'The name of the index'
     required: true
diff --git a/.github/actions/test-dependency-grpc/action.yaml b/.github/actions/test-dependency-grpc/action.yaml
@@ -8,7 +8,7 @@ inputs:
   PINECONE_ADDITIONAL_HEADERS:
     description: 'Additional headers to send with the request'
     required: false
-    default: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+    default: '{"sdk-test-suite": "pinecone-python-client"}'
   index_name:
     description: 'The name of the index'
     required: true
diff --git a/.github/actions/test-dependency-rest/action.yaml b/.github/actions/test-dependency-rest/action.yaml
@@ -8,7 +8,7 @@ inputs:
   PINECONE_ADDITIONAL_HEADERS:
     description: 'Additional headers to send with the request'
     required: false
-    default: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+    default: '{"sdk-test-suite": "pinecone-python-client"}'
   index_name:
     description: 'The name of the index'
     required: true
diff --git a/.github/workflows/testing-dependency-asyncio.yaml b/.github/workflows/testing-dependency-asyncio.yaml
@@ -27,5 +27,5 @@ jobs:
           python_version: '${{ matrix.python_version }}'
           index_name: '${{ inputs.index_name }}'
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
           aiohttp_version: '${{ matrix.aiohttp_version }}'
diff --git a/.github/workflows/testing-dependency-grpc.yaml b/.github/workflows/testing-dependency-grpc.yaml
@@ -53,7 +53,7 @@ jobs:
           python_version: '${{ matrix.python_version }}'
           index_name: '${{ inputs.index_name }}'
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
           grpcio_version: '${{ matrix.grpcio_version }}'
           lz4_version: '${{ matrix.lz4_version }}'
           protobuf_version: '${{ matrix.protobuf_version }}'
@@ -87,7 +87,7 @@ jobs:
           python_version: '${{ matrix.python_version }}'
           index_name: '${{ inputs.index_name }}'
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
           grpcio_version: '${{ matrix.grpcio_version }}'
           lz4_version: '${{ matrix.lz4_version }}'
           protobuf_version: '${{ matrix.protobuf_version }}'
@@ -120,7 +120,7 @@ jobs:
           python_version: '${{ matrix.python_version }}'
           index_name: '${{ inputs.index_name }}'
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
           grpcio_version: '${{ matrix.grpcio_version }}'
           lz4_version: '${{ matrix.lz4_version }}'
           protobuf_version: '${{ matrix.protobuf_version }}'
diff --git a/.github/workflows/testing-dependency-rest.yaml b/.github/workflows/testing-dependency-rest.yaml
@@ -30,7 +30,7 @@ jobs:
           python_version: '${{ matrix.python_version }}'
           index_name: '${{ inputs.index_name }}'
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
           urllib3_version: '${{ matrix.urllib3_version }}'
 
 
@@ -54,7 +54,7 @@ jobs:
           python_version: '${{ matrix.python_version }}'
           index_name: '${{ inputs.index_name }}'
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
           urllib3_version: '${{ matrix.urllib3_version }}'
 
   dependency-matrix-rest-313:
@@ -77,5 +77,5 @@ jobs:
           python_version: '${{ matrix.python_version }}'
           index_name: '${{ inputs.index_name }}'
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
           urllib3_version: '${{ matrix.urllib3_version }}'
diff --git a/.github/workflows/testing-dependency.yaml b/.github/workflows/testing-dependency.yaml
@@ -19,7 +19,7 @@ jobs:
           name_prefix: depstest-${{ github.run_number }}
           dimension: 2
           PINECONE_API_KEY: ${{ secrets.PINECONE_API_KEY }}
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
 
   dependency-test-rest:
     uses: './.github/workflows/testing-dependency-rest.yaml'
@@ -56,4 +56,4 @@ jobs:
         with:
           index_name: '${{ needs.deps-test-setup.outputs.index_name }}'
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
diff --git a/.github/workflows/testing-integration-asyncio.yaml b/.github/workflows/testing-integration-asyncio.yaml
@@ -29,7 +29,7 @@ jobs:
         run: poetry run pytest tests/integration/data_asyncio --retries 2 --retry-delay 35 -s -vv --log-cli-level=DEBUG
         env:
             PINECONE_API_KEY: ${{ secrets.PINECONE_API_KEY }}
-            PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+            PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
 
   db-control-asyncio:
     name: db_control asyncio
@@ -55,4 +55,4 @@ jobs:
         run: poetry run pytest tests/integration/control_asyncio/*.py --retries 2 --retry-delay 35 -s -vv --log-cli-level=DEBUG
         env:
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
diff --git a/.github/workflows/testing-integration.yaml b/.github/workflows/testing-integration.yaml
@@ -8,7 +8,7 @@ jobs:
     runs-on: ubuntu-latest
     env:
       PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-      PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+      PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
     strategy:
       fail-fast: false
       matrix:
@@ -54,7 +54,7 @@ jobs:
         run: poetry run pytest tests/integration/inference --retries 2 --retry-delay 35 -s -vv --log-cli-level=DEBUG
         env:
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
 
   plugins:
     name: Plugin installation
@@ -76,7 +76,7 @@ jobs:
         run: poetry run pytest tests/integration/plugins --retries 2 --retry-delay 35 -s -vv --log-cli-level=DEBUG
         env:
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
 
 
 
@@ -107,7 +107,7 @@ jobs:
           metric: 'cosine'
           spec: '{ "serverless": { "region": "us-west-2", "cloud": "aws" }}'
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
           freshness_timeout_seconds: 600
           skip_weird_id_tests: 'true'
 
@@ -172,6 +172,6 @@ jobs:
         run: poetry run pytest tests/integration/control/serverless --retries 2 --retry-delay 35 -s -vv --log-cli-level=DEBUG
         env:
           PINECONE_API_KEY: '${{ secrets.PINECONE_API_KEY }}'
-          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client", "x-environment": "preprod-aws-0"}'
+          PINECONE_ADDITIONAL_HEADERS: '{"sdk-test-suite": "pinecone-python-client"}'
           SERVERLESS_CLOUD: '${{ matrix.testConfig.serverless.cloud }}'
           SERVERLESS_REGION: '${{ matrix.testConfig.serverless.region }}'
diff --git a/tests/integration/data_asyncio/test_query.py b/tests/integration/data_asyncio/test_query.py
@@ -4,50 +4,57 @@
 from .conftest import build_asyncioindex_client, poll_for_freshness
 from ..helpers import random_string, embedding_values
 
+import logging
+
+logger = logging.getLogger(__name__)
+
 
 @pytest.mark.asyncio
 @pytest.mark.parametrize("target_namespace", [random_string(20)])
 async def test_query(index_host, dimension, target_namespace):
     asyncio_idx = build_asyncioindex_client(index_host)
+    logger.info(f"Testing query on index {index_host}")
+    logger.info(f"Target namespace: {target_namespace}")
+    logger.info(f"Dimension: {dimension}")
 
     def emb():
         return embedding_values(dimension)
 
     # Upsert with tuples
-    await asyncio_idx.upsert(
-        vectors=[("1", emb()), ("2", emb()), ("3", emb())], namespace=target_namespace
-    )
+    tuple_vectors = [("1", emb()), ("2", emb()), ("3", emb())]
+    logger.info(f"Upserting {len(tuple_vectors)} vectors")
+    await asyncio_idx.upsert(vectors=tuple_vectors, namespace=target_namespace)
 
     # Upsert with objects
-    await asyncio_idx.upsert(
-        vectors=[
-            Vector(id="4", values=emb(), metadata={"genre": "action"}),
-            Vector(id="5", values=emb(), metadata={"genre": "action"}),
-            Vector(id="6", values=emb(), metadata={"genre": "horror"}),
-        ],
-        namespace=target_namespace,
-    )
+    object_vectors = [
+        Vector(id="4", values=emb(), metadata={"genre": "action"}),
+        Vector(id="5", values=emb(), metadata={"genre": "action"}),
+        Vector(id="6", values=emb(), metadata={"genre": "horror"}),
+    ]
+    logger.info(f"Upserting {len(object_vectors)} vectors")
+    await asyncio_idx.upsert(vectors=object_vectors, namespace=target_namespace)
 
     # Upsert with dict
-    await asyncio_idx.upsert(
-        vectors=[
-            {"id": "7", "values": emb()},
-            {"id": "8", "values": emb()},
-            {"id": "9", "values": emb()},
-        ],
-        namespace=target_namespace,
-    )
+    dict_vectors = [
+        {"id": "7", "values": emb()},
+        {"id": "8", "values": emb()},
+        {"id": "9", "values": emb()},
+    ]
+    logger.info(f"Upserting {len(dict_vectors)} vectors")
+    await asyncio_idx.upsert(vectors=dict_vectors, namespace=target_namespace)
 
     await poll_for_freshness(asyncio_idx, target_namespace, 9)
 
     # Check the vector count reflects some data has been upserted
     stats = await asyncio_idx.describe_index_stats()
+    logger.info(f"Index stats: {stats}")
     assert stats.total_vector_count >= 9
     # default namespace could have other stuff from other tests
     if target_namespace != "":
         assert stats.namespaces[target_namespace].vector_count == 9
 
     results1 = await asyncio_idx.query(top_k=4, namespace=target_namespace, vector=emb())
+    logger.info(f"Results 1: {results1}")
     assert results1 is not None
     assert len(results1.matches) == 4
     assert results1.namespace == target_namespace
@@ -67,6 +74,7 @@ def emb():
     results2 = await asyncio_idx.query(
         top_k=4, namespace=target_namespace, vector=emb(), include_values=True
     )
+    logger.info(f"Results 2: {results2}")
     assert results2 is not None
     assert len(results2.matches) == 4
     assert results2.namespace == target_namespace
@@ -81,6 +89,7 @@ def emb():
         include_metadata=True,
         include_values=True,
     )
+    logger.info(f"Results 3: {results3}")
     assert results3 is not None
     assert len(results3.matches) == 2
     assert results3.namespace == target_namespace
@@ -97,6 +106,7 @@ def emb():
         include_metadata=True,
         include_values=True,
     )
+    logger.info(f"Results 4: {results4}")
     assert results4 is not None
     assert len(results4.matches) == 1
     assert results4.namespace == target_namespace
@@ -113,6 +123,7 @@ def emb():
         include_metadata=True,
         include_values=True,
     )
+    logger.info(f"Results 5: {results5}")
     assert results5 is not None
     assert len(results5.matches) == 0
     assert results5.namespace == target_namespace
@@ -122,27 +133,32 @@ def emb():
 
     # Query by id
     results6 = await asyncio_idx.query(top_k=4, id="1", namespace=target_namespace)
+    logger.info(f"Results 6: {results6}")
     assert results6 is not None
     assert len(results6.matches) == 4
 
     # Query by id, when id doesn't exist gives empty result set
     results7 = await asyncio_idx.query(top_k=10, id="unknown", namespace=target_namespace)
+    logger.info(f"Results 7: {results7}")
     assert results7 is not None
     assert len(results7.matches) == 0
 
     # When missing required top_k kwarg
     with pytest.raises(TypeError) as e:
         await asyncio_idx.query(id="1", namespace=target_namespace)
+    logger.info(f"Error Msg 1: {e.value}")
     assert "top_k" in str(e.value)
 
     # When incorrectly passing top_k as a positional argument
     with pytest.raises(TypeError) as e:
         await asyncio_idx.query(4, id="1", namespace=target_namespace)
+    logger.info(f"Error Msg 2: {e.value}")
     assert "top_k" in str(e.value)
 
     # When trying to pass both id and vector as query params
     with pytest.raises(ValueError) as e:
         await asyncio_idx.query(top_k=10, id="1", vector=emb(), namespace=target_namespace)
+    logger.info(f"Error Msg 3: {e.value}")
     assert "Cannot specify both `id` and `vector`" in str(e.value)
 
     # When trying to pass sparse vector as query params to dense index
@@ -152,5 +168,6 @@ def emb():
             sparse_vector={"indices": [i for i in range(dimension)], "values": emb()},
             namespace=target_namespace,
         )
+    logger.info(f"Error Msg 4: {e.value}")
     assert "Cannot query index with dense 'vector_type' with only sparse vector" in str(e.value)
     await asyncio_idx.close()
