40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
|
"""API: Save model routing configuration."""
|
||
|
|
from helpers.api import ApiHandler, Request, Response
|
||
|
|
import yaml
|
||
|
|
import os
|
||
|
|
|
||
|
|
class ModelRoutingSave(ApiHandler):
|
||
|
|
async def process(self, input: dict, request: Request) -> dict | Response:
|
||
|
|
role = input.get("role", "")
|
||
|
|
preset = input.get("preset", "")
|
||
|
|
project_name = input.get("project_name", "")
|
||
|
|
agent_profile = input.get("agent_profile", "")
|
||
|
|
|
||
|
|
if not role or not preset:
|
||
|
|
return {"ok": False, "error": "role and preset are required"}
|
||
|
|
|
||
|
|
# Update default_config.yaml or per-project config
|
||
|
|
plugin_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
|
|
config_path = os.path.join(plugin_dir, "default_config.yaml")
|
||
|
|
|
||
|
|
try:
|
||
|
|
with open(config_path, "r") as f:
|
||
|
|
config = yaml.safe_load(f) or {}
|
||
|
|
except Exception:
|
||
|
|
config = {}
|
||
|
|
|
||
|
|
routing = config.get("model_routing", {})
|
||
|
|
roles = routing.get("roles", {})
|
||
|
|
if role not in roles:
|
||
|
|
roles[role] = {}
|
||
|
|
roles[role]["preset"] = preset
|
||
|
|
routing["roles"] = roles
|
||
|
|
config["model_routing"] = routing
|
||
|
|
|
||
|
|
try:
|
||
|
|
with open(config_path, "w") as f:
|
||
|
|
yaml.dump(config, f, default_flow_style=False, allow_unicode=True)
|
||
|
|
return {"ok": True, "role": role, "preset": preset}
|
||
|
|
except Exception as e:
|
||
|
|
return {"ok": False, "error": str(e)}
|