add tool calling test

Author: ajac-zero

pydantic_ai_slim/pydantic_ai/models/openrouter.py

@@ -466,8 +466,10 @@ def _map_model_response(self, message: ModelResponse) -> chat.ChatCompletionMess
             if isinstance(item, TextPart):
                 texts.append(item.content)
             elif isinstance(item, ThinkingPart):
-                if isinstance(item, OpenRouterThinkingPart) and item.provider_name == self.system:
+                if item.provider_name == self.system and isinstance(item, OpenRouterThinkingPart):
                     reasoning_details.append(item.into_reasoning_detail())
+                else:  # pragma: no cover
+                    pass
             elif isinstance(item, ToolCallPart):
                 tool_calls.append(self._map_tool_call(item))
             elif isinstance(item, BuiltinToolCallPart | BuiltinToolReturnPart):  # pragma: no cover

tests/models/cassettes/test_openrouter/test_openrouter_tool_calling.yaml

@@ -0,0 +1,97 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '514'
+      content-type:
+      - application/json
+      host:
+      - openrouter.ai
+    method: POST
+    parsed_body:
+      messages:
+      - content: What is 123 / 456?
+        role: user
+      model: mistralai/mistral-small
+      stream: false
+      tool_choice: auto
+      tools:
+      - function:
+          description: Divide two numbers.
+          name: divide
+          parameters:
+            additionalProperties: false
+            description: Divide two numbers.
+            properties:
+              denominator:
+                type: number
+              numerator:
+                type: number
+              on_inf:
+                default: infinity
+                enum:
+                - error
+                - infinity
+                type: string
+            required:
+            - numerator
+            - denominator
+            type: object
+        type: function
+    uri: https://openrouter.ai/api/v1/chat/completions
+  response:
+    headers:
+      access-control-allow-origin:
+      - '*'
+      connection:
+      - keep-alive
+      content-length:
+      - '585'
+      content-type:
+      - application/json
+      permissions-policy:
+      - payment=(self "https://checkout.stripe.com" "https://connect-js.stripe.com" "https://js.stripe.com" "https://*.js.stripe.com"
+        "https://hooks.stripe.com")
+      referrer-policy:
+      - no-referrer, strict-origin-when-cross-origin
+      transfer-encoding:
+      - chunked
+      vary:
+      - Accept-Encoding
+    parsed_body:
+      choices:
+      - finish_reason: tool_calls
+        index: 0
+        logprobs: null
+        message:
+          content: ''
+          reasoning: null
+          refusal: null
+          role: assistant
+          tool_calls:
+          - function:
+              arguments: '{"numerator": 123, "denominator": 456, "on_inf": "infinity"}'
+              name: divide
+            id: 3sniiMddS
+            index: 0
+            type: function
+        native_finish_reason: tool_calls
+      created: 1762047030
+      id: gen-1762047030-dJUcJW4ildNGqK4UV6iJ
+      model: mistralai/mistral-small
+      object: chat.completion
+      provider: Mistral
+      usage:
+        completion_tokens: 43
+        prompt_tokens: 134
+        total_tokens: 177
+    status:
+      code: 200
+      message: OK
+version: 1

tests/models/test_openrouter.py

@@ -1,8 +1,9 @@
 from collections.abc import Sequence
-from typing import cast
+from typing import Literal, cast
 
 import pytest
 from inline_snapshot import snapshot
+from pydantic import BaseModel
 
 from pydantic_ai import (
     Agent,
@@ -11,9 +12,12 @@
     ModelRequest,
     TextPart,
     ThinkingPart,
+    ToolCallPart,
+    ToolDefinition,
     UnexpectedModelBehavior,
 )
 from pydantic_ai.direct import model_request
+from pydantic_ai.models import ModelRequestParameters
 
 from ..conftest import try_import
 
@@ -69,6 +73,58 @@ async def test_openrouter_with_native_options(allow_model_requests: None, openro
     assert response.provider_details['native_finish_reason'] == 'stop'
 
 
+async def test_openrouter_tool_calling(allow_model_requests: None, openrouter_api_key: str) -> None:
+    provider = OpenRouterProvider(api_key=openrouter_api_key)
+
+    class Divide(BaseModel):
+        """Divide two numbers."""
+
+        numerator: float
+        denominator: float
+        on_inf: Literal['error', 'infinity'] = 'infinity'
+
+    model = OpenRouterModel('mistralai/mistral-small', provider=provider)
+    response = await model_request(
+        model,
+        [ModelRequest.user_text_prompt('What is 123 / 456?')],
+        model_request_parameters=ModelRequestParameters(
+            function_tools=[
+                ToolDefinition(
+                    name=Divide.__name__.lower(),
+                    description=Divide.__doc__,
+                    parameters_json_schema=Divide.model_json_schema(),
+                )
+            ],
+            allow_text_output=True,  # Allow model to either use tools or respond directly
+        ),
+    )
+
+    assert len(response.parts) == 1
+
+    tool_call_part = response.parts[0]
+    assert isinstance(tool_call_part, ToolCallPart)
+    assert tool_call_part.tool_call_id == snapshot('3sniiMddS')
+    assert tool_call_part.tool_name == 'divide'
+    assert tool_call_part.args == snapshot('{"numerator": 123, "denominator": 456, "on_inf": "infinity"}')
+
+    mapped_messages = await model._map_messages([response])  # type: ignore[reportPrivateUsage]
+    tool_call_message = mapped_messages[0]
+    assert tool_call_message['role'] == 'assistant'
+    assert tool_call_message.get('content') is None
+    assert tool_call_message.get('tool_calls') == snapshot(
+        [
+            {
+                'id': '3sniiMddS',
+                'type': 'function',
+                'function': {
+                    'name': 'divide',
+                    'arguments': '{"numerator": 123, "denominator": 456, "on_inf": "infinity"}',
+                },
+            }
+        ]
+    )
+
+
 async def test_openrouter_with_reasoning(allow_model_requests: None, openrouter_api_key: str) -> None:
     provider = OpenRouterProvider(api_key=openrouter_api_key)
     request = ModelRequest.user_text_prompt(