yumoqing
6890af7638
删除已在 ahserver/ServerEnv 预加载的模块: json, datetime, getID, debug, curDateString, timestampstr, get_sor_context, time, DictObject, partial, FileStorage, os 删除模块内部 import: from llmage.utils import get_llmusage_by_id, read_ioinfo_content 同步将 get_llmusage_by_id, read_ioinfo_content 加入 load_llmage() 导出, dspy 中通过 ServerEnv 全局调用。
63 lines
2.7 KiB
Python
63 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
|
|
result = {'widgettype': 'Message', 'options': {'title': 'Error', 'message': 'Invalid', 'type': 'error'}}
|
|
|
|
try:
|
|
dbname = get_module_dbname('llmage')
|
|
llmid = params_kw.get('llmid', '')
|
|
catelogid = params_kw.get('llmcatelogid', '')
|
|
apiname = params_kw.get('apiname', '')
|
|
|
|
if not llmid or not catelogid or not apiname:
|
|
result['options'] = {'title': 'Error', 'message': '模型、分类和API接口为必填项', 'type': 'error'}
|
|
else:
|
|
user_orgid = await get_userorgid()
|
|
from appPublic.uniqueID import getID
|
|
new_id = getID()
|
|
|
|
async with DBPools().sqlorContext(dbname) as sor:
|
|
# 验证模型属于当前用户的机构
|
|
check_llm = await sor.sqlExe(
|
|
"select id from llm where id = ${llmid}$ and ownerid = ${ownerid}$",
|
|
{'llmid': llmid, 'ownerid': user_orgid}
|
|
)
|
|
if not check_llm:
|
|
result['options'] = {'title': 'Error', 'message': '无权操作该模型', 'type': 'error'}
|
|
else:
|
|
# 检查是否已存在
|
|
check_sql = """select id from llm_api_map
|
|
where llmid = ${llmid}$ and apiname = ${apiname}$"""
|
|
exists = await sor.sqlExe(check_sql, {'llmid': llmid, 'apiname': apiname})
|
|
|
|
if exists:
|
|
result['options'] = {'title': '提示', 'message': '该模型的此API映射已存在', 'type': 'warning'}
|
|
else:
|
|
data = {
|
|
'id': new_id,
|
|
'llmid': llmid,
|
|
'llmcatelogid': catelogid,
|
|
'apiname': apiname
|
|
}
|
|
query_apiname = params_kw.get('query_apiname', '').strip()
|
|
if query_apiname:
|
|
data['query_apiname'] = query_apiname
|
|
query_period = params_kw.get('query_period', '').strip()
|
|
if query_period:
|
|
try:
|
|
data['query_period'] = int(query_period)
|
|
except ValueError:
|
|
data['query_period'] = 30
|
|
else:
|
|
data['query_period'] = 30
|
|
ppid = params_kw.get('ppid', '').strip()
|
|
if ppid:
|
|
data['ppid'] = ppid
|
|
|
|
await sor.C('llm_api_map', data)
|
|
result['options'] = {'title': 'Success', 'message': '添加成功', 'type': 'success'}
|
|
|
|
except Exception as e:
|
|
result['options'] = {'title': 'Error', 'message': f'添加失败: {str(e)}', 'type': 'error'}
|
|
|
|
return json.dumps(result, ensure_ascii=False)
|