fix: update extra module for new speakeasy type generation

- Remove `type=` parameter from FunctionTool constructor (now a constant)
- Use isinstance() check instead of .type attribute access
- Add proper type annotation for updated_tools list

Author: louis-sanna-dev

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mistralai"
-version = "2.0.0a1"
+version = "2.0.0a2"
 description = "Python Client SDK for the Mistral AI API."
 authors = [{ name = "Mistral" }]
 requires-python = ">=3.10"

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,6 +22,7 @@
     create_tool_call,
 )
 from mistralai.client.models import (
+    AgentTool,
     CompletionArgs,
     CompletionArgsTypedDict,
     ConversationInputs,
@@ -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

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),