Upgrade to Inference 1.1.0 (w/ Rerank API)

pyproject.toml

@@ -68,7 +68,7 @@ lz4 = { version = ">=3.1.3", optional = true }
 protobuf = { version = "^4.25", optional = true }
 protoc-gen-openapiv2 = {version = "^0.0.1", optional = true }
 pinecone-plugin-interface = "^0.0.7"
-pinecone-plugin-inference = "^1.0.3"
+pinecone-plugin-inference = "^1.1.0"
 
 [tool.poetry.group.types]
 optional = true

tests/integration/inference/test_rerank.py

@@ -0,0 +1,40 @@
+from pinecone import Pinecone
+from pinecone.grpc import PineconeGRPC
+
+
+class TestInferencePluginRerank:
+    def test_rerank(self, api_key):
+        pc = Pinecone(api_key=api_key)
+
+        model = "bge-reranker-v2-m3"
+        result = pc.inference.rerank(
+            model=model,
+            query="i love dogs",
+            documents=["dogs are pretty cool", "everyone loves dogs", "I'm a cat person"],
+            top_n=1,
+            return_documents=True,
+        )
+        assert len(result.data) == 1
+        assert result.data[0].index == 1
+        assert result.data[0].document.text == "everyone loves dogs"
+        assert result.model == model
+        assert isinstance(result.usage.rerank_units, int)
+        assert result.usage.rerank_units == 1
+
+    def test_rerank_grpc(self, api_key):
+        pc = PineconeGRPC(api_key=api_key)
+
+        model = "bge-reranker-v2-m3"
+        result = pc.inference.rerank(
+            model=model,
+            query="i love dogs",
+            documents=["dogs are pretty cool", "everyone loves dogs", "I'm a cat person"],
+            top_n=1,
+            return_documents=True,
+        )
+        assert len(result.data) == 1
+        assert result.data[0].index == 1
+        assert result.data[0].document.text == "everyone loves dogs"
+        assert result.model == model
+        assert isinstance(result.usage.rerank_units, int)
+        assert result.usage.rerank_units == 1