From b706d55b95fe16c2f31d3f02868dc5d526072150 Mon Sep 17 00:00:00 2001
From: alex-ac <alexandre.abouchahine@mistral.ai>
Date: Mon, 16 Jun 2025 16:58:06 +0200
Subject: [PATCH] fix: response format in run context badly set

---
 src/mistralai/extra/run/context.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/mistralai/extra/run/context.py b/src/mistralai/extra/run/context.py
index a79fd59e..08350a84 100644
--- a/src/mistralai/extra/run/context.py
+++ b/src/mistralai/extra/run/context.py
@@ -30,7 +30,6 @@
     FunctionTool,
     MessageInputEntry,
     InputEntries,
-    ResponseFormatTypedDict,
 )
 
 from logging import getLogger
@@ -229,9 +228,8 @@ async def prepare_model_request(
         elif isinstance(completion_args, CompletionArgs) and self.output_format:
             completion_args.response_format = self.response_format
         elif isinstance(completion_args, dict) and self.output_format:
-            completion_args["response_format"] = typing.cast(
-                ResponseFormatTypedDict, self.response_format.model_dump()
-            )
+            completion_args = CompletionArgs.model_validate(completion_args)
+            completion_args.response_format = self.response_format
         request_tools = []
         if isinstance(tools, list):
             for tool in tools: