Add functools.wraps so we get some arg type hinting

Author: DouweM

pydantic_evals/pydantic_evals/_utils.py

@@ -5,7 +5,7 @@
 import warnings
 from collections.abc import Awaitable, Callable, Generator, Sequence
 from contextlib import contextmanager
-from functools import partial
+from functools import partial, wraps
 from pathlib import Path
 from typing import Any, TypeVar
 
@@ -118,6 +118,7 @@ class LogfireNotConfiguredWarning(UserWarning):
         pass
 
 
+@wraps(_logfire.span)
 @contextmanager
 def logfire_span(*args: Any, **kwargs: Any) -> Generator[logfire_api.LogfireSpan, None, None]:
     """Create a Logfire span without warning if logfire is not configured."""

pydantic_graph/pydantic_graph/_utils.py

@@ -5,7 +5,7 @@
 import warnings
 from collections.abc import Callable, Generator
 from contextlib import contextmanager
-from functools import partial
+from functools import partial, wraps
 from typing import TYPE_CHECKING, Any, TypeAlias, TypeVar, get_args, get_origin
 
 from logfire_api import Logfire, LogfireSpan
@@ -149,6 +149,7 @@ class LogfireNotConfiguredWarning(UserWarning):
         pass
 
 
+@wraps(_logfire.span)
 @contextmanager
 def logfire_span(*args: Any, **kwargs: Any) -> Generator[LogfireSpan, None, None]:
     """Create a Logfire span without warning if logfire is not configured."""