(camel) 01/01: CAMEL-20739 - Camel-Pinecone: Add a datatype for transforming langchain embeddings in Pinecone objects - Manage index id and index name

[acosentino]

This is an automated email from the ASF dual-hosted git repository.

acosentino pushed a commit to branch pinecone-refine
in repository https://gitbox.apache.org/repos/asf/camel.git

commit c55da88e4820408252265811450dd234a9756dfc
Author: Andrea Cosentino <anco...@gmail.com>
AuthorDate: Tue May 7 09:17:24 2024 +0200

    CAMEL-20739 - Camel-Pinecone: Add a datatype for transforming langchain 
embeddings in Pinecone objects - Manage index id and index name
    
    Signed-off-by: Andrea Cosentino <anco...@gmail.com>
---
 .../pinecone/transform/PineconeEmbeddingsDataTypeTransformer.java   | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git 
a/components/camel-pinecone/src/main/java/org/apache/camel/component/pinecone/transform/PineconeEmbeddingsDataTypeTransformer.java
 
b/components/camel-pinecone/src/main/java/org/apache/camel/component/pinecone/transform/PineconeEmbeddingsDataTypeTransformer.java
index c3d4329bbbe..3824c53e269 100644
--- 
a/components/camel-pinecone/src/main/java/org/apache/camel/component/pinecone/transform/PineconeEmbeddingsDataTypeTransformer.java
+++ 
b/components/camel-pinecone/src/main/java/org/apache/camel/component/pinecone/transform/PineconeEmbeddingsDataTypeTransformer.java
@@ -37,9 +37,11 @@ public class PineconeEmbeddingsDataTypeTransformer extends 
Transformer {
     @Override
     public void transform(Message message, DataType fromType, DataType toType) 
{
         Embedding embedding = 
message.getHeader("CamelLangChain4jEmbeddingsVector", Embedding.class);
+        String indexId = message.getHeader(PineconeVectorDb.Headers.INDEX_ID, 
UUID.randomUUID(), String.class);
+        String indexName = 
message.getHeader(PineconeVectorDb.Headers.INDEX_NAME, "embeddings", 
String.class);
 
-        message.setHeader(PineconeVectorDb.Headers.INDEX_NAME, "embeddings");
-        message.setHeader(PineconeVectorDb.Headers.INDEX_ID, 
UUID.randomUUID());
+        message.setHeader(PineconeVectorDb.Headers.INDEX_NAME, indexName);
+        message.setHeader(PineconeVectorDb.Headers.INDEX_ID, indexId);
         message.setBody(embedding.vectorAsList());
     }
 }