INNER CODE UNIT · Python

_convert_toml_agent_llm_to_json_format

derisk-ai/OpenDerisk · packages/derisk-app/src/derisk_app/app.py:267

def _convert_toml_agent_llm_to_json_format(toml_agent_llm: Dict[str, Any]) -> Dict[str, Any]:
    """Convert TOML agent.llm format to JSON agent_llm format.

    TOML format (agent.llm):
        { "temperature": 0.5, "provider": [{ "provider": "openai", "api_base": "...", "model": [...] }] }

    JSON format (agent_llm):
        { "temperature": 0.5, "providers": [{ "provider": "openai", "api_base": "...", "models": [...] }] }
    """
    result = dict(toml_agent_llm)
    providers = result.pop("provider", [])
    json_providers = []
    for p in providers:
        if not isinstance(p, dict):
            continue
        converted = dict(p)
        if "model" in converted:
            converted["models"] = converted.pop("model")

View source record →