fix(ai): Agent chat handler 精确选择 FC-capable provider + 环境变量适配

- chat_handler: 使用 get_provider("claude") 精确获取,避免 resolve fallback 到 Ollama
- ProviderRegistry: 新增 get_provider() 方法(无 health check,无 fallback)
- orchestrator: 从 ANTHROPIC_DEFAULT_SONNET_MODEL 读取模型名,兼容智谱代理
- erp-server: Claude provider 注册优先读 ANTHROPIC_AUTH_TOKEN + ANTHROPIC_BASE_URL

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
iven
2026-05-18 03:58:38 +08:00
parent e47fe547c8
commit 882b27ab7a
4 changed files with 30 additions and 15 deletions

View File

@@ -51,7 +51,8 @@ impl AgentOrchestrator {
messages.clone(),
tools.clone(),
system_prompt,
"auto",
&std::env::var("ANTHROPIC_DEFAULT_SONNET_MODEL")
.unwrap_or_else(|_| "claude-sonnet-4-6".to_string()),
0.7,
2048,
)

View File

@@ -128,14 +128,16 @@ where
tool_call_id: None,
});
// 解析 Provider
let resolved = ai_state
// 解析 Provider — Agent 需要 Function Calling精确获取 Claude/OpenAI
let provider_arc = ai_state
.provider_registry
.resolve("auto")
.await
.map_err(|e| {
tracing::error!(error = %e, "AI provider resolve failed");
erp_core::error::AppError::Internal("AI 服务暂时不可用,请稍后再试".into())
.get_provider("claude")
.or_else(|| ai_state.provider_registry.get_provider("openai"))
.ok_or_else(|| {
tracing::error!("No FC-capable provider found (need claude or openai)");
erp_core::error::AppError::Internal(
"AI Agent 暂时不可用,需要 Claude 或 OpenAI 提供商".into(),
)
})?;
// 构建 ToolRegistry — Phase 0 只有 query_patient_vitals
@@ -159,7 +161,6 @@ where
);
// 执行 Agent ReAct 循环
let provider_arc = resolved.into_arc();
let orchestrator = AgentOrchestrator::new(provider_arc, std::sync::Arc::new(registry));
let result = orchestrator
.run(SYSTEM_PROMPT, &mut messages, &tool_ctx)

View File

@@ -105,6 +105,11 @@ impl ProviderRegistry {
pub fn provider_names(&self) -> Vec<String> {
self.entries.iter().map(|e| e.key().to_string()).collect()
}
/// 精确获取指定名称的 provider不做 health check不做 fallback
pub fn get_provider(&self, name: &str) -> Option<Arc<dyn AiProvider>> {
self.entries.get(name).map(|e| e.provider.clone())
}
}
pub struct ResolvedProvider {

View File

@@ -477,13 +477,21 @@ async fn main() -> anyhow::Result<()> {
// 构建多 Provider 注册表
let registry = std::sync::Arc::new(erp_ai::provider::registry::ProviderRegistry::new());
// 始终注册默认 Claude provider(兼容旧配置)
// 始终注册默认 Claude provider — 优先用环境变量
{
let mut claude =
erp_ai::provider::claude::ClaudeProvider::new(config.ai.api_key.clone());
if let Some(ref base_url) = config.ai.base_url {
claude = claude.with_base_url(base_url.clone());
}
let api_key = if !config.ai.api_key.is_empty() {
config.ai.api_key.clone()
} else {
std::env::var("ANTHROPIC_AUTH_TOKEN")
.or_else(|_| std::env::var("ANTHROPIC_API_KEY"))
.unwrap_or_default()
};
let base_url = std::env::var("ANTHROPIC_BASE_URL")
.ok()
.or_else(|| config.ai.base_url.clone())
.unwrap_or_else(|| "https://api.anthropic.com".to_string());
let claude =
erp_ai::provider::claude::ClaudeProvider::new(api_key).with_base_url(base_url);
registry.register("claude".to_string(), std::sync::Arc::new(claude));
}