test(langchain): types in test_tool_call_limit and test_model_retry (#34629)

Co-authored-by: Mason Daugherty <github@mdrxy.com>
Co-authored-by: Mason Daugherty <mason@langchain.dev>

Author: cbornet

libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_model_retry.py

@@ -1,6 +1,7 @@
 """Tests for ModelRetryMiddleware functionality."""
 
 import time
+from collections.abc import Callable
 from typing import Any
 
 import pytest
@@ -13,7 +14,12 @@
 from langchain.agents.factory import create_agent
 from langchain.agents.middleware._retry import calculate_delay
 from langchain.agents.middleware.model_retry import ModelRetryMiddleware
-from langchain.agents.middleware.types import wrap_model_call
+from langchain.agents.middleware.types import (
+    ModelCallResult,
+    ModelRequest,
+    ModelResponse,
+    wrap_model_call,
+)
 from tests.unit_tests.agents.model import FakeToolCallingModel
 
 
@@ -659,7 +665,10 @@ def test_model_retry_multiple_middleware_composition() -> None:
 
     # Custom middleware that logs calls
     @wrap_model_call
-    def logging_middleware(request, handler):
+    def logging_middleware(
+        request: ModelRequest,
+        handler: Callable[[ModelRequest], ModelResponse],
+    ) -> ModelCallResult:
         call_log.append("before_model")
         response = handler(request)
         call_log.append("after_model")