Skip to content

datarobot_genai.dragent.plugins.llm_clients

llm_clients

apply_reasoning_config 

apply_reasoning_config(config: dict[str, Any], llm_config: SupportsReasoningConfig) -> dict[str, Any]

Map workflow reasoning to provider extra_body and temperature.

Source code in datarobot_genai/dragent/plugins/llm_clients.py 
84
85
86
87
88
89
90
91
92
93
94
def apply_reasoning_config(
    config: dict[str, Any],
    llm_config: SupportsReasoningConfig,
) -> dict[str, Any]:
    """Map workflow ``reasoning`` to provider ``extra_body`` and temperature."""
    return apply_reasoning_to_parameters(
        config,
        reasoning=llm_config.reasoning,
        model_name=_resolve_model_name(llm_config, config),
        explicit_extra_body="extra_body" in llm_config.model_fields_set,
    )

prepare_llm_parameters 

prepare_llm_parameters(llm_config: SupportsReasoningConfig, *, exclude_unset: bool = False) -> dict[str, Any]

Dump LLM config kwargs and apply reasoning / extra_body mapping.

Source code in datarobot_genai/dragent/plugins/llm_clients.py 
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
def prepare_llm_parameters(
    llm_config: SupportsReasoningConfig,
    *,
    exclude_unset: bool = False,
) -> dict[str, Any]:
    """Dump LLM config kwargs and apply ``reasoning`` / ``extra_body`` mapping."""
    dumped = llm_config.model_dump(
        exclude=EXCLUDE_FIELDS,
        by_alias=True,
        exclude_none=True,
        exclude_unset=exclude_unset,
    )
    return apply_reasoning_config(dumped, llm_config)