Fix OpenAiApi tool_choice value and refactor MistralAi class

spring-projects · Mar 21, 2024 · 8510f00 · 8510f00
1 parent 766b420
commit 8510f00
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/...spring-ai-mistral-ai/src/main/java/org/springframework/ai/mistralai/api/MistralAiApi.java b/...spring-ai-mistral-ai/src/main/java/org/springframework/ai/mistralai/api/MistralAiApi.java
@@ -705,7 +705,7 @@ public ResponseEntity<ChatCompletion> chatCompletionEntity(ChatCompletionRequest
 			.toEntity(ChatCompletion.class);
 	}
 
-	private MIstralAiStreamFunctionCallingHelper chunkMerger = new MIstralAiStreamFunctionCallingHelper();
+	private MistralAiStreamFunctionCallingHelper chunkMerger = new MistralAiStreamFunctionCallingHelper();
 
 	/**
 	 * Creates a streaming chat response for the given chat conversation.

diff --git a/...MIstralAiStreamFunctionCallingHelper.java → ...MistralAiStreamFunctionCallingHelper.java b/...MIstralAiStreamFunctionCallingHelper.java → ...MistralAiStreamFunctionCallingHelper.java
@@ -37,7 +37,7 @@
  * @author Christian Tzolov
  * @since 0.8.1
  */
-public class MIstralAiStreamFunctionCallingHelper {
+public class MistralAiStreamFunctionCallingHelper {
 
 	/**
 	 * Merge the previous and current ChatCompletionChunk into a single one.

diff --git a/models/spring-ai-openai/src/main/java/org/springframework/ai/openai/api/OpenAiApi.java b/models/spring-ai-openai/src/main/java/org/springframework/ai/openai/api/OpenAiApi.java
@@ -380,7 +380,7 @@ public static class ToolChoiceBuilder {
 			/**
 			 * Model can pick between generating a message or calling a function.
 			 */
-			public static final String AUTO = "none";
+			public static final String AUTO = "auto";
 			/**
 			 * Model will not call a function and instead generates a message
 			 */