Function pydantic_model_from_param_schema

python/composio/utils/shared.py:460–544

Dynamically creates a Pydantic model from a schema dictionary. :param param_schema: Schema with 'title', 'properties', and optionally 'required' keys. :return: A Pydantic model class for the defined schema. :raised ValueError: Invalid 'type' for property or recursive model creatio

(param_schema: t.Dict)

def pydantic_model_from_param_schema(param_schema: t.Dict) -> t.Type:
    """
    Dynamically creates a Pydantic model from a schema dictionary.

    :param param_schema: Schema with 'title', 'properties', and optionally 'required' keys.
    :return: A Pydantic model class for the defined schema.

    :raised ValueError: Invalid 'type' for property or recursive model creation.

    Note: Requires global `schema_type_python_type_dict` for type mapping and
        `fallback_values` for default values.
    """
    required_fields = {}
    optional_fields = {}
    if "title" not in param_schema:
        raise ValueError(f"Missing 'title' in param_schema: {param_schema}")

    param_title = str(param_schema["title"]).replace(" ", "")
    required_props = param_schema.get("required", [])

    if param_schema.get("type") == "array":
        # print("param_schema inside array - ", param_schema)
        item_schema = param_schema.get("items")
        if item_schema:
            ItemType = t.cast(
                t.Type,
                json_schema_to_pydantic_type(
                    json_schema=item_schema,
                ),
            )
            return t.List[ItemType]  # type: ignore
        return t.List

    for prop_name, prop_info in param_schema.get("properties", {}).items():
        prop_type = prop_info.get("type")
        prop_title = prop_info.get("title", prop_name).replace(" ", "")
        prop_default = prop_info.get("default", FALLBACK_VALUES.get(prop_type))
        if (
            prop_type is not None
            and prop_type in PYDANTIC_TYPE_TO_PYTHON_TYPE
            and prop_type not in CONTAINER_TYPE
        ):
            signature_prop_type = PYDANTIC_TYPE_TO_PYTHON_TYPE[prop_type]
        elif prop_type is None:
            # Schema uses anyOf/allOf/oneOf/$ref instead of a top-level "type" key.
            # Delegate to json_schema_to_pydantic_type which handles all combiners.
            signature_prop_type = t.cast(
                t.Type,
                json_schema_to_pydantic_type(json_schema=prop_info),
            )
        else:
            signature_prop_type = pydantic_model_from_param_schema(prop_info)

        field_kwargs = {
            "description": prop_info.get(
                "description", prop_info.get("desc", prop_title)
            ),
        }