yumoqing
1c4b158a3f
- agent_chat.dspy: 提取前端 model_id 传给 gateway.run_message（校验+持久化到项目） - agent_model_options.dspy: selected 标记=当前项目已设模型（按 session_id 解析， 无项目回退个人全局）——下拉每次打开都恢复项目当前模型，不回退首项 - agent/index.ui: 模型下拉带 session_id（多 tab 各自项目，下拉解析对的项目）
57 lines
2.4 KiB
Plaintext
57 lines
2.4 KiB
Plaintext
# agent_model_options.dspy - 返回 active 模型列表（供 AgentIO 模型选择下拉）
|
||
# llm 表是 pipeline 库的通用配置，产线无关
|
||
# org_id 多租户隔离：非系统级机构只看到本机构的模型（不含系统级兜底）
|
||
# selected 标记 = 当前项目的已设模型（sd_projects.default_model，产线通用、跨会话持久）；
|
||
# 项目未设模型时回退个人全局选择（default_llm_id）。前端 UiCode 按 selected 恢复选中项，
|
||
# 不再每次重建都回退第一项。
|
||
|
||
dbname = get_module_dbname('pipeline_core')
|
||
uid = await get_user()
|
||
org_id = (await get_userorgid()) or ''
|
||
session_id = (params_kw or {}).get('session_id', '') or ''
|
||
|
||
async with DBPools().sqlorContext(dbname) as sor:
|
||
sql = "SELECT id, name, provider, model_id, capabilities FROM llm WHERE status='active'"
|
||
params = {}
|
||
if org_id and org_id != '0':
|
||
sql += " AND org_id=${org}$"
|
||
params['org'] = org_id
|
||
sql += " ORDER BY name"
|
||
recs = await sor.sqlExe(sql, params)
|
||
|
||
# 项目级模型（优先级最高）：按会话解析当前项目 → sd_projects.default_model。
|
||
# session_id 为空/无会话记录时自动回退全局设置，与消息链路的项目解析同一函数，不漂移。
|
||
project_model = ''
|
||
try:
|
||
from pipeline_service.workspace import get_session_project_id
|
||
_pid = await get_session_project_id(sor, uid or '', session_id)
|
||
if _pid:
|
||
_p = await sor.sqlExe(
|
||
"SELECT default_model FROM sd_projects WHERE id=${p}$", {"p": _pid})
|
||
await sor.sqlExe("COMMIT", {})
|
||
if _p:
|
||
project_model = getattr(_p[0], 'default_model', '') or ''
|
||
except Exception as e:
|
||
debug(f'agent_model_options project_model error: {e}')
|
||
|
||
# 个人之前选择的默认模型（项目未设模型时的回退选中项）
|
||
current_llm_id = ''
|
||
if uid:
|
||
_s = await sor.sqlExe(
|
||
"SELECT default_llm_id FROM pipeline_agent_settings WHERE user_id=${u}$", {"u": uid})
|
||
if _s:
|
||
current_llm_id = getattr(_s[0], 'default_llm_id', '') or ''
|
||
|
||
rows = []
|
||
for r in recs:
|
||
rows.append({
|
||
'value': r.id,
|
||
'text': f"{r.name} ({r.provider})",
|
||
'provider': r.provider,
|
||
'model_id': r.model_id,
|
||
'capabilities': r.capabilities or 'text',
|
||
'selected': (r.name == project_model) if project_model else (r.id == current_llm_id),
|
||
})
|
||
|
||
return rows
|