54 lines
1.8 KiB
C#
54 lines
1.8 KiB
C#
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Logging;
|
|
using Microsoft.Agents.AI;
|
|
|
|
namespace Service.AgentFramework;
|
|
|
|
/// <summary>
|
|
/// Agent Framework 依赖注入扩展
|
|
/// </summary>
|
|
public static class AgentFrameworkExtensions
|
|
{
|
|
/// <summary>
|
|
/// 注册 Agent Framework 相关服务
|
|
/// </summary>
|
|
public static IServiceCollection AddAgentFramework(this IServiceCollection services)
|
|
{
|
|
// 注册 Tool Registry (Singleton - 无状态，全局共享)
|
|
services.AddSingleton<IToolRegistry, ToolRegistry>();
|
|
|
|
// 注册 Tools (Scoped - 因为依赖 Scoped Repository)
|
|
services.AddSingleton<ITransactionQueryTools, TransactionQueryTools>();
|
|
services.AddSingleton<ITextProcessingTools, TextProcessingTools>();
|
|
services.AddSingleton<IAITools, AITools>();
|
|
|
|
// 注册 Agents (Scoped - 因为依赖 Scoped Tools)
|
|
services.AddSingleton<ClassificationAgent>();
|
|
services.AddSingleton<ParsingAgent>();
|
|
services.AddSingleton<ImportAgent>();
|
|
|
|
// 注册 Service Facade (Scoped - 避免生命周期冲突)
|
|
services.AddSingleton<ISmartHandleServiceV2, SmartHandleServiceV2>();
|
|
|
|
return services;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 初始化 Agent 框架的 Tools
|
|
/// 在应用启动时调用此方法
|
|
/// </summary>
|
|
public static void InitializeAgentTools(
|
|
this IServiceProvider serviceProvider)
|
|
{
|
|
var toolRegistry = serviceProvider.GetRequiredService<IToolRegistry>();
|
|
var logger = serviceProvider.GetRequiredService<ILogger<IToolRegistry>>();
|
|
|
|
logger.LogInformation("开始初始化 Agent Tools...");
|
|
|
|
// 这里可以注册更多的 Tool
|
|
// 目前大部分 Tool 被整合到了工具类中，后续可根据需要扩展
|
|
|
|
logger.LogInformation("Agent Tools 初始化完成");
|
|
}
|
|
}
|