This is an automated email from the ASF dual-hosted git repository. acosentino pushed a commit to branch pinecone-3.0.0 in repository https://gitbox.apache.org/repos/asf/camel.git
commit e30fbb46973960c93c6b2fa622a7cac7c2ad97ee Author: Andrea Cosentino <anco...@gmail.com> AuthorDate: Tue Nov 12 15:57:01 2024 +0100 Upgrade Pinecone Client to version 3.0.0 Signed-off-by: Andrea Cosentino <anco...@gmail.com> --- .../apache/camel/component/pinecone/PineconeVectorDbProducer.java | 7 ++++--- .../apache/camel/component/pinecone/it/PineconeComponentIT.java | 6 ++---- parent/pom.xml | 2 +- 3 files changed, 7 insertions(+), 8 deletions(-) diff --git a/components/camel-ai/camel-pinecone/src/main/java/org/apache/camel/component/pinecone/PineconeVectorDbProducer.java b/components/camel-ai/camel-pinecone/src/main/java/org/apache/camel/component/pinecone/PineconeVectorDbProducer.java index f787c8342a6..351c79ce528 100644 --- a/components/camel-ai/camel-pinecone/src/main/java/org/apache/camel/component/pinecone/PineconeVectorDbProducer.java +++ b/components/camel-ai/camel-pinecone/src/main/java/org/apache/camel/component/pinecone/PineconeVectorDbProducer.java @@ -29,8 +29,9 @@ import org.apache.camel.Exchange; import org.apache.camel.Message; import org.apache.camel.NoSuchHeaderException; import org.apache.camel.support.DefaultProducer; -import org.openapitools.client.model.CollectionModel; -import org.openapitools.client.model.IndexModel; +import org.openapitools.db_control.client.model.CollectionModel; +import org.openapitools.db_control.client.model.DeletionProtection; +import org.openapitools.db_control.client.model.IndexModel; public class PineconeVectorDbProducer extends DefaultProducer { private Pinecone client; @@ -125,7 +126,7 @@ public class PineconeVectorDbProducer extends DefaultProducer { String collectionCloudRegionName = in.getHeader(PineconeVectorDb.Headers.COLLECTION_CLOUD_REGION, String.class); IndexModel result = this.client.createServerlessIndex(indexName, collectionSimilarityMetricName, collectionDimension, - collectionCloudName, collectionCloudRegionName); + collectionCloudName, collectionCloudRegionName, DeletionProtection.DISABLED); populateIndexResponse(result, exchange); diff --git a/components/camel-ai/camel-pinecone/src/test/java/org/apache/camel/component/pinecone/it/PineconeComponentIT.java b/components/camel-ai/camel-pinecone/src/test/java/org/apache/camel/component/pinecone/it/PineconeComponentIT.java index f1ad3fcbc59..211f6fdb797 100644 --- a/components/camel-ai/camel-pinecone/src/test/java/org/apache/camel/component/pinecone/it/PineconeComponentIT.java +++ b/components/camel-ai/camel-pinecone/src/test/java/org/apache/camel/component/pinecone/it/PineconeComponentIT.java @@ -108,9 +108,7 @@ public class PineconeComponentIT extends CamelTestSupport { assertThat(result).isNotNull(); assertThat(result.getException()).isNull(); - assertThat(((QueryResponseWithUnsignedIndices) result.getMessage().getBody()).getMatchesList()).isNotEmpty(); - assertThat( - ((QueryResponseWithUnsignedIndices) result.getMessage().getBody()).getMatches(0).getScore() > 0.9f); + assertThat(!result.getMessage().getBody(QueryResponseWithUnsignedIndices.class).getMatchesList().isEmpty()); } @Test @@ -126,7 +124,7 @@ public class PineconeComponentIT extends CamelTestSupport { assertThat(result).isNotNull(); assertThat(result.getException()).isNull(); - assertThat(((QueryResponseWithUnsignedIndices) result.getMessage().getBody()).getMatchesList()).isNotEmpty(); + assertThat(!result.getMessage().getBody(QueryResponseWithUnsignedIndices.class).getMatchesList().isEmpty()); } @Test diff --git a/parent/pom.xml b/parent/pom.xml index caa9bf0dead..84de4b85754 100644 --- a/parent/pom.xml +++ b/parent/pom.xml @@ -406,7 +406,7 @@ <pgjdbc-driver-version>42.7.4</pgjdbc-driver-version> <pgjdbc-ng-driver-version>0.8.9</pgjdbc-ng-driver-version> <picocli-version>4.7.6</picocli-version> - <pinecone-client-version>1.2.2</pinecone-client-version> + <pinecone-client-version>3.0.0</pinecone-client-version> <plc4x-version>0.12.0</plc4x-version> <plexus-archiver-version>4.10.0</plexus-archiver-version> <plexus-build-api-version>1.2.0</plexus-build-api-version>