Program.cs

using AIAgentTaskPlanner;
using AIAgentTaskPlanner.Examples;
using AIAgentTaskPlanner.Models;
using AIAgentTaskPlanner.Services;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using Microsoft.SemanticKernel;
using Microsoft.SemanticKernel.ChatCompletion;
using Microsoft.SemanticKernel.Connectors.OpenAI;
using Microsoft.SemanticKernel.PromptTemplates.Handlebars;

// 构建配置
var configuration = new ConfigurationBuilder()
    .SetBasePath(Directory.GetCurrentDirectory())
    .AddJsonFile("appsettings.json", optional: false, reloadOnChange: true)
    .AddUserSecrets<Program>() // 添加用户机密支持
    .Build();

// 配置 AI 服务
Console.WriteLine("🔧 配置 AI 服务...");
Kernel kernel;

// 从 appsettings.json 读取配置
var openAiApiKey = configuration["OpenAI:ApiKey"];
var openAiModel = configuration["OpenAI:Model"] ?? "gpt-4.1";

var azureOpenAiApiKey = configuration["AzureOpenAI:ApiKey"];
var azureOpenAiEndpoint = configuration["AzureOpenAI:Endpoint"];
var azureOpenAiDeployment = configuration["AzureOpenAI:DeploymentName"] ?? "gpt-4.1";

if (!string.IsNullOrWhiteSpace(openAiApiKey) && openAiApiKey != "your-openai-api-key-here")
{
    Console.WriteLine("✅ 使用 OpenAI 服务");
    kernel = KernelConfiguration.CreateWithOpenAI(openAiApiKey, openAiModel);
}
else if (!string.IsNullOrWhiteSpace(azureOpenAiApiKey) && !string.IsNullOrWhiteSpace(azureOpenAiEndpoint))
{
    Console.WriteLine("✅ 使用 Azure OpenAI 服务");
    kernel = KernelConfiguration.CreateWithAzureOpenAI(azureOpenAiDeployment, azureOpenAiEndpoint, azureOpenAiApiKey);
    // 测试 Azure OpenAI 连接
    await TestAIServiceConnection(kernel, "Azure OpenAI");
}
else
{
    Console.WriteLine("⚠️  未检测到 AI 服务配置，使用演示模式");
    Console.WriteLine("   要使用真实 AI 服务，请修改 appsettings.json 中的配置:");
    Console.WriteLine("   - OpenAI:ApiKey (对于 OpenAI)");
    Console.WriteLine("   - AzureOpenAI:ApiKey 和 AzureOpenAI:Endpoint (对于 Azure OpenAI)");
    kernel = KernelConfiguration.CreateDemo();
}

// 配置服务容器
var serviceProvider = KernelConfiguration.ConfigureServices(kernel);
var logger = serviceProvider.GetRequiredService<ILogger<Program>>();
var taskPlanningService = serviceProvider.GetRequiredService<TaskPlanningService>();

Console.WriteLine("🤖 AI Agent 任务规划系统");
Console.WriteLine("基于 .NET 9 + Semantic Kernel");
Console.WriteLine("=====================================\n");

// 主程序循环
while (true)
{
    try
    {
        Console.WriteLine("请选择操作：");
        Console.WriteLine("1. 创建任务计划");
        Console.WriteLine("2. 搜索知识库");
        Console.WriteLine("3. 提交反馈");
        Console.WriteLine("4. 查看知识库统计");
        Console.WriteLine("5. 测试流式输出");
        Console.WriteLine("9. 测试重构功能");
        Console.WriteLine("0. 退出");
        Console.Write("\n请输入选项 (0-9): ");

        var choice = Console.ReadLine()?.Trim();

        // 如果输入为null（如管道输入结束），默认退出
        if (choice == null)
        {
            Console.WriteLine("\n检测到输入结束，程序退出。再见！👋");
            return;
        }

        switch (choice)
        {
            case "1":
                await CreateTaskPlan(taskPlanningService);
                break;
            case "2":
                await SearchKnowledge(taskPlanningService);
                break;
            case "3":
                await SubmitFeedback(taskPlanningService);
                break;
            case "4":
                await ExampleUtils.ShowKnowledgeStats(taskPlanningService);
                break;
            case "0":
                Console.WriteLine("再见！👋");
                return;
            default:
                Console.WriteLine("无效选项，请重试。\n");
                break;
        }
    }
    catch (Exception ex)
    {
        logger.LogError(ex, "程序执行出错");
        Console.WriteLine($"❌ 发生错误: {ex.Message}\n");
    }
}

// 创建任务计划
static async Task CreateTaskPlan(TaskPlanningService service)
{
    Console.WriteLine("\n📝 创建任务计划");
    Console.WriteLine("请输入你的需求描述：");

    var input = Console.ReadLine();

    if (string.IsNullOrWhiteSpace(input))
    {
        Console.WriteLine("\n❌ 无法分析结果：输入不能为空");
        Console.WriteLine();
        return;
    }

    Console.WriteLine("⏳ 正在分析并生成任务计划...");

    try
    {
        SystemOutput? result = null;
        #region 流式处理

        //var lastMessage = "";

        // 使用流式处理显示实时进度
        //await foreach (var update in service.ProcessUserInputStreamAsync(input))
        //{
        //    // 如果消息改变了，显示新消息
        //    if (update.AIPartialContent != lastMessage)
        //    {
        //        if (!string.IsNullOrEmpty(lastMessage))
        //        {
        //            Console.WriteLine(); // 完成上一个消息
        //        }
        //        Console.Write($"🔄 {update.AIPartialContent}");
        //        lastMessage = update.AIPartialContent;
        //    }

        //    // 如果任务完成，从Data中提取结果
        //    if (update.IsComplete && update.Data is SystemOutput systemOutput)
        //    {
        //        result = systemOutput;
        //        Console.WriteLine(" ✅");
        //        break;
        //    }

        //    // 显示进度动画
        //    Console.Write(".");
        //    await Task.Delay(50);
        //}

        #endregion

        // 如果流式处理没有返回最终结果，使用后备方案
        if (result == null)
        {
            Console.WriteLine("\n🔄 完成处理，正在整理结果...");
            result = await service.ProcessUserInputAsync(input);
        }

        // 检查结果是否有效
        if (result == null)
        {
            Console.WriteLine("\n❌ 无法分析结果：系统未能生成有效的任务计划");
            Console.WriteLine("可能原因：");
            Console.WriteLine("- 输入内容无法被理解或识别");
            Console.WriteLine("- AI服务暂时不可用");
            Console.WriteLine("- 输入格式不正确");
            Console.WriteLine("\n建议：请尝试重新描述您的需求，使用更清晰的表达方式");
        }
        else if (string.IsNullOrWhiteSpace(result.TaskAnalysis) &&
                 string.IsNullOrWhiteSpace(result.RagOperations) &&
                 (result.TodoList == null || string.IsNullOrWhiteSpace(result.TodoList.Title)))
        {
            Console.WriteLine("\n❌ 无法分析结果：生成的任务计划为空");
            Console.WriteLine("可能原因：");
            Console.WriteLine("- 输入内容过于简单或模糊");
            Console.WriteLine("- 缺少必要的任务描述信息");
            Console.WriteLine("- 系统无法识别具体的操作需求");
            Console.WriteLine("\n建议：请提供更详细的任务描述和具体要求");
        }
        else
        {
            Console.WriteLine("\n✅ 任务计划生成完成！\n");

            // 显示结果
            try
            {
                var markdown = await service.ExportToMarkdownAsync(result);

                if (string.IsNullOrWhiteSpace(markdown))
                {
                    Console.WriteLine("❌ 无法分析结果：任务计划格式化失败");
                    Console.WriteLine("系统生成了任务计划，但无法将其转换为可读格式");
                    Console.WriteLine("\n建议：请重试或联系技术支持");
                }
                else
                {
                    Console.WriteLine(markdown);

                    // 询问是否保存
                    Console.Write("\n是否要保存为文件？(y/n): ");
                    if (Console.ReadLine()?.ToLower() == "y")
                    {
                        var fileName = $"TaskPlan_{DateTime.Now:yyyyMMdd_HHmmss}.md";
                        await File.WriteAllTextAsync(fileName, markdown);
                        Console.WriteLine($"✅ 已保存为: {fileName}");
                    }
                }
            }
            catch (Exception markdownEx)
            {
                Console.WriteLine("❌ 无法分析结果：格式化输出时发生错误");
                Console.WriteLine($"错误详情: {markdownEx.Message}");
                Console.WriteLine("\n任务计划已生成，但无法正确显示");
                Console.WriteLine("建议：请重试或检查系统配置");
            }
        }
    }
    catch (Exception ex)
    {
        Console.WriteLine($"\n❌ 无法分析结果：生成任务计划时发生错误");
        Console.WriteLine($"错误详情: {ex.Message}");
        Console.WriteLine("\n可能原因：");
        Console.WriteLine("- AI服务连接问题");
        Console.WriteLine("- 输入内容包含无法处理的特殊字符");
        Console.WriteLine("- 系统资源不足");
        Console.WriteLine("\n建议：请检查网络连接并重试，或使用英文描述您的需求");
    }

    Console.WriteLine("\n按任意键继续...");
    Console.ReadKey();
    Console.WriteLine();
}

// 搜索知识库
static async Task SearchKnowledge(TaskPlanningService service)
{
    Console.WriteLine("\n🔍 搜索知识库");
    Console.Write("请输入搜索关键词: ");
    var query = Console.ReadLine();

    if (string.IsNullOrWhiteSpace(query))
    {
        Console.WriteLine("搜索关键词不能为空。\n");
        return;
    }

    try
    {
        var results = await service.SearchKnowledgeAsync(query);

        if (!results.Any())
        {
            Console.WriteLine("未找到相关结果。");
        }
        else
        {
            Console.WriteLine($"\n找到 {results.Count} 个相关结果：\n");

            for (int i = 0; i < results.Count; i++)
            {
                var entry = results[i];
                Console.WriteLine($"📄 结果 {i + 1}");
                Console.WriteLine($"ID: {entry.Id}");
                Console.WriteLine($"分类: {entry.Classification.Category}");
                Console.WriteLine($"标签: {string.Join(", ", entry.Classification.Tags)}");
                Console.WriteLine($"内容预览: {entry.Content[..Math.Min(100, entry.Content.Length)]}...");
                Console.WriteLine($"用户反馈: {entry.UserFeedback?.ToString() ?? "未评价"}");
                Console.WriteLine($"更新时间: {entry.LastUpdated:yyyy-MM-dd HH:mm:ss}");
                Console.WriteLine();
            }
        }
    }
    catch (Exception ex)
    {
        Console.WriteLine($"❌ 搜索时出错: {ex.Message}");
    }

    Console.WriteLine("按任意键继续...");
    Console.ReadKey();
    Console.WriteLine();
}

// 提交反馈
static async Task SubmitFeedback(TaskPlanningService service)
{
    Console.WriteLine("\n💬 提交用户反馈");
    Console.Write("请输入知识条目ID: ");
    var entryId = Console.ReadLine();

    if (string.IsNullOrWhiteSpace(entryId))
    {
        Console.WriteLine("条目ID不能为空。\n");
        return;
    }

    Console.Write("分类是否准确？(y/n): ");
    var isAccurate = Console.ReadLine()?.ToLower() == "y";

    Console.Write("请输入评论（可选）: ");
    var comments = Console.ReadLine() ?? "";

    var feedback = new UserFeedback
    {
        EntryId = entryId,
        IsAccurate = isAccurate,
        Comments = comments
    };

    try
    {
        var result = await service.ProcessUserFeedbackAsync(feedback);
        Console.WriteLine($"✅ {result}");
    }
    catch (Exception ex)
    {
        Console.WriteLine($"❌ 提交反馈时出错: {ex.Message}");
    }

    Console.WriteLine("\n按任意键继续...");
    Console.ReadKey();
    Console.WriteLine();
}

// 测试 AI 服务连接的辅助方法
static async Task TestAIServiceConnection(Kernel kernel, string serviceName)
{
    try
    {
        Console.Write("🔍 正在测试 AI 服务连接...");

        // 获取聊天完成服务
        var chatService = kernel.GetRequiredService<IChatCompletionService>();

        // 创建一个简单的测试消息
        var chatHistory = new ChatHistory();
        chatHistory.AddUserMessage("你好，请回复'连接成功',并告诉我你是用的模型。来确认服务正常");

        // 设置超时时间为10秒
        using var cts = new CancellationTokenSource(TimeSpan.FromSeconds(10));

        // 发送测试请求
        var response = await chatService.GetChatMessageContentAsync(
            chatHistory,
            new OpenAIPromptExecutionSettings
            {
                MaxTokens = 50,
                Temperature = 0.1
            },
            cancellationToken: cts.Token);

        if (!string.IsNullOrWhiteSpace(response?.Content))
        {
            Console.WriteLine(" ✅ 连接成功！");
            Console.WriteLine($"   📡 服务: {serviceName}");
            Console.WriteLine($"   💬 测试响应: {response.Content[..Math.Min(50, response.Content.Length)]}...");
        }
        else
        {
            Console.WriteLine(" ⚠️ 连接成功但响应为空");
        }
    }
    catch (HttpRequestException ex)
    {
        Console.WriteLine(" ❌ 网络连接失败");
        Console.WriteLine($"   错误: {ex.Message}");
        Console.WriteLine("   💡 请检查网络连接和服务端点配置");
    }
    catch (UnauthorizedAccessException ex)
    {
        Console.WriteLine(" ❌ 认证失败");
        Console.WriteLine($"   错误: {ex.Message}");
        Console.WriteLine("   💡 请检查 API 密钥是否正确");
    }
    catch (TaskCanceledException)
    {
        Console.WriteLine(" ❌ 连接超时");
        Console.WriteLine("   💡 请检查网络连接或尝试稍后重试");
    }
    catch (Exception ex)
    {
        Console.WriteLine(" ❌ 连接失败");
        Console.WriteLine($"   错误类型: {ex.GetType().Name}");
        Console.WriteLine($"   错误信息: {ex.Message}");

        // 提供具体的故障排除建议
        if (ex.Message.Contains("401"))
        {
            Console.WriteLine("   💡 建议: API 密钥可能无效，请检查 appsettings.json 中的配置");
        }
        else if (ex.Message.Contains("404"))
        {
            Console.WriteLine("   💡 建议: 部署名称或端点可能不正确");
        }
        else if (ex.Message.Contains("429"))
        {
            Console.WriteLine("   💡 建议: 请求频率过高，请稍后重试");
        }
        else if (ex.Message.Contains("timeout"))
        {
            Console.WriteLine("   💡 建议: 网络连接超时，请检查网络连接");
        }
    }

    Console.WriteLine(); // 添加一个空行
}