Spaces:
Running
Running
File size: 3,029 Bytes
57d4e72 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 | 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,
}
|