Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Resolve Mistral AI Function Call issue and activate testing.

Browse Source 
Signed-off-by: Ricken Bazolo <ricken.bazolo@gmail.com>
This commit is contained in:
Ricken Bazolo
2025-01-18 15:55:07 +01:00
committed by Ilayaperumal Gopinathan
parent b77e084bf9
commit a0e0508cc0
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
models/spring-ai-mistral-ai/src/test/java/org/springframework/ai/mistralai/api/tool/MistralAiApiToolFunctionCallIT.java
View File

@@ -21,7 +21,6 @@ import java.util.List;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
import org.slf4j.Logger;
@@ -37,14 +36,15 @@ import org.springframework.ai.mistralai.api.MistralAiApi.ChatCompletionRequest.T
import org.springframework.ai.mistralai.api.MistralAiApi.FunctionTool.Type;
import org.springframework.ai.model.ModelOptionsUtils;
import org.springframework.http.ResponseEntity;
import org.springframework.util.ObjectUtils;
import static org.assertj.core.api.Assertions.assertThat;
/**
* @author Christian Tzolov
* @author Ricken Bazolo
*/
@EnabledIfEnvironmentVariable(named = "MISTRAL_AI_API_KEY", matches = ".+")
@Disabled
public class MistralAiApiToolFunctionCallIT {
static final String MISTRAL_AI_CHAT_MODEL = MistralAiApi.ChatModel.LARGE.getValue();
@@ -121,7 +121,7 @@ public class MistralAiApiToolFunctionCallIT {
assertThat(responseMessage.toolCalls()).isNotNull();
// Check if the model wanted to call a function
if (responseMessage.toolCalls() != null) {
if (!ObjectUtils.isEmpty(responseMessage.toolCalls())) {
// extend conversation with assistant's reply.
messages.add(responseMessage);
@@ -137,7 +137,7 @@ public class MistralAiApiToolFunctionCallIT {
// extend conversation with function response.
messages.add(new ChatCompletionMessage("" + weatherResponse.temp() + weatherRequest.unit(),
Role.TOOL, functionName, null));
Role.TOOL, functionName, null, toolCall.id()));
}
}