fix(ai): 审计问题修复 — 错误映射/性能/SSE/依赖规范化
- C3: handler 中 .map_err(AppError::Internal) 改为 ? 操作符,
利用 From<AiError> for AppError 实现正确的 HTTP 状态码映射
- H1: AiState 预构建在 AppState 初始化时,避免每次请求重建
ClaudeProvider/AnalysisService/PromptService/UsageService
- H3: stream_analyze 的 user_id 参数传递到 created_by/updated_by
- H5: SSE 事件添加 .event("chunk"/"error"/"done") 类型字段
- L3: erp-ai Cargo.toml 依赖改用 workspace 引用
(reqwest/handlebars/sha2/hex)
- 修复 erp-health 编译错误: points_handler 缺少 ColumnTrait 导入,
points_service 版本字段部分移动问题
This commit is contained in:
@@ -38,8 +38,7 @@ where
|
||||
let prompt = state
|
||||
.prompt
|
||||
.get_active_prompt(ctx.tenant_id, "lab_report_interpretation")
|
||||
.await
|
||||
.map_err(|e| erp_core::error::AppError::Internal(e.to_string()))?;
|
||||
.await?;
|
||||
|
||||
let model_config = &prompt.model_config;
|
||||
let model = model_config["model"].as_str().unwrap_or("claude-sonnet-4-6").to_string();
|
||||
@@ -61,8 +60,7 @@ where
|
||||
temperature,
|
||||
max_tokens,
|
||||
)
|
||||
.await
|
||||
.map_err(|e| erp_core::error::AppError::Internal(e.to_string()))?;
|
||||
.await?;
|
||||
|
||||
let analysis_id_clone = analysis_id;
|
||||
let state_clone = state.clone();
|
||||
@@ -82,14 +80,14 @@ where
|
||||
index,
|
||||
};
|
||||
let data = serde_json::to_string(&event).unwrap_or_default();
|
||||
yield Ok(Event::default().data(data));
|
||||
yield Ok(Event::default().event("chunk").data(data));
|
||||
}
|
||||
Err(e) => {
|
||||
let event = AnalysisSseEvent::Error {
|
||||
message: e.to_string(),
|
||||
};
|
||||
let data = serde_json::to_string(&event).unwrap_or_default();
|
||||
yield Ok(Event::default().data(data));
|
||||
yield Ok(Event::default().event("error").data(data));
|
||||
let _ = state_clone
|
||||
.analysis
|
||||
.fail_analysis(analysis_id_clone, e.to_string())
|
||||
@@ -111,7 +109,7 @@ where
|
||||
status: "completed".into(),
|
||||
};
|
||||
let data = serde_json::to_string(&done_event).unwrap_or_default();
|
||||
yield Ok(Event::default().data(data));
|
||||
yield Ok(Event::default().event("done").data(data));
|
||||
};
|
||||
|
||||
Ok(Sse::new(sse_stream).keep_alive(KeepAlive::default()))
|
||||
@@ -134,8 +132,7 @@ where
|
||||
let prompt = state
|
||||
.prompt
|
||||
.get_active_prompt(ctx.tenant_id, "health_trend_analysis")
|
||||
.await
|
||||
.map_err(|e| erp_core::error::AppError::Internal(e.to_string()))?;
|
||||
.await?;
|
||||
|
||||
let model_config = &prompt.model_config;
|
||||
let model = model_config["model"].as_str().unwrap_or("claude-sonnet-4-6").to_string();
|
||||
@@ -157,8 +154,7 @@ where
|
||||
temperature,
|
||||
max_tokens,
|
||||
)
|
||||
.await
|
||||
.map_err(|e| erp_core::error::AppError::Internal(e.to_string()))?;
|
||||
.await?;
|
||||
|
||||
let analysis_id_clone = analysis_id;
|
||||
let state_clone = state.clone();
|
||||
@@ -184,8 +180,7 @@ where
|
||||
let prompt = state
|
||||
.prompt
|
||||
.get_active_prompt(ctx.tenant_id, "personalized_checkup_plan")
|
||||
.await
|
||||
.map_err(|e| erp_core::error::AppError::Internal(e.to_string()))?;
|
||||
.await?;
|
||||
|
||||
let model_config = &prompt.model_config;
|
||||
let model = model_config["model"].as_str().unwrap_or("claude-sonnet-4-6").to_string();
|
||||
@@ -207,8 +202,7 @@ where
|
||||
temperature,
|
||||
max_tokens,
|
||||
)
|
||||
.await
|
||||
.map_err(|e| erp_core::error::AppError::Internal(e.to_string()))?;
|
||||
.await?;
|
||||
|
||||
let analysis_id_clone = analysis_id;
|
||||
let state_clone = state.clone();
|
||||
@@ -234,8 +228,7 @@ where
|
||||
let prompt = state
|
||||
.prompt
|
||||
.get_active_prompt(ctx.tenant_id, "report_summary_generation")
|
||||
.await
|
||||
.map_err(|e| erp_core::error::AppError::Internal(e.to_string()))?;
|
||||
.await?;
|
||||
|
||||
let model_config = &prompt.model_config;
|
||||
let model = model_config["model"].as_str().unwrap_or("claude-sonnet-4-6").to_string();
|
||||
@@ -257,8 +250,7 @@ where
|
||||
temperature,
|
||||
max_tokens,
|
||||
)
|
||||
.await
|
||||
.map_err(|e| erp_core::error::AppError::Internal(e.to_string()))?;
|
||||
.await?;
|
||||
|
||||
let analysis_id_clone = analysis_id;
|
||||
let state_clone = state.clone();
|
||||
@@ -323,12 +315,12 @@ fn build_sse_stream(
|
||||
index += 1;
|
||||
let event = AnalysisSseEvent::Chunk { content: chunk, index };
|
||||
let data = serde_json::to_string(&event).unwrap_or_default();
|
||||
yield Ok(Event::default().data(data));
|
||||
yield Ok(Event::default().event("chunk").data(data));
|
||||
}
|
||||
Err(e) => {
|
||||
let event = AnalysisSseEvent::Error { message: e.to_string() };
|
||||
let data = serde_json::to_string(&event).unwrap_or_default();
|
||||
yield Ok(Event::default().data(data));
|
||||
yield Ok(Event::default().event("error").data(data));
|
||||
let _ = state.analysis.fail_analysis(analysis_id, e.to_string()).await;
|
||||
return;
|
||||
}
|
||||
@@ -343,6 +335,6 @@ fn build_sse_stream(
|
||||
status: "completed".into(),
|
||||
};
|
||||
let data = serde_json::to_string(&done_event).unwrap_or_default();
|
||||
yield Ok(Event::default().data(data));
|
||||
yield Ok(Event::default().event("done").data(data));
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user