bbit/AILab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

仪评指标联分析模块

Browse Source
This commit is contained in:
BBIT-Kai
2025-09-24 13:59:00 +08:00
parent 0ab82b00d6
commit d2775f60a7
28 changed files with 1106 additions and 181 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bbit_ai/app/db/postgres.py
+240 -90
View File
@@ -1,8 +1,9 @@
from langchain_postgres import PostgresChatMessageHistory
from config.pgDb import pg_pool
from config.ssDb import mssql_pool
from typing import List, Dict
import json
from typing import List, Dict
from langchain_postgres import PostgresChatMessageHistory
from config.pgDb import pg_pool
# ————————————————————————————————————————————————————AI角色———————————————————————————————
@@ -13,8 +14,7 @@ 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,)
"SELECT ai_personality FROM ai_chat_profiles WHERE id = %s", (ai_id,)
)
row = cur.fetchone()
if row:
@@ -27,8 +27,7 @@ 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,)
"SELECT description FROM ai_chat_profiles WHERE id = %s", (ai_id,)
)
row = cur.fetchone()
if row:
@@ -40,8 +39,7 @@ def get_description(ai_id: str):
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,)
"SELECT available_kn_bases FROM ai_chat_profiles WHERE id = %s", (ai_id,)
)
return result.fetchone()[0]
@@ -50,10 +48,7 @@ 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 users WHERE id = %s",
(user_id,)
)
cur.execute("SELECT roles FROM users WHERE id = %s", (user_id,))
role_row = cur.fetchone()
if not role_row:
return [] # 用户不存在
@@ -69,27 +64,37 @@ def get_all_ai_bot(user_id: str, module: str) -> List[Dict]:
AND is_active = TRUE
AND available_roles::jsonb ?| %s
""",
(module, user_roles)
(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
(
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
})
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
@@ -98,9 +103,7 @@ def get_all_ai_bot(user_id: str, module: str) -> List[Dict]:
def insert_message(session_id: str, isAI: bool, content: str):
with pg_pool.getConn() as conn:
history = PostgresChatMessageHistory(
database_name,
session_id,
sync_connection=conn
database_name, session_id, sync_connection=conn
)
if isAI:
history.add_ai_message(content)
@@ -112,15 +115,10 @@ def get_history(session_id: str):
simplified = []
with pg_pool.getConn() as conn:
history = PostgresChatMessageHistory(
database_name,
session_id,
sync_connection=conn
database_name, session_id, sync_connection=conn
)
for msg in history.messages:
simplified.append({
"type": msg.type,
"content": msg.content
})
simplified.append({"type": msg.type, "content": msg.content})
return simplified
@@ -129,28 +127,33 @@ def get_history_with_time(session_id: str, number: int):
with pg_pool.getConn() as conn:
with conn.cursor() as cur:
cur.execute(
f"SELECT message, created_at FROM ai_chat_history WHERE session_id = '{session_id}' ORDER BY created_at DESC LIMIT {number}")
f"SELECT message, created_at FROM ai_chat_history WHERE session_id = '{session_id}' ORDER BY created_at DESC LIMIT {number}"
)
rows = cur.fetchall()
simplified = []
for row in rows:
msg_dict = row[0]
simplified.append({
"type": msg_dict.get("type"),
"created_at": row[1].isoformat(),
"content": msg_dict.get("data", {}).get("content")
})
simplified.append(
{
"type": msg_dict.get("type"),
"created_at": row[1].isoformat(),
"content": msg_dict.get("data", {}).get("content"),
}
)
return simplified
# ————————————————————————————————————————————————————会话———————————————————————————————
def insert_session(user_id: str, ai_id: str, session_id: str, session_title: str, available_module):
def insert_session(
user_id: str, ai_id: str, session_id: str, session_title: str, available_module
):
with pg_pool.getConn() as coon:
with coon.cursor() as cur:
cur.execute(
"INSERT INTO ai_chat_sessions (id ,user_id, ai_id, title, available_module, created_at, updated_at) VALUES (%s, %s, %s, %s,%s, NOW(), NOW())",
(session_id, user_id, ai_id, session_title, available_module)
(session_id, user_id, ai_id, session_title, available_module),
)
coon.commit()
@@ -160,7 +163,7 @@ def update_session_updated_at(session_id: str):
with conn.cursor() as cur:
cur.execute(
"UPDATE ai_chat_sessions SET updated_at = NOW() WHERE id = %s",
(session_id,)
(session_id,),
)
conn.commit()
@@ -173,16 +176,12 @@ def get_sessions(user_id: str, available_module: str):
"FROM ai_chat_sessions "
"WHERE user_id = %s AND available_module = %s "
"ORDER BY updated_at DESC",
(user_id, available_module)
(user_id, available_module),
)
sessions = cur.fetchall()
return [
{
"id": row[0],
"title": row[1],
"updated_at": row[2]
}
{"id": row[0], "title": row[1], "updated_at": row[2]}
for row in sessions
]
@@ -193,16 +192,10 @@ def get_reports(user_id: str):
with conn.cursor() as cur:
cur.execute(
"SELECT id, title FROM ai_reports WHERE created_by = %s AND is_masked = TRUE ORDER BY created_at DESC",
(user_id,)
(user_id,),
)
reports = cur.fetchall()
return [
{
"id": row[0],
"title": row[1]
}
for row in reports
]
return [{"id": row[0], "title": row[1]} for row in reports]
def save_report(id: str, user_id: str, title: str, sql: str):
@@ -210,7 +203,7 @@ def save_report(id: str, user_id: str, title: str, sql: str):
with conn.cursor() as cur:
cur.execute(
"INSERT INTO ai_reports (id, title, sql, created_at, created_by , is_masked) VALUES (%s, %s, %s, NOW(), %s, FALSE) RETURNING id",
(id, title, sql, user_id)
(id, title, sql, user_id),
)
report_id = cur.fetchone()[0]
conn.commit()
@@ -222,7 +215,7 @@ def maked_report(report_id: str, title: str):
with conn.cursor() as cur:
cur.execute(
"UPDATE ai_reports SET title = %s, is_masked = TRUE WHERE id = %s",
(title, report_id)
(title, report_id),
)
conn.commit()
@@ -230,10 +223,7 @@ def maked_report(report_id: str, title: str):
def getSQL(reportId: str):
with pg_pool.getConn() as conn:
with conn.cursor() as cur:
cur.execute(
"SELECT sql FROM ai_reports WHERE id = %s",
(reportId,)
)
cur.execute("SELECT sql FROM ai_reports WHERE id = %s", (reportId,))
row = cur.fetchone()
if row:
return row[0]
@@ -247,7 +237,7 @@ def get_available_tables_str(aiId: str):
# 1. 先取 AI 可用的数据库表
cur.execute(
"SELECT available_report_tables FROM ai_chat_profiles WHERE id = %s",
(aiId,)
(aiId,),
)
role_row = cur.fetchone()
if not role_row:
@@ -258,7 +248,7 @@ def get_available_tables_str(aiId: str):
return "无数据库表可用"
# 2. 构造 IN 查询占位符
placeholders = ','.join(['%s'] * len(available_tables))
placeholders = ",".join(["%s"] * len(available_tables))
sql_query = f"""
SELECT id, name, description
FROM ai_reports_tables
@@ -272,7 +262,7 @@ def get_available_tables_str(aiId: str):
for table in tableIds:
cur.execute(
"SELECT name, type, description FROM ai_reports_fields WHERE table_id = %s AND is_active = TRUE",
(table[0],)
(table[0],),
)
columns = cur.fetchall()
result += f"{table[1]}：{table[2]}\n"
@@ -291,8 +281,15 @@ def get_available_tables():
cursor.execute(
"SELECT id, name, description,is_active FROM ai_reports_tables",
)
return [{"id": row[0], "name": row[1], "description": row[2], "is_active": row[3]} for row in
cursor.fetchall()]
return [
{
"id": row[0],
"name": row[1],
"description": row[2],
"is_active": row[3],
}
for row in cursor.fetchall()
]
# 新增表
@@ -305,7 +302,7 @@ def add_table(name, description, user_id):
VALUES (%s, %s, %s)
RETURNING id
""",
(name, description, user_id)
(name, description, user_id),
)
new_id = cursor.fetchone()[0] # 取返回的 id
return new_id
@@ -319,8 +316,16 @@ def get_fields_by_table_id(table_id):
"SELECT id, name, type, description, is_active FROM ai_reports_fields WHERE table_id = %s",
(table_id,),
)
return [{"id": row[0], "name": row[1], "type": row[2], "description": row[3], "is_active": row[4]} for row
in cursor.fetchall()]
return [
{
"id": row[0],
"name": row[1],
"type": row[2],
"description": row[3],
"is_active": row[4],
}
for row in cursor.fetchall()
]
# 新增字段
@@ -329,18 +334,26 @@ def add_field(name, type, description, is_active, table_id, user_id):
with conn.cursor() as cursor:
cursor.execute(
"INSERT INTO ai_reports_fields (name,type,description, is_active, create_by, table_id) VALUES (%s, %s, %s, %s, %s, %s) RETURNING id",
(name, type, description, is_active, user_id, table_id)
(name, type, description, is_active, user_id, table_id),
)
new_id = cursor.fetchone()[0] # 取返回的 id
return new_id
# 新增报表智能体
def insert_bot(title: str, description: str, welcome_words: str, ai_personality: str, available_module: str,
available_report_tables: str, available_kn_bases: str, user_id: str):
def insert_bot(
title: str,
description: str,
welcome_words: str,
ai_personality: str,
available_module: str,
available_report_tables: str,
available_kn_bases: str,
user_id: str,
):
with pg_pool.getConn() as conn:
with conn.cursor() as cursor:
available_roles = json.dumps(['user'])
available_roles = json.dumps(["user"])
cursor.execute(
"""
INSERT INTO ai_chat_profiles
@@ -348,19 +361,38 @@ def insert_bot(title: str, description: str, welcome_words: str, ai_personality:
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, now())
RETURNING id
""",
(available_module, available_roles, title, description, welcome_words, ai_personality,
available_report_tables, available_kn_bases, user_id)
(
available_module,
available_roles,
title,
description,
welcome_words,
ai_personality,
available_report_tables,
available_kn_bases,
user_id,
),
)
report_id = cursor.fetchone()[0]
return report_id
# 更新报表智能体
def update_bot(id: str, title: str, description: str, welcome_words: str, ai_personality: str, available_module: str,
available_report_tables: str, available_kn_bases: str, user_id: str):
def update_bot(
id: str,
title: str,
description: str,
welcome_words: str,
ai_personality: str,
available_module: str,
available_report_tables: str,
available_kn_bases: str,
user_id: str,
):
with pg_pool.getConn() as conn:
with conn.cursor() as cursor:
cursor.execute("""
cursor.execute(
"""
UPDATE ai_chat_profiles
SET title = %s,
description = %s,
@@ -373,9 +405,18 @@ def update_bot(id: str, title: str, description: str, welcome_words: str, ai_per
updated_by = %s
WHERE id = %s
""",
(title, description, ai_personality, welcome_words, available_report_tables,
available_kn_bases, available_module, user_id, id)
)
(
title,
description,
ai_personality,
welcome_words,
available_report_tables,
available_kn_bases,
available_module,
user_id,
id,
),
)
# ————————————————————————————————————————————————————知识库———————————————————————————————
@@ -388,11 +429,18 @@ def get_available_knowledge_bases(available_module: str):
FROM ai_knowledge
WHERE available_module::jsonb @> %s::jsonb
""",
(f'["{available_module}"]',)
(f'["{available_module}"]',),
)
return [{"id": row[0], "name": row[1], "description": row[2], "is_active": row[3]} for row in
cursor.fetchall()]
return [
{
"id": row[0],
"name": row[1],
"description": row[2],
"is_active": row[3],
}
for row in cursor.fetchall()
]
def add_knowledge_base(name, description, user_id):
@@ -404,6 +452,108 @@ def add_knowledge_base(name, description, user_id):
VALUES (%s, %s, %s, now())
RETURNING id
""",
(name, description, user_id)
(name, description, user_id),
)
new_id = cursor.fetchone()[0] # 取返回的 id
return new_id
# ————————————————————————————————————————————————————仪评指标联识别———————————————————————————————
from config.minIO import get_temp_url
import utils.MyUtils as MyUtils
def get_ticket_image_list(user_id):
with pg_pool.getConn() as conn:
with conn.cursor() as cursor:
cursor.execute(
"""
SELECT created_at, file_name, resolution, size, name,
moisture_content, cocoon_weight, defective_pupa_count,
fresh_shell_weight, sample_count, barcode, oss,
net_weight_total, evaluator, reviewer,id
FROM ticket_images
WHERE created_by = %s
""",
(user_id,),
)
rows = cursor.fetchall()
result = []
for row in rows:
result.append(
{
"created_at": MyUtils.format_datetime(row[0]),
"file_name": row[1],
"resolution": row[2],
"size": round(row[3], 2),
"name": row[4],
"moisture_content": row[5],
"cocoon_weight": row[6],
"defective_pupa_count": row[7],
"fresh_shell_weight": row[8],
"sample_count": row[9],
"barcode": row[10],
# "oss_url": f"http://{SERVER_PATH_OSS}:9000/image-ticket/{row[11]}",
"oss_url": get_temp_url("image-ticket", row[11]),
"net_weight_total": row[12],
"evaluator": row[13],
"reviewer": row[14],
"id": row[15],
}
)
return result
def insert_ticket_image(
created_by,
file_name,
resolution,
size,
name,
moisture_content,
cocoon_weight,
defective_pupa_count,
fresh_shell_weight,
sample_count,
barcode,
oss,
net_weight_total,
evaluator,
reviewer,
):
with pg_pool.getConn() as conn:
with conn.cursor() as cursor:
cursor.execute(
"""
INSERT INTO ticket_images (
created_by, file_name, resolution, size, name,
moisture_content, cocoon_weight, defective_pupa_count,
fresh_shell_weight, sample_count, barcode, oss,
net_weight_total, evaluator, reviewer, created_at
)
VALUES (%s, %s, %s, %s, %s,
%s, %s, %s, %s, %s,
%s, %s, %s, %s, %s, NOW())
RETURNING id
""",
(
created_by,
file_name,
resolution,
size,
name,
moisture_content,
cocoon_weight,
defective_pupa_count,
fresh_shell_weight,
sample_count,
barcode,
oss,
net_weight_total,
evaluator,
reviewer,
),
)
new_id = cursor.fetchone()[0]
conn.commit()
return new_id