mistral: read tool calls from AIMessage

libs/partners/mistralai/langchain_mistralai/chat_models.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import json
 import logging
 import uuid
 from operator import itemgetter
@@ -42,8 +43,10 @@
     ChatMessageChunk,
     HumanMessage,
     HumanMessageChunk,
+    InvalidToolCall,
     SystemMessage,
     SystemMessageChunk,
+    ToolCall,
     ToolMessage,
 )
 from langchain_core.output_parsers.base import OutputParserLike
@@ -223,6 +226,34 @@ def _convert_delta_to_message_chunk(
         return default_class(content=content)
 
 
+def _format_tool_call_for_mistral(tool_call: ToolCall) -> dict:
+    """Format Langchain ToolCall to dict expected by Mistral."""
+    result: Dict[str, Any] = {
+        "function": {
+            "name": tool_call["name"],
+            "arguments": json.dumps(tool_call["args"]),
+        }
+    }
+    if _id := tool_call.get("id"):
+        result["id"] = _id
+
+    return result
+
+
+def _format_invalid_tool_call_for_mistral(tool_call: InvalidToolCall) -> dict:
+    """Format Langchain ToolCall to dict expected by Mistral."""
+    result: Dict[str, Any] = {
+        "function": {
+            "name": tool_call["name"],
+            "arguments": tool_call["args"],
+        }
+    }
+    if _id := tool_call.get("id"):
+        result["id"] = _id
+
+    return result
+
+
 def _convert_message_to_mistral_chat_message(
     message: BaseMessage,
 ) -> Dict:
@@ -231,8 +262,8 @@ def _convert_message_to_mistral_chat_message(
     elif isinstance(message, HumanMessage):
         return dict(role="user", content=message.content)
     elif isinstance(message, AIMessage):
+        tool_calls = []
         if "tool_calls" in message.additional_kwargs:
-            tool_calls = []
             for tc in message.additional_kwargs["tool_calls"]:
                 chunk = {
                     "function": {
@@ -243,8 +274,15 @@ def _convert_message_to_mistral_chat_message(
                 if _id := tc.get("id"):
                     chunk["id"] = _id
                 tool_calls.append(chunk)
+        elif message.tool_calls or message.invalid_tool_calls:
+            for tool_call in message.tool_calls:
+                tool_calls.append(_format_tool_call_for_mistral(tool_call))
+            for invalid_tool_call in message.invalid_tool_calls:
+                tool_calls.append(
+                    _format_invalid_tool_call_for_mistral(invalid_tool_call)
+                )
         else:
-            tool_calls = []
+            pass
         return {
             "role": "assistant",
             "content": message.content,

libs/partners/mistralai/tests/integration_tests/test_chat_models.py

@@ -138,7 +138,7 @@ def test_structured_output() -> None:
 
 
 def test_streaming_structured_output() -> None:
-    llm = ChatMistralAI(model="mistral-large", temperature=0)
+    llm = ChatMistralAI(model="mistral-large-latest", temperature=0)
 
     class Person(BaseModel):
         name: str
@@ -156,7 +156,7 @@ class Person(BaseModel):
 
 
 def test_tool_call() -> None:
-    llm = ChatMistralAI(model="mistral-large", temperature=0)
+    llm = ChatMistralAI(model="mistral-large-latest", temperature=0)
 
     class Person(BaseModel):
         name: str
@@ -173,7 +173,7 @@ class Person(BaseModel):
 
 
 def test_streaming_tool_call() -> None:
-    llm = ChatMistralAI(model="mistral-large", temperature=0)
+    llm = ChatMistralAI(model="mistral-large-latest", temperature=0)
 
     class Person(BaseModel):
         name: str

libs/partners/mistralai/tests/integration_tests/test_standard.py

@@ -13,3 +13,10 @@ class TestMistralStandard(ChatModelIntegrationTests):
     @pytest.fixture
     def chat_model_class(self) -> Type[BaseChatModel]:
         return ChatMistralAI
+
+    @pytest.fixture
+    def chat_model_params(self) -> dict:
+        return {
+            "model": "mistral-large-latest",
+            "temperature": 0,
+        }