From 1ca21077afc12bb3c4d5351b96b8570c29d96175 Mon Sep 17 00:00:00 2001 From: Alexandros Pappas Date: Mon, 17 Feb 2025 11:45:15 +0100 Subject: [PATCH] fix: remove default temperature on all QueryTransformer implementations Signed-off-by: Alexandros Pappas --- .../query/transformation/CompressionQueryTransformer.java | 2 +- .../query/transformation/RewriteQueryTransformer.java | 2 +- .../query/transformation/TranslationQueryTransformer.java | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/CompressionQueryTransformer.java b/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/CompressionQueryTransformer.java index 11cb23b2271..7795dcf902a 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/CompressionQueryTransformer.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/CompressionQueryTransformer.java @@ -84,7 +84,7 @@ public Query transform(Query query) { .user(user -> user.text(this.promptTemplate.getTemplate()) .param("history", formatConversationHistory(query.history())) .param("query", query.text())) - .options(ChatOptions.builder().temperature(0.0).build()) + .options(ChatOptions.builder().build()) .call() .content(); diff --git a/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/RewriteQueryTransformer.java b/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/RewriteQueryTransformer.java index 8dd495029ec..861df1a8e12 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/RewriteQueryTransformer.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/RewriteQueryTransformer.java @@ -82,7 +82,7 @@ public Query transform(Query query) { .user(user -> user.text(this.promptTemplate.getTemplate()) .param("target", this.targetSearchSystem) .param("query", query.text())) - .options(ChatOptions.builder().temperature(0.0).build()) + .options(ChatOptions.builder().build()) .call() .content(); diff --git a/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/TranslationQueryTransformer.java b/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/TranslationQueryTransformer.java index 7ddf2c7d673..6c77cffdd3f 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/TranslationQueryTransformer.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/rag/preretrieval/query/transformation/TranslationQueryTransformer.java @@ -91,7 +91,7 @@ public Query transform(Query query) { .user(user -> user.text(this.promptTemplate.getTemplate()) .param("targetLanguage", this.targetLanguage) .param("query", query.text())) - .options(ChatOptions.builder().temperature(0.0).build()) + .options(ChatOptions.builder().build()) .call() .content();