Fix mypy

Author: gopidesupavan

providers/common/ai/src/airflow/providers/common/ai/utils/function_schema.py

@@ -80,7 +80,7 @@ def extract_function_description(fn: Callable[..., Any]) -> str:
             or type(fn).__name__
         )
 
-    return _first_docstring_paragraph(fn) or fn.__name__  # type: ignore[return-value]
+    return _first_docstring_paragraph(fn) or fn.__name__
 
 
 def build_function_json_schema(fn: Callable[..., Any]) -> dict[str, Any]:

providers/common/ai/tests/unit/common/ai/utils/test_function_schema.py

@@ -34,7 +34,7 @@
 # ---------------------------------------------------------------------------
 
 
-def _plain(x: int, y: str = "hi") -> str:
+def _plain(x: int, y: str = "hi") -> str:  # type: ignore[empty-body]
     """Do something useful.
 
     Args:
@@ -46,22 +46,22 @@ def _plain(x: int, y: str = "hi") -> str:
     """
 
 
-def _no_doc(x: int) -> str:
+def _no_doc(x: int) -> str:  # type: ignore[empty-body]
     pass
 
 
 def _no_params() -> None:
     """No parameters at all."""
 
 
-def _annotated(q: Annotated[str, "The search query"], limit: Annotated[int, "Max results"] = 10) -> list:
+def _annotated(q: Annotated[str, "The search query"], limit: Annotated[int, "Max results"] = 10) -> list:  # type: ignore[empty-body]
     """Search."""
 
 
 class _CallableObj:
     """Callable object used to test non-function callables."""
 
-    def __call__(self, value: str) -> str:
+    def __call__(self, value: str) -> str:  # type: ignore[empty-body]
         """Process value."""
 
 
@@ -96,13 +96,13 @@ def test_callable_object_falls_back_to_class_docstring(self):
         class _NoCallDoc:
             """Describes the class."""
 
-            def __call__(self, x: int) -> int: ...
+            def __call__(self, x: int) -> int: ...  # type: ignore[empty-body]
 
         assert extract_function_description(_NoCallDoc()) == "Describes the class."
 
     def test_callable_object_falls_back_to_class_name(self):
         class _NoDocs:
-            def __call__(self, x: int) -> int: ...
+            def __call__(self, x: int) -> int: ...  # type: ignore[empty-body]
 
         assert extract_function_description(_NoDocs()) == "_NoDocs"
 
@@ -270,15 +270,15 @@ def test_additional_properties_stripped(self):
         assert "additionalProperties" not in schema
 
     def test_partial_positional_bind_removes_param(self):
-        def add(a: int, b: int) -> int: ...
+        def add(a: int, b: int) -> int: ...  # type: ignore[empty-body]
 
         p = functools.partial(add, 1)
         schema = build_function_json_schema(p)
         assert "a" not in schema.get("properties", {})
         assert "b" in schema["properties"]
 
     def test_partial_keyword_bind_keeps_param_as_optional(self):
-        def add(a: int, b: int) -> int: ...
+        def add(a: int, b: int) -> int: ...  # type: ignore[empty-body]
 
         p = functools.partial(add, a=1)
         schema = build_function_json_schema(p)
@@ -287,7 +287,7 @@ def add(a: int, b: int) -> int: ...
         assert "b" in schema["required"]
 
     def test_nested_partial_unwraps_hint_source(self):
-        def fn(x: int, y: str) -> str: ...
+        def fn(x: int, y: str) -> str: ...  # type: ignore[empty-body]
 
         p = functools.partial(functools.partial(fn, 1), "hello")
         schema = build_function_json_schema(p)