mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-08-11 01:16:25 +00:00
15a2b553be
Make model presets a single global collection and treat presets as reusable setups (main, utility, embedding). Persisted scope configs now store only a preset name (Default is always present and immutable); project-scoped preset definitions were removed and attempts to edit/reset project presets are rejected. Add embedding-model support throughout: include embedding slot in presets, surface embedding in the switcher and override APIs, and notify extensions when the effective embedding model changes. Implement rename/retire propagation to update plugin configs, saved chats, and live AgentContext objects when preset names change or presets are removed. Update CLI/integration commands to use an "inherit" action for returning to the scoped preset and to report the effective scoped-or-chat preset. Add startup migration/bootstrap to initialize presets from remote or bundled fallback and numerous docs/UI text updates to reflect the new preset model. Extra: refactor config resolution helpers and add safety checks (validation, atomic writes) when saving presets.
69 lines
3 KiB
Python
69 lines
3 KiB
Python
from helpers.api import ApiHandler, Request, Response
|
|
from helpers.providers import get_raw_providers
|
|
from plugins._model_config.helpers import model_config
|
|
from agent import AgentContext
|
|
import models
|
|
|
|
|
|
class ModelConfigGet(ApiHandler):
|
|
async def process(self, input: dict, request: Request) -> dict | Response:
|
|
project_name = input.get("project_name", "")
|
|
agent_profile = input.get("agent_profile", "")
|
|
context_id = str(input.get("context_id") or "").strip()
|
|
context = AgentContext.get(context_id) if context_id else None
|
|
agent = context.agent0 if context else None
|
|
|
|
if agent:
|
|
config = model_config.get_effective_config(agent)
|
|
configured_preset = model_config.get_configured_preset_name(agent=agent)
|
|
selected_preset = model_config.get_effective_preset_name(agent)
|
|
else:
|
|
config = model_config.get_config(
|
|
project_name=project_name or None,
|
|
agent_profile=agent_profile or None,
|
|
)
|
|
configured_preset = model_config.get_configured_preset_name(
|
|
project_name=project_name or None,
|
|
agent_profile=agent_profile or None,
|
|
)
|
|
selected_preset = str(
|
|
config.get(model_config.MODEL_PRESET_CONFIG_KEY)
|
|
or model_config.DEFAULT_PRESET_NAME
|
|
)
|
|
|
|
# Add provider lists for UI dropdowns
|
|
chat_providers = model_config.get_chat_providers()
|
|
embedding_providers = model_config.get_embedding_providers()
|
|
chat_provider_details = get_raw_providers("chat")
|
|
embedding_provider_details = get_raw_providers("embedding")
|
|
|
|
# Mask API keys - show status only
|
|
api_key_status = {}
|
|
all_providers = chat_providers + embedding_providers
|
|
seen = set()
|
|
for p in all_providers:
|
|
pid = p.get("value", "")
|
|
if pid and pid not in seen:
|
|
seen.add(pid)
|
|
key = models.get_api_key(pid)
|
|
api_key_status[pid] = bool(key and key.strip() and key != "None")
|
|
|
|
chat_model = config.get("chat_model", {}) if isinstance(config, dict) else {}
|
|
chat_provider = str(chat_model.get("provider") or "").strip()
|
|
chat_name = str(chat_model.get("name") or "").strip()
|
|
|
|
return {
|
|
"config": config,
|
|
"chat_providers": chat_providers,
|
|
"embedding_providers": embedding_providers,
|
|
"chat_provider_details": chat_provider_details,
|
|
"embedding_provider_details": embedding_provider_details,
|
|
"api_key_status": api_key_status,
|
|
"model_configured": model_config.is_chat_model_configured(config),
|
|
"model_configured_label": " / ".join(
|
|
part for part in (chat_provider, chat_name) if part
|
|
),
|
|
"presets": model_config.get_presets(),
|
|
"configured_preset": configured_preset,
|
|
"selected_preset": selected_preset,
|
|
}
|