mistral: respect ChatMessageTypeTool

tmc · Mar 26, 2024 · 5e9b694 · 5e9b694
1 parent 87b90eb
commit 5e9b694
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llms/mistral/mistralmodel.go b/llms/mistral/mistralmodel.go
@@ -233,7 +233,7 @@ func setMistralChatMessageRole(msg *llms.MessageContent, chatMsg *sdk.ChatMessag
 		chatMsg.Role = "assistant"
 	case schema.ChatMessageTypeGeneric, schema.ChatMessageTypeHuman:
 		chatMsg.Role = "user"
-	case schema.ChatMessageTypeFunction:
+	case schema.ChatMessageTypeFunction, schema.ChatMessageTypeTool:
 		chatMsg.Role = "tool"
 	case schema.ChatMessageTypeSystem:
 		chatMsg.Role = "system"