wunianxiao/iAs
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: token统计+聊天记录入库静默失败, 列名/类型/去重不匹配

Browse Source 
根因：insert_llm_usage / insert_chat_record 因旧 iPet 遗留列 (kind/entry NOT NULL,
context_token_present boolean→context_token text) 导致 INSERT 静默失败,
四个调用点 let _ = 吞掉错误。ias usage 显示 token 总量为 0。

修复：
- llm_usage: 迁移 0010 ADD COLUMN IF NOT EXISTS + 条件 DROP NOT NULL,
  insert_llm_usage 写入 kind/entry 默认值
- chat_records: 迁移 0011 列名 context_token_present→context_token 类型转换+重命名,
  迁移 0012 清理存量重复→partial UNIQUE index→恢复 ON CONFLICT
- main.rs: 所有 let _ = insert_* 改为 if let Err(e) 日志输出,
  message_id 缺失时归为 '' 而非 '0'
This commit is contained in:
无念晓
2026-06-02 16:31:48 +08:00
parent 1aef0c2f69
commit dcfaf47809
5 changed files with 94 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main.rs
+29 -12
View File
@@ -162,7 +162,9 @@ async fn cmd_listen(
account_id: fa.account_id,
base_url: fa.base_url,
};
let _ = db::models::save_auth(db.pool(), &a).await;
if let Err(e) = db::models::save_auth(db.pool(), &a).await {
error!("保存认证信息失败: {}", e);
}
break 'load Some(a);
}
} else if let Some(a) = file_state.load_auth() {
@@ -481,13 +483,13 @@ async fn listen_loop(
let from = msg.from_user_id.as_deref().unwrap_or("unknown");
let text = msg.text_content().unwrap_or("(非文本消息)");
let ctx_token = msg.context_token.as_deref();
let msg_id = msg.message_id.unwrap_or(0).to_string();
let msg_id = msg.message_id.map(|id| id.to_string()).unwrap_or_default();
info!("收到消息 from={}: {}", from, text);
// === 入库：收到的消息 ===
if let Some(db) = database {
let _ = db::models::insert_chat_record(
if let Err(e) = db::models::insert_chat_record(
db.pool(),
"inbound",
from,
@@ -497,7 +499,10 @@ async fn listen_loop(
ctx_token,
&msg_id,
)
.await;
.await
{
error!("保存聊天记录失败: {}", e);
}
}
// Echo 模式
@@ -507,7 +512,7 @@ async fn listen_loop(
info!("回显成功 msg_id={}", sent_id);
// === 入库：发送的回显 ===
if let Some(db) = database {
let _ = db::models::insert_chat_record(
if let Err(e) = db::models::insert_chat_record(
db.pool(),
"outbound",
from,
@@ -517,7 +522,10 @@ async fn listen_loop(
ctx_token,
&sent_id,
)
.await;
.await
{
error!("保存聊天记录失败: {}", e);
}
}
}
Err(e) => error!("回显失败: {}", e),
@@ -579,7 +587,7 @@ async fn listen_loop(
Ok(sent_id) => {
info!("LLM 回复成功 msg_id={}", sent_id);
if let Some(db) = database {
let _ = db::models::insert_chat_record(
if let Err(e) = db::models::insert_chat_record(
db.pool(),
"outbound",
&from_owned,
@@ -589,7 +597,10 @@ async fn listen_loop(
ctx_owned.as_deref(),
&sent_id,
)
.await;
.await
{
error!("保存聊天记录失败: {}", e);
}
}
}
Err(e) => error!("发送 LLM 回复失败: {}", e),
@@ -659,7 +670,7 @@ async fn store_usage(
u: &Usage,
) {
if let Some(db) = db {
let _ = db::models::insert_llm_usage(
if let Err(e) = db::models::insert_llm_usage(
db.pool(),
user_id,
model,
@@ -669,7 +680,10 @@ async fn store_usage(
u.prompt_cache_hit_tokens,
u.prompt_cache_miss_tokens,
)
.await;
.await
{
tracing::error!("存储 LLM 用量失败: {}", e);
}
}
}
@@ -797,7 +811,7 @@ async fn cmd_send(
// 入库：发送的消息
if let Some(db) = database {
let _ = db::models::insert_chat_record(
if let Err(e) = db::models::insert_chat_record(
db.pool(),
"outbound",
to,
@@ -807,7 +821,10 @@ async fn cmd_send(
context_token,
&msg_id,
)
.await;
.await
{
error!("保存聊天记录失败: {}", e);
}
}
}
Err(e) => error!("发送失败: {}", e),