# load_basemodels_if_needed
## autogen.tools.load_basemodels_if_needed
```
load_basemodels_if_needed(func)
```
A decorator to load the parameters of a function if they are Pydantic models
| PARAMETER | DESCRIPTION |
| --- | --- |
| `func` | The function with annotated parameters
**TYPE:**`Callable[..., Any]` |
| RETURNS | DESCRIPTION |
| --- | --- |
| `Callable[..., Any]` | A function that loads the parameters before calling the original function |
Source code in `autogen/tools/function_utils.py`
| | |
| --- | --- |
| ```
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
``` | ```
@export_module("autogen.tools")
def load_basemodels_if_needed(func: Callable[..., Any]) -> Callable[..., Any]:
"""A decorator to load the parameters of a function if they are Pydantic models
Args:
func: The function with annotated parameters
Returns:
A function that loads the parameters before calling the original function
"""
# get the type annotations of the parameters
typed_signature = get_typed_signature(func)
param_annotations = get_param_annotations(typed_signature)
# get functions for loading BaseModels when needed based on the type annotations
kwargs_mapping_with_nones = {k: get_load_param_if_needed_function(t) for k, t in param_annotations.items()}
# remove the None values
kwargs_mapping = {k: f for k, f in kwargs_mapping_with_nones.items() if f is not None}
# a function that loads the parameters before calling the original function
@functools.wraps(func)
def _load_parameters_if_needed(*args: Any, **kwargs: Any) -> Any:
# load the BaseModels if needed
for k, f in kwargs_mapping.items():
kwargs[k] = f(kwargs[k], param_annotations[k])
# call the original function
return func(*args, **kwargs)
@functools.wraps(func)
async def _a_load_parameters_if_needed(*args: Any, **kwargs: Any) -> Any:
# load the BaseModels if needed
for k, f in kwargs_mapping.items():
kwargs[k] = f(kwargs[k], param_annotations[k])
# call the original function
return await func(*args, **kwargs)
if is_coroutine_callable(func):
return _a_load_parameters_if_needed
else:
return _load_parameters_if_needed
``` |