fix: update extra module and docs for new speakeasy types

- Update type references: Tools -> ConversationRequestTool
- Remove type= parameter from FunctionTool constructor (now a constant)
- Use isinstance() check instead of .type attribute access
- Document type name changes in MIGRATION.md

Author: louis-sanna-dev

MIGRATION.md

@@ -35,10 +35,22 @@ from mistralai.client.types import BaseModel
 
 ### What Stays the Same
 
-- All method names and signatures remain identical
 - The `Mistral` client API is unchanged
 - All models (`UserMessage`, `AssistantMessage`, etc.) work the same way
 
+### Type Name Changes
+
+Some type names have been updated for clarity and consistency:
+
+| Old Name | New Name |
+|---|---|
+| `Tools` | `ConversationRequestTool` |
+| `ToolsTypedDict` | `ConversationRequestToolTypedDict` |
+| `HandoffExecution` | `ConversationRequestHandoffExecution` |
+| `AgentVersion` | `ConversationRequestAgentVersion` |
+
+Enums now accept unknown values for forward compatibility with API changes.
+
 ---
 
 ## Migrating from v0.x to v1.x

src/mistralai/extra/mcp/base.py

@@ -84,7 +84,6 @@ async def get_tools(self) -> list[FunctionTool]:
         for mcp_tool in mcp_tools.tools:
             tools.append(
                 FunctionTool(
-                    type="function",
                     function=Function(
                         name=mcp_tool.name,
                         description=mcp_tool.description,

src/mistralai/extra/run/context.py

@@ -22,18 +22,19 @@
     create_tool_call,
 )
 from mistralai.client.models import (
+    AgentTool,
     CompletionArgs,
     CompletionArgsTypedDict,
     ConversationInputs,
     ConversationInputsTypedDict,
+    ConversationRequestTool,
+    ConversationRequestToolTypedDict,
     FunctionCallEntry,
     FunctionResultEntry,
     FunctionTool,
     InputEntries,
     MessageInputEntry,
     ResponseFormat,
-    Tools,
-    ToolsTypedDict,
 )
 from mistralai.client.types.basemodel import BaseModel, OptionalNullable, UNSET
 
@@ -50,7 +51,7 @@ class AgentRequestKwargs(typing.TypedDict):
 class ModelRequestKwargs(typing.TypedDict):
     model: str
     instructions: OptionalNullable[str]
-    tools: OptionalNullable[list[Tools] | list[ToolsTypedDict]]
+    tools: OptionalNullable[list[ConversationRequestTool] | list[ConversationRequestToolTypedDict]]
     completion_args: OptionalNullable[CompletionArgs | CompletionArgsTypedDict]
 
 
@@ -186,10 +187,9 @@ async def prepare_agent_request(self, beta_client: "Beta") -> AgentRequestKwargs
             )
         agent = await beta_client.agents.get_async(agent_id=self.agent_id)
         agent_tools = agent.tools or []
-        updated_tools = []
-        for i in range(len(agent_tools)):
-            tool = agent_tools[i]
-            if tool.type != "function":
+        updated_tools: list[AgentTool] = []
+        for tool in agent_tools:
+            if not isinstance(tool, FunctionTool):
                 updated_tools.append(tool)
             elif tool.function.name in self._callable_tools:
                 # function already exists in the agent, don't add it again
@@ -209,7 +209,7 @@ async def prepare_agent_request(self, beta_client: "Beta") -> AgentRequestKwargs
 
     async def prepare_model_request(
         self,
-        tools: OptionalNullable[list[Tools] | list[ToolsTypedDict]] = UNSET,
+        tools: OptionalNullable[list[ConversationRequestTool] | list[ConversationRequestToolTypedDict]] = UNSET,
         completion_args: OptionalNullable[CompletionArgs | CompletionArgsTypedDict] = UNSET,
         instructions: OptionalNullable[str] = None,
     ) -> ModelRequestKwargs:
@@ -225,7 +225,7 @@ async def prepare_model_request(
         request_tools = []
         if isinstance(tools, list):
             for tool in tools:
-                request_tools.append(typing.cast(Tools, tool))
+                request_tools.append(typing.cast(ConversationRequestTool, tool))
         for tool in self.get_tools():
             request_tools.append(tool)
         return ModelRequestKwargs(
@@ -248,7 +248,7 @@ async def _validate_run(
     run_ctx: RunContext,
     inputs: ConversationInputs | ConversationInputsTypedDict,
     instructions: OptionalNullable[str] = UNSET,
-    tools: OptionalNullable[list[Tools] | list[ToolsTypedDict]] = UNSET,
+    tools: OptionalNullable[list[ConversationRequestTool] | list[ConversationRequestToolTypedDict]] = UNSET,
     completion_args: OptionalNullable[CompletionArgs | CompletionArgsTypedDict] = UNSET,
 ) -> tuple[
     AgentRequestKwargs | ModelRequestKwargs, RunResult, list[InputEntries]

src/mistralai/extra/run/tools.py

@@ -168,7 +168,6 @@ def create_tool_call(func: Callable) -> FunctionTool:
     type_hints = get_type_hints(func, include_extras=True, localns=None, globalns=None)
 
     return FunctionTool(
-        type="function",
         function=Function(
             name=name,
             description=_get_function_description(docstring_sections),