Function create_ide_agent

python/vibe-coding-ide/backend/src/agent/agent.py:47–83

Factory to construct the IDE Agent with an optional model override. If model is provided, attempt to set it on the Agent. If the underlying Agent class does not accept a model parameter, gracefully ignore it.

(model: str | None = None, oidc_token: str | None = None)

def create_ide_agent(model: str | None = None, oidc_token: str | None = None) -> Agent:
    """Factory to construct the IDE Agent with an optional model override.

    If model is provided, attempt to set it on the Agent. If the underlying
    Agent class does not accept a model parameter, gracefully ignore it.
    """
    base_kwargs = {
        "name": "IDE Agent",
        "instructions": instructions,
        "tools": [
            think,
            # fs ops
            edit_code,
            create_file,
            delete_file,
            rename_file,
            create_folder,
            delete_folder,
            rename_folder,
            # sandbox controls
            sandbox_create,
            sandbox_stop,
            sandbox_run,
            sandbox_set_env,
            sandbox_show_preview,
        ],
    }
    os.environ["VERCEL_AI_GATEWAY_API_KEY"] = os.getenv("VERCEL_AI_GATEWAY_API_KEY") or os.getenv("AI_GATEWAY_API_KEY", "")
    os.environ["VERCEL_OIDC_TOKEN"] = oidc_token or os.getenv("VERCEL_OIDC_TOKEN", "")

    if model:
        try:
            formatted_model = f"litellm/vercel_ai_gateway/{model}"
            return Agent(**base_kwargs, model=formatted_model)
        except TypeError:
            return Agent(**base_kwargs)
    return Agent(**base_kwargs)


async def run_agent_flow(