mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-05-19 07:59:34 +00:00
7c59ac9e57
Expose sanitized active main and utility model metadata through the model override endpoint, then render those names in the chat model switcher even when no preset override is active. Keep the inline model names hidden on narrow screens and cover the behavior with a regression check. Refresh model names after settings save Refresh the active chat model switcher after _model_config settings are saved so changes to main and utility models appear immediately. Extend the model switcher regression check to cover the save-refresh hook.
88 lines
3.4 KiB
Python
88 lines
3.4 KiB
Python
from helpers.api import ApiHandler, Request, Response
|
|
from helpers.persist_chat import save_tmp_chat
|
|
from agent import AgentContext
|
|
from plugins._model_config.helpers import model_config
|
|
|
|
|
|
def _public_model_config(config: dict) -> dict | None:
|
|
if not isinstance(config, dict):
|
|
return None
|
|
provider = str(config.get("provider", "") or "").strip()
|
|
name = str(config.get("name", "") or "").strip()
|
|
if not provider and not name:
|
|
return None
|
|
return {"provider": provider, "name": name}
|
|
|
|
|
|
def _active_models(ctx: AgentContext) -> dict:
|
|
return {
|
|
"main": _public_model_config(model_config.get_chat_model_config(ctx.agent0)),
|
|
"utility": _public_model_config(model_config.get_utility_model_config(ctx.agent0)),
|
|
}
|
|
|
|
|
|
class ModelOverride(ApiHandler):
|
|
async def process(self, input: dict, request: Request) -> dict | Response:
|
|
context_id = input.get("context_id", "")
|
|
action = input.get("action", "get") # get | set | set_preset | clear
|
|
|
|
if not context_id:
|
|
return Response(status=400, response="Missing context_id")
|
|
|
|
ctx = AgentContext.get(context_id)
|
|
if not ctx:
|
|
return Response(status=404, response="Context not found")
|
|
|
|
if action == "get":
|
|
override = ctx.get_data("chat_model_override")
|
|
allowed = model_config.is_chat_override_allowed(ctx.agent0)
|
|
return {
|
|
"override": override,
|
|
"allowed": allowed,
|
|
"active_models": _active_models(ctx),
|
|
}
|
|
|
|
elif action == "set":
|
|
if not model_config.is_chat_override_allowed(ctx.agent0):
|
|
return Response(status=403, response="Per-chat override is disabled")
|
|
override_config = input.get("override")
|
|
if not override_config or not isinstance(override_config, dict):
|
|
return Response(status=400, response="Missing or invalid override config")
|
|
ctx.set_data("chat_model_override", override_config)
|
|
save_tmp_chat(ctx)
|
|
return {
|
|
"ok": True,
|
|
"override": override_config,
|
|
"active_models": _active_models(ctx),
|
|
}
|
|
|
|
elif action == "set_preset":
|
|
if not model_config.is_chat_override_allowed(ctx.agent0):
|
|
return Response(status=403, response="Per-chat override is disabled")
|
|
preset_name = input.get("preset_name", "")
|
|
if not preset_name:
|
|
return Response(status=400, response="Missing preset_name")
|
|
# Verify preset exists
|
|
preset = model_config.get_preset_by_name(preset_name)
|
|
if not preset:
|
|
return Response(status=404, response=f"Preset '{preset_name}' not found")
|
|
# Store as a preset reference
|
|
override_value = {"preset_name": preset_name}
|
|
ctx.set_data("chat_model_override", override_value)
|
|
save_tmp_chat(ctx)
|
|
return {
|
|
"ok": True,
|
|
"preset_name": preset_name,
|
|
"active_models": _active_models(ctx),
|
|
}
|
|
|
|
elif action == "clear":
|
|
ctx.set_data("chat_model_override", None)
|
|
save_tmp_chat(ctx)
|
|
return {
|
|
"ok": True,
|
|
"override": None,
|
|
"active_models": _active_models(ctx),
|
|
}
|
|
|
|
return Response(status=400, response=f"Unknown action: {action}")
|