style: auto-format code with ruff (cron)
This commit is contained in:
@@ -283,7 +283,7 @@ class AIManager:
|
||||
def get_custom_model(self, model_id: str) -> CustomModel | None:
|
||||
"""获取自定义模型"""
|
||||
with self._get_db() as conn:
|
||||
row = conn.execute("SELECT * FROM custom_models WHERE id = ?", (model_id, )).fetchone()
|
||||
row = conn.execute("SELECT * FROM custom_models WHERE id = ?", (model_id,)).fetchone()
|
||||
|
||||
if not row:
|
||||
return None
|
||||
@@ -350,7 +350,8 @@ class AIManager:
|
||||
"""获取训练样本"""
|
||||
with self._get_db() as conn:
|
||||
rows = conn.execute(
|
||||
"SELECT * FROM training_samples WHERE model_id = ? ORDER BY created_at", (model_id, )
|
||||
"SELECT * FROM training_samples WHERE model_id = ? ORDER BY created_at",
|
||||
(model_id,),
|
||||
).fetchall()
|
||||
|
||||
return [self._row_to_training_sample(row) for row in rows]
|
||||
@@ -712,7 +713,7 @@ class AIManager:
|
||||
def get_kg_rag(self, rag_id: str) -> KnowledgeGraphRAG | None:
|
||||
"""获取知识图谱 RAG 配置"""
|
||||
with self._get_db() as conn:
|
||||
row = conn.execute("SELECT * FROM kg_rag_configs WHERE id = ?", (rag_id, )).fetchone()
|
||||
row = conn.execute("SELECT * FROM kg_rag_configs WHERE id = ?", (rag_id,)).fetchone()
|
||||
|
||||
if not row:
|
||||
return None
|
||||
@@ -1122,7 +1123,7 @@ class AIManager:
|
||||
"""获取预测模型"""
|
||||
with self._get_db() as conn:
|
||||
row = conn.execute(
|
||||
"SELECT * FROM prediction_models WHERE id = ?", (model_id, )
|
||||
"SELECT * FROM prediction_models WHERE id = ?", (model_id,)
|
||||
).fetchone()
|
||||
|
||||
if not row:
|
||||
@@ -1238,7 +1239,7 @@ class AIManager:
|
||||
# 更新预测计数
|
||||
conn.execute(
|
||||
"UPDATE prediction_models SET prediction_count = prediction_count + 1 WHERE id = ?",
|
||||
(model_id, ),
|
||||
(model_id,),
|
||||
)
|
||||
conn.commit()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user