Initial perf test setup

Author: jhamon

poetry.lock

@@ -101,6 +101,7 @@ pytest-timeout = "2.2.0"
 urllib3_mock = "0.3.3"
 responses = ">=0.8.1"
 black = "^24.4.2"
+pytest-benchmark = "^4.0.0"
 
 [tool.poetry.extras]
 grpc = ["grpcio", "googleapis-common-protos", "lz4", "protobuf", "protoc-gen-openapiv2"]

pyproject.toml

@@ -0,0 +1,20 @@
+services:
+  index1:
+    image: ghcr.io/pinecone-io/pinecone-index:latest
+    environment:
+      PORT: 5081
+      INDEX_TYPE: serverless
+      DIMENSION: 1536
+      METRIC: cosine
+    ports:
+      - "5081:5081"
+
+  index2:
+    image: ghcr.io/pinecone-io/pinecone-index:latest
+    environment:
+      PORT: 5082
+      INDEX_TYPE: pod
+      DIMENSION: 1536
+      METRIC: dot-product
+    ports:
+      - "5082:5082"

tests/perf/__init__.py

@@ -0,0 +1,22 @@
+import random
+import uuid
+from pinecone.grpc import PineconeGRPC, GRPCClientConfig
+
+# Initialize a client. An API key must be passed, but the 
+# value does not matter.
+pc = PineconeGRPC(api_key="test_api_key")
+
+# Target the indexes. Use the host and port number along with disabling tls.
+index = pc.Index(host="localhost:5081", grpc_config=GRPCClientConfig(secure=False))
+dimension = 3
+
+def upserts():
+    vectors = []
+    for batch in range(0, 10):
+        for i in range(0, 100):
+            vectors.append((str(uuid.uuid4()), [random.random()] * dimension))
+        index.upsert(vectors=vectors, namespace="ns2")
+
+def test_upsert(benchmark):
+    benchmark(upserts)
+    print(index.describe_index_stats())