self-evolve-framework 1.4.0 → 1.6.0

package/template/skills/skillopt-sleep/scripts/framework/skillopt/model/__init__.py

@@ -0,0 +1,514 @@
+"""ReflACT model API with runtime backend selection for the target path."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from skillopt.model import azure_openai as _openai
+from skillopt.model import claude_backend as _claude
+from skillopt.model import minimax_backend as _minimax
+from skillopt.model import qwen_backend as _qwen
+from skillopt.model.backend_config import (  # noqa: F401
+    configure_claude_code_exec,
+    configure_codex_exec,
+    get_claude_code_exec_config,
+    get_codex_exec_config,
+    get_target_backend,
+    get_optimizer_backend,
+    is_target_chat_backend,
+    is_target_exec_backend,
+    is_optimizer_chat_backend,
+    set_target_backend,
+    set_optimizer_backend,
+)
+
+
+def set_backend(name: str | None) -> str:
+    """Backward-compatible global backend setter.
+
+    Historically the codebase used one shared backend for both optimizer and
+    target. Keep that entry point so older scripts continue to work, while
+    mapping it onto the split optimizer/target backend model.
+    """
+    normalized = str(name or "azure_openai").strip().lower()
+    if normalized in {"azure_openai", "openai_chat", "azure", "azure-openai"}:
+        set_optimizer_backend("openai_chat")
+        set_target_backend("openai_chat")
+        return "azure_openai"
+    if normalized in {"claude", "claude_chat", "anthropic"}:
+        set_optimizer_backend("claude_chat")
+        set_target_backend("claude_chat")
+        return "claude_chat"
+    if normalized == "codex":
+        set_optimizer_backend("openai_chat")
+        set_target_backend("codex_exec")
+        return "codex"
+    if normalized in {"codex_exec", "claude_code_exec"}:
+        set_optimizer_backend("openai_chat")
+        set_target_backend(normalized)
+        return normalized
+    if normalized in {"qwen", "qwen_chat"}:
+        set_optimizer_backend("openai_chat")
+        set_target_backend("qwen_chat")
+        return "qwen_chat"
+    if normalized in {"minimax", "minimax_chat"}:
+        set_optimizer_backend("openai_chat")
+        set_target_backend("minimax_chat")
+        return "minimax_chat"
+    raise ValueError(f"Unsupported legacy backend: {name!r}")
+
+
+def get_backend_name() -> str:
+    """Best-effort backward-compatible backend summary."""
+    optimizer = get_optimizer_backend()
+    target = get_target_backend()
+    if optimizer == "claude_chat" and target == "claude_chat":
+        return "claude_chat"
+    if optimizer == "qwen_chat" and target == "qwen_chat":
+        return "qwen_chat"
+    if optimizer == "openai_chat" and target == "openai_chat":
+        return "azure_openai"
+    if optimizer == "openai_chat" and target == "codex_exec":
+        return "codex"
+    if optimizer == "openai_chat" and target == "qwen_chat":
+        return "qwen_chat"
+    if optimizer == "openai_chat" and target == "minimax_chat":
+        return "minimax_chat"
+    return f"{optimizer}+{target}"
+
+
+def chat_optimizer(
+    system: str,
+    user: str,
+    max_completion_tokens: int = 16384,
+    retries: int = 5,
+    stage: str = "optimizer",
+    reasoning_effort: str | None = None,
+    timeout: int | None = None,
+) -> tuple[str, dict]:
+    if get_optimizer_backend() == "claude_chat":
+        return _claude.chat_optimizer(
+            system=system,
+            user=user,
+            max_completion_tokens=max_completion_tokens,
+            retries=retries,
+            stage=stage,
+            timeout=timeout,
+        )
+    if get_optimizer_backend() == "qwen_chat":
+        return _qwen.chat_optimizer(
+            system=system,
+            user=user,
+            max_completion_tokens=max_completion_tokens,
+            retries=retries,
+            stage=stage,
+            reasoning_effort=reasoning_effort,
+            timeout=timeout,
+        )
+    return _openai.chat_optimizer(
+        system=system,
+        user=user,
+        max_completion_tokens=max_completion_tokens,
+        retries=retries,
+        stage=stage,
+        reasoning_effort=reasoning_effort,
+        timeout=timeout,
+    )
+
+
+def chat_target(
+    system: str,
+    user: str,
+    max_completion_tokens: int = 16384,
+    retries: int = 5,
+    stage: str = "target",
+    reasoning_effort: str | None = None,
+    timeout: int | None = None,
+) -> tuple[str, dict]:
+    if get_target_backend() == "claude_chat":
+        return _claude.chat_target(
+            system=system,
+            user=user,
+            max_completion_tokens=max_completion_tokens,
+            retries=retries,
+            stage=stage,
+            timeout=timeout,
+        )
+    if get_target_backend() == "qwen_chat":
+        return _qwen.chat_target(
+            system=system,
+            user=user,
+            max_completion_tokens=max_completion_tokens,
+            retries=retries,
+            stage=stage,
+            reasoning_effort=reasoning_effort,
+            timeout=timeout,
+        )
+    if get_target_backend() == "minimax_chat":
+        return _minimax.chat_target(
+            system=system,
+            user=user,
+            max_completion_tokens=max_completion_tokens,
+            retries=retries,
+            stage=stage,
+            reasoning_effort=reasoning_effort,
+        )
+    if not is_target_chat_backend():
+        raise NotImplementedError(
+            "chat_target is only supported with target_backend=openai_chat, claude_chat, qwen_chat, or minimax_chat. "
+            "Exec backends are handled in environment-specific rollout code."
+        )
+    return _openai.chat_target(
+        system=system,
+        user=user,
+        max_completion_tokens=max_completion_tokens,
+        retries=retries,
+        stage=stage,
+        reasoning_effort=reasoning_effort,
+        timeout=timeout,
+    )
+
+
+def chat_optimizer_messages(
+    messages: list[dict[str, Any]],
+    max_completion_tokens: int = 16384,
+    retries: int = 5,
+    stage: str = "optimizer",
+    reasoning_effort: str | None = None,
+    *,
+    tools: list[dict[str, Any]] | None = None,
+    tool_choice: str | dict[str, Any] | None = None,
+    return_message: bool = False,
+    timeout: int | None = None,
+) -> tuple[Any, dict]:
+    if get_optimizer_backend() == "claude_chat":
+        return _claude.chat_optimizer_messages(
+            messages=messages,
+            max_completion_tokens=max_completion_tokens,
+            retries=retries,
+            stage=stage,
+            tools=tools,
+            tool_choice=tool_choice,
+            return_message=return_message,
+            timeout=timeout,
+        )
+    if get_optimizer_backend() == "qwen_chat":
+        return _qwen.chat_optimizer_messages(
+            messages=messages,
+            max_completion_tokens=max_completion_tokens,
+            retries=retries,
+            stage=stage,
+            reasoning_effort=reasoning_effort,
+            tools=tools,
+            tool_choice=tool_choice,
+            return_message=return_message,
+            timeout=timeout,
+        )
+    return _openai.chat_optimizer_messages(
+        messages=messages,
+        max_completion_tokens=max_completion_tokens,
+        retries=retries,
+        stage=stage,
+        reasoning_effort=reasoning_effort,
+        tools=tools,
+        tool_choice=tool_choice,
+        return_message=return_message,
+        timeout=timeout,
+    )
+
+
+def chat_target_messages(
+    messages: list[dict[str, Any]],
+    max_completion_tokens: int = 16384,
+    retries: int = 5,
+    stage: str = "target",
+    reasoning_effort: str | None = None,
+    *,
+    tools: list[dict[str, Any]] | None = None,
+    tool_choice: str | dict[str, Any] | None = None,
+    return_message: bool = False,
+    timeout: int | None = None,
+) -> tuple[Any, dict]:
+    if get_target_backend() == "claude_chat":
+        return _claude.chat_target_messages(
+            messages=messages,
+            max_completion_tokens=max_completion_tokens,
+            retries=retries,
+            stage=stage,
+            tools=tools,
+            tool_choice=tool_choice,
+            return_message=return_message,
+            timeout=timeout,
+        )
+    if get_target_backend() == "qwen_chat":
+        return _qwen.chat_target_messages(
+            messages=messages,
+            max_completion_tokens=max_completion_tokens,
+            retries=retries,
+            stage=stage,
+            reasoning_effort=reasoning_effort,
+            tools=tools,
+            tool_choice=tool_choice,
+            return_message=return_message,
+            timeout=timeout,
+        )
+    if get_target_backend() == "minimax_chat":
+        return _minimax.chat_target_messages(
+            messages=messages,
+            max_completion_tokens=max_completion_tokens,
+            retries=retries,
+            stage=stage,
+            reasoning_effort=reasoning_effort,
+            tools=tools,
+            tool_choice=tool_choice,
+            return_message=return_message,
+        )
+    if not is_target_chat_backend():
+        raise NotImplementedError(
+            "chat_target_messages is only supported with target_backend=openai_chat, claude_chat, qwen_chat, or minimax_chat. "
+            "Exec backends are handled in environment-specific rollout code."
+        )
+    return _openai.chat_target_messages(
+        messages=messages,
+        max_completion_tokens=max_completion_tokens,
+        retries=retries,
+        stage=stage,
+        reasoning_effort=reasoning_effort,
+        tools=tools,
+        tool_choice=tool_choice,
+        return_message=return_message,
+        timeout=timeout,
+    )
+
+
+def chat_messages_with_deployment(
+    deployment: str,
+    messages: list[dict[str, Any]],
+    max_completion_tokens: int = 16384,
+    retries: int = 5,
+    stage: str = "custom",
+    reasoning_effort: str | None = None,
+    *,
+    tools: list[dict[str, Any]] | None = None,
+    tool_choice: str | dict[str, Any] | None = None,
+    return_message: bool = False,
+    timeout: int | None = None,
+) -> tuple[Any, dict]:
+    return _openai.chat_messages_with_deployment(
+        deployment=deployment,
+        messages=messages,
+        max_completion_tokens=max_completion_tokens,
+        retries=retries,
+        stage=stage,
+        reasoning_effort=reasoning_effort,
+        tools=tools,
+        tool_choice=tool_choice,
+        return_message=return_message,
+        timeout=timeout,
+    )
+
+
+def chat_with_deployment(
+    deployment: str,
+    system: str,
+    user: str,
+    max_completion_tokens: int = 16384,
+    retries: int = 5,
+    stage: str = "custom",
+    reasoning_effort: str | None = None,
+    timeout: int | None = None,
+) -> tuple[str, dict]:
+    return _openai.chat_with_deployment(
+        deployment=deployment,
+        system=system,
+        user=user,
+        max_completion_tokens=max_completion_tokens,
+        retries=retries,
+        stage=stage,
+        reasoning_effort=reasoning_effort,
+        timeout=timeout,
+    )
+
+
+def get_token_summary() -> dict:
+    summary = _openai.get_token_summary()
+    claude_summary = _claude.get_token_summary()
+    for stage, values in claude_summary.items():
+        if stage == "_total":
+            continue
+        if stage not in summary:
+            summary[stage] = values
+            continue
+        summary[stage]["calls"] += values["calls"]
+        summary[stage]["prompt_tokens"] += values["prompt_tokens"]
+        summary[stage]["completion_tokens"] += values["completion_tokens"]
+        summary[stage]["total_tokens"] += values["total_tokens"]
+    qwen_summary = _qwen.get_token_summary()
+    for stage, values in qwen_summary.items():
+        if stage == "_total":
+            continue
+        if stage not in summary:
+            summary[stage] = values
+            continue
+        summary[stage]["calls"] += values["calls"]
+        summary[stage]["prompt_tokens"] += values["prompt_tokens"]
+        summary[stage]["completion_tokens"] += values["completion_tokens"]
+        summary[stage]["total_tokens"] += values["total_tokens"]
+    minimax_summary = _minimax.get_token_summary()
+    for stage, values in minimax_summary.items():
+        if stage == "_total":
+            continue
+        if stage not in summary:
+            summary[stage] = values
+            continue
+        summary[stage]["calls"] += values["calls"]
+        summary[stage]["prompt_tokens"] += values["prompt_tokens"]
+        summary[stage]["completion_tokens"] += values["completion_tokens"]
+        summary[stage]["total_tokens"] += values["total_tokens"]
+    total = {
+        "calls": 0,
+        "prompt_tokens": 0,
+        "completion_tokens": 0,
+        "total_tokens": 0,
+    }
+    for stage, values in summary.items():
+        if stage == "_total":
+            continue
+        total["calls"] += values["calls"]
+        total["prompt_tokens"] += values["prompt_tokens"]
+        total["completion_tokens"] += values["completion_tokens"]
+        total["total_tokens"] += values["total_tokens"]
+    summary["_total"] = total
+    return summary
+
+
+def reset_token_tracker() -> None:
+    _openai.reset_token_tracker()
+    _claude.reset_token_tracker()
+    _qwen.reset_token_tracker()
+    _minimax.reset_token_tracker()
+
+
+def configure_azure_openai(
+    *,
+    endpoint: str | None = None,
+    api_version: str | None = None,
+    api_key: str | None = None,
+    auth_mode: str | None = None,
+    ad_scope: str | None = None,
+    managed_identity_client_id: str | None = None,
+    optimizer_endpoint: str | None = None,
+    optimizer_api_version: str | None = None,
+    optimizer_api_key: str | None = None,
+    optimizer_auth_mode: str | None = None,
+    optimizer_ad_scope: str | None = None,
+    optimizer_managed_identity_client_id: str | None = None,
+    target_endpoint: str | None = None,
+    target_api_version: str | None = None,
+    target_api_key: str | None = None,
+    target_auth_mode: str | None = None,
+    target_ad_scope: str | None = None,
+    target_managed_identity_client_id: str | None = None,
+) -> None:
+    _openai.configure_azure_openai(
+        endpoint=endpoint,
+        api_version=api_version,
+        api_key=api_key,
+        auth_mode=auth_mode,
+        ad_scope=ad_scope,
+        managed_identity_client_id=managed_identity_client_id,
+        optimizer_endpoint=optimizer_endpoint,
+        optimizer_api_version=optimizer_api_version,
+        optimizer_api_key=optimizer_api_key,
+        optimizer_auth_mode=optimizer_auth_mode,
+        optimizer_ad_scope=optimizer_ad_scope,
+        optimizer_managed_identity_client_id=optimizer_managed_identity_client_id,
+        target_endpoint=target_endpoint,
+        target_api_version=target_api_version,
+        target_api_key=target_api_key,
+        target_auth_mode=target_auth_mode,
+        target_ad_scope=target_ad_scope,
+        target_managed_identity_client_id=target_managed_identity_client_id,
+    )
+
+
+def configure_qwen_chat(
+    *,
+    base_url: str | None = None,
+    api_key: str | None = None,
+    temperature: float | str | None = None,
+    timeout_seconds: float | str | None = None,
+    max_tokens: int | str | None = None,
+    enable_thinking: bool | str | None = None,
+    optimizer_base_url: str | None = None,
+    optimizer_api_key: str | None = None,
+    optimizer_temperature: float | str | None = None,
+    optimizer_timeout_seconds: float | str | None = None,
+    optimizer_max_tokens: int | str | None = None,
+    optimizer_enable_thinking: bool | str | None = None,
+    target_base_url: str | None = None,
+    target_api_key: str | None = None,
+    target_temperature: float | str | None = None,
+    target_timeout_seconds: float | str | None = None,
+    target_max_tokens: int | str | None = None,
+    target_enable_thinking: bool | str | None = None,
+) -> None:
+    _qwen.configure_qwen_chat(
+        base_url=base_url,
+        api_key=api_key,
+        temperature=temperature,
+        timeout_seconds=timeout_seconds,
+        max_tokens=max_tokens,
+        enable_thinking=enable_thinking,
+        optimizer_base_url=optimizer_base_url,
+        optimizer_api_key=optimizer_api_key,
+        optimizer_temperature=optimizer_temperature,
+        optimizer_timeout_seconds=optimizer_timeout_seconds,
+        optimizer_max_tokens=optimizer_max_tokens,
+        optimizer_enable_thinking=optimizer_enable_thinking,
+        target_base_url=target_base_url,
+        target_api_key=target_api_key,
+        target_temperature=target_temperature,
+        target_timeout_seconds=target_timeout_seconds,
+        target_max_tokens=target_max_tokens,
+        target_enable_thinking=target_enable_thinking,
+    )
+
+
+def configure_minimax_chat(
+    *,
+    base_url: str | None = None,
+    api_key: str | None = None,
+    temperature: float | str | None = None,
+    timeout_seconds: float | str | None = None,
+    max_tokens: int | str | None = None,
+    enable_thinking: bool | str | None = None,
+) -> None:
+    _minimax.configure_minimax_chat(
+        base_url=base_url,
+        api_key=api_key,
+        temperature=temperature,
+        timeout_seconds=timeout_seconds,
+        max_tokens=max_tokens,
+        enable_thinking=enable_thinking,
+    )
+
+
+def set_reasoning_effort(effort: str | None) -> None:
+    _openai.set_reasoning_effort(effort)
+    _claude.set_reasoning_effort(effort)
+    _qwen.set_reasoning_effort(effort)
+    _minimax.set_reasoning_effort(effort)
+
+
+def set_target_deployment(deployment: str) -> None:
+    _openai.set_target_deployment(deployment)
+    _claude.set_target_deployment(deployment)
+    _qwen.set_target_deployment(deployment)
+    _minimax.set_target_deployment(deployment)
+
+
+def set_optimizer_deployment(deployment: str) -> None:
+    _openai.set_optimizer_deployment(deployment)
+    _claude.set_optimizer_deployment(deployment)
+    _qwen.set_optimizer_deployment(deployment)