yumoqing
a72637b2a3
- Updated core modules: __init__, ai_config, ai_core, attachment, contract, enhanced_contract - Updated mysql.ddl.sql - Updated UI files: contract_detail, contract_edit, contract_list - Added new files: api/*.dspy, contract_list.dspy
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
import os
|
|
import json
|
|
from sqlor.dbpools import DBPools
|
|
|
|
async def save_ai_config(request_data: dict, org_id: str) -> dict:
|
|
"""保存AI配置"""
|
|
dbp = await getDBP(org_id)
|
|
|
|
# 检查是否已存在配置
|
|
check_sql = "SELECT id FROM contract_ai_config WHERE org_id = %(org_id)s"
|
|
check_result = await sor.doQuery(check_sql, {'org_id': org_id})
|
|
|
|
if check_result:
|
|
# 更新现有配置
|
|
sql = """
|
|
UPDATE contract_ai_config
|
|
SET ai_service_url = %(ai_service_url)s, api_key = %(api_key)s, updated_at = NOW()
|
|
WHERE org_id = %(org_id)s
|
|
"""
|
|
else:
|
|
# 插入新配置
|
|
sql = """
|
|
INSERT INTO contract_ai_config (id, ai_service_url, api_key, org_id, created_at, updated_at)
|
|
VALUES (%(id)s, %(ai_service_url)s, %(api_key)s, %(org_id)s, NOW(), NOW())
|
|
"""
|
|
request_data['id'] = str(uuid.uuid4()).replace('-', '')
|
|
|
|
request_data['org_id'] = org_id
|
|
await sor.doTransaction([{'sql': sql, 'params': request_data}])
|
|
return {'success': True, 'message': 'AI配置保存成功'}
|
|
|
|
# 在 init.py 中添加这个函数的暴露 |