ag2 0.10.2__py3-none-any.whl

autogen/fast_depends/core/build.py

@@ -0,0 +1,206 @@
+# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# Portions derived from  https://github.com/https://github.com/Lancetnik/FastDepends are under the MIT License.
+# SPDX-License-Identifier: MIT
+
+import inspect
+from collections.abc import Awaitable, Callable, Sequence
+from copy import deepcopy
+from typing import Annotated, Any, TypeVar, get_args, get_origin
+
+from typing_extensions import ParamSpec
+
+from .._compat import ConfigDict, create_model, get_config_base
+from ..dependencies import Depends
+from ..library import CustomField
+from ..utils import (
+    get_typed_signature,
+    is_async_gen_callable,
+    is_coroutine_callable,
+    is_gen_callable,
+)
+from .model import CallModel, ResponseModel
+
+CUSTOM_ANNOTATIONS = (Depends, CustomField)
+
+
+P = ParamSpec("P")
+T = TypeVar("T")
+
+
+def build_call_model(
+    call: Callable[P, T] | Callable[P, Awaitable[T]],
+    *,
+    cast: bool = True,
+    use_cache: bool = True,
+    is_sync: bool | None = None,
+    extra_dependencies: Sequence[Depends] = (),
+    pydantic_config: ConfigDict | None = None,
+) -> CallModel[P, T]:
+    name = getattr(call, "__name__", type(call).__name__)
+
+    is_call_async = is_coroutine_callable(call) or is_async_gen_callable(call)
+    if is_sync is None:
+        is_sync = not is_call_async
+    else:
+        assert not (is_sync and is_call_async), f"You cannot use async dependency `{name}` at sync main"
+
+    typed_params, return_annotation = get_typed_signature(call)
+    if (is_call_generator := is_gen_callable(call) or is_async_gen_callable(call)) and (
+        return_args := get_args(return_annotation)
+    ):
+        return_annotation = return_args[0]
+
+    class_fields: dict[str, tuple[Any, Any]] = {}
+    dependencies: dict[str, CallModel[..., Any]] = {}
+    custom_fields: dict[str, CustomField] = {}
+    positional_args: list[str] = []
+    keyword_args: list[str] = []
+    var_positional_arg: str | None = None
+    var_keyword_arg: str | None = None
+
+    for param_name, param in typed_params.parameters.items():
+        dep: Depends | None = None
+        custom: CustomField | None = None
+
+        if param.annotation is inspect.Parameter.empty:
+            annotation = Any
+
+        elif get_origin(param.annotation) is Annotated:
+            annotated_args = get_args(param.annotation)
+            type_annotation = annotated_args[0]
+
+            custom_annotations = []
+            regular_annotations = []
+            for arg in annotated_args[1:]:
+                if isinstance(arg, CUSTOM_ANNOTATIONS):
+                    custom_annotations.append(arg)
+                else:
+                    regular_annotations.append(arg)
+
+            assert len(custom_annotations) <= 1, (
+                f"Cannot specify multiple `Annotated` Custom arguments for `{param_name}`!"
+            )
+
+            next_custom = next(iter(custom_annotations), None)
+            if next_custom is not None:
+                if isinstance(next_custom, Depends):
+                    dep = next_custom
+                elif isinstance(next_custom, CustomField):
+                    custom = deepcopy(next_custom)
+                else:  # pragma: no cover
+                    raise AssertionError("unreachable")
+
+                annotation = param.annotation if regular_annotations else type_annotation
+            else:
+                annotation = param.annotation
+        else:
+            annotation = param.annotation
+
+        default: Any
+        if param.kind == inspect.Parameter.VAR_POSITIONAL:
+            default = ()
+            var_positional_arg = param_name
+        elif param.kind == inspect.Parameter.VAR_KEYWORD:
+            default = {}
+            var_keyword_arg = param_name
+        elif param.default is inspect.Parameter.empty:
+            default = Ellipsis
+        else:
+            default = param.default
+
+        if isinstance(default, Depends):
+            assert not dep, "You can not use `Depends` with `Annotated` and default both"
+            dep, default = default, Ellipsis
+
+        elif isinstance(default, CustomField):
+            assert not custom, "You can not use `CustomField` with `Annotated` and default both"
+            custom, default = default, Ellipsis
+
+        else:
+            class_fields[param_name] = (annotation, default)
+
+        if dep:
+            if not cast:
+                dep.cast = False
+
+            dependencies[param_name] = build_call_model(
+                dep.dependency,
+                cast=dep.cast,
+                use_cache=dep.use_cache,
+                is_sync=is_sync,
+                pydantic_config=pydantic_config,
+            )
+
+            if dep.cast is True:
+                class_fields[param_name] = (annotation, Ellipsis)
+
+            keyword_args.append(param_name)
+
+        elif custom:
+            assert not (is_sync and is_coroutine_callable(custom.use)), (
+                f"You cannot use async custom field `{type(custom).__name__}` at sync `{name}`"
+            )
+
+            custom.set_param_name(param_name)
+            custom_fields[param_name] = custom
+
+            if custom.cast is False:
+                annotation = Any
+
+            if custom.required:
+                class_fields[param_name] = (annotation, default)
+
+            else:
+                class_fields[param_name] = class_fields.get(param_name, (annotation | None, None))
+
+            keyword_args.append(param_name)
+
+        else:
+            if param.kind is param.KEYWORD_ONLY:
+                keyword_args.append(param_name)
+            elif param.kind not in (inspect.Parameter.VAR_POSITIONAL, inspect.Parameter.VAR_KEYWORD):
+                positional_args.append(param_name)
+
+    func_model = create_model(  # type: ignore[call-overload]
+        name,
+        __config__=get_config_base(pydantic_config),
+        **class_fields,
+    )
+
+    response_model: type[ResponseModel[T]] | None = None
+    if cast and return_annotation and return_annotation is not inspect.Parameter.empty:
+        response_model = create_model(  # type: ignore[call-overload,assignment]
+            "ResponseModel",
+            __config__=get_config_base(pydantic_config),
+            response=(return_annotation, Ellipsis),
+        )
+
+    return CallModel(
+        call=call,
+        model=func_model,
+        response_model=response_model,
+        params=class_fields,
+        cast=cast,
+        use_cache=use_cache,
+        is_async=is_call_async,
+        is_generator=is_call_generator,
+        dependencies=dependencies,
+        custom_fields=custom_fields,
+        positional_args=positional_args,
+        keyword_args=keyword_args,
+        var_positional_arg=var_positional_arg,
+        var_keyword_arg=var_keyword_arg,
+        extra_dependencies=[
+            build_call_model(
+                d.dependency,
+                cast=d.cast,
+                use_cache=d.use_cache,
+                is_sync=is_sync,
+                pydantic_config=pydantic_config,
+            )
+            for d in extra_dependencies
+        ],
+    )