BBIT-Kai
93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
from typing import List, Dict
|
|
|
|
from config.pgDb import pg_pool
|
|
|
|
|
|
def get_ai_personality(ai_id: str):
|
|
with pg_pool.getConn() as conn:
|
|
with conn.cursor() as cur:
|
|
cur.execute(
|
|
"SELECT ai_personality FROM ai_chat_profiles WHERE id = %s", (ai_id,)
|
|
)
|
|
row = cur.fetchone()
|
|
if row:
|
|
return row[0]
|
|
else:
|
|
return "你是一个乐于助人的AI助手，请保持中文简洁回答用户。"
|
|
|
|
|
|
def get_description(ai_id: str):
|
|
with pg_pool.getConn() as conn:
|
|
with conn.cursor() as cur:
|
|
cur.execute(
|
|
"SELECT description FROM ai_chat_profiles WHERE id = %s", (ai_id,)
|
|
)
|
|
row = cur.fetchone()
|
|
if row:
|
|
return row[0]
|
|
else:
|
|
return "你是一个乐于助人的AI助手，请保持中文简洁回答用户。"
|
|
|
|
|
|
def get_ai_available_kn_bases(ai_id: str) -> List[str]:
|
|
with pg_pool.getConn() as conn:
|
|
result = conn.execute(
|
|
"SELECT available_kn_bases FROM ai_chat_profiles WHERE id = %s", (ai_id,)
|
|
)
|
|
return result.fetchone()[0]
|
|
|
|
|
|
def get_all_ai_bot(user_id: str, module: str) -> List[Dict]:
|
|
with pg_pool.getConn() as conn:
|
|
with conn.cursor() as cur:
|
|
# 查询用户角色
|
|
cur.execute("SELECT roles FROM sys_users WHERE id = %s", (user_id,))
|
|
role_row = cur.fetchone()
|
|
if not role_row:
|
|
return [] # 用户不存在
|
|
user_roles = role_row[0]
|
|
|
|
# 查询 AI 角色 JSON 字段包含用户角色
|
|
|
|
cur.execute(
|
|
"""
|
|
SELECT id, title, description, welcome_words, ai_personality, available_report_tables, available_kn_bases
|
|
FROM ai_chat_profiles
|
|
WHERE available_module = %s
|
|
AND is_active = TRUE
|
|
AND available_roles::jsonb ?| %s
|
|
""",
|
|
(module, user_roles),
|
|
)
|
|
rows = cur.fetchall()
|
|
result = []
|
|
for row in rows:
|
|
# row 索引对应 SELECT 字段顺序
|
|
(
|
|
id_,
|
|
title,
|
|
description,
|
|
welcome_words,
|
|
ai_personality,
|
|
available_report_tables,
|
|
available_kn_bases,
|
|
) = row
|
|
|
|
# 解析 JSON
|
|
roles_json = ai_personality if ai_personality else {}
|
|
result.append(
|
|
{
|
|
"id": id_,
|
|
"title": title,
|
|
"description": description,
|
|
"welcome_words": welcome_words,
|
|
"name": roles_json.get("名字", ""),
|
|
"role": roles_json.get("性格", ""),
|
|
"service": roles_json.get("业务", ""),
|
|
"available_report_tables": available_report_tables,
|
|
"available_kn_bases": available_kn_bases,
|
|
}
|
|
)
|
|
|
|
return result
|