82 lines
3.1 KiB
C#
82 lines
3.1 KiB
C#
using System.Text.Json;
|
||
using MAF1.Tools;
|
||
using MAF1.Utils;
|
||
using Microsoft.Agents.AI;
|
||
using Microsoft.Extensions.AI;
|
||
|
||
namespace MAF1.Agents.FileCity;
|
||
|
||
public static class FileCityAgent
|
||
{
|
||
public static AIAgent Create(AgentFactory factory)
|
||
{
|
||
return factory.CreateAgent(
|
||
name: "FileCityAgent",
|
||
instructions:
|
||
"""
|
||
你负责读取用户指定的本地文件，并从中提取有效城市名。
|
||
必须先调用 ReadTextFile 读取文件，不要猜测文件内容。
|
||
有效城市名：现实世界中真实存在的城市（如 成都、北京、Amsterdam）。
|
||
忽略空行、注释、人名、水果、随意单词等不是城市的内容。
|
||
只输出一个 JSON 对象，不要 Markdown，不要其它说明：
|
||
{"hasValidCities": true, "cities": ["成都"], "reason": "说明"}
|
||
若没有有效城市：{"hasValidCities": false, "cities": [], "reason": "原因"}
|
||
cities 去重，保持文件中的出现顺序。
|
||
""",
|
||
tools: [AIFunctionFactory.Create(FileTools.ReadTextFile)]);
|
||
}
|
||
|
||
public static async Task<AgentStepResult> RunAsync(
|
||
AIAgent agent,
|
||
IReadOnlyDictionary<string, object?> inputs,
|
||
CancellationToken cancellationToken = default)
|
||
{
|
||
string filePath = AgentInputs.ReadString(inputs, "filePath") ?? "Data/cities.txt";
|
||
AgentResponse response = await agent.RunAsync(
|
||
$"请读取这个文件并提取有效城市名：{filePath}",
|
||
cancellationToken: cancellationToken);
|
||
CityExtraction extraction = Parse(response.Text);
|
||
return new AgentStepResult
|
||
{
|
||
Message = response.Text,
|
||
Inputs = new Dictionary<string, object?> { ["filePath"] = filePath },
|
||
Outputs = new Dictionary<string, object?>
|
||
{
|
||
["hasValidCities"] = extraction.HasValidCities,
|
||
["cities"] = extraction.Cities,
|
||
["reason"] = extraction.Reason,
|
||
["raw"] = response.Text,
|
||
},
|
||
};
|
||
}
|
||
|
||
public static CityExtraction Parse(string text)
|
||
{
|
||
try
|
||
{
|
||
CityExtraction? result = JsonSerializer.Deserialize<CityExtraction>(
|
||
JsonText.UnwrapObject(text),
|
||
new JsonSerializerOptions { PropertyNameCaseInsensitive = true });
|
||
if (result is null)
|
||
{
|
||
return Invalid("无法解析文件 Agent 的输出。");
|
||
}
|
||
|
||
result.Cities = result.Cities
|
||
.Where(city => !string.IsNullOrWhiteSpace(city))
|
||
.Select(city => city.Trim())
|
||
.Distinct(StringComparer.OrdinalIgnoreCase)
|
||
.ToList();
|
||
result.HasValidCities = result.HasValidCities && result.Cities.Count > 0;
|
||
return result;
|
||
}
|
||
catch (JsonException ex)
|
||
{
|
||
return Invalid($"文件 Agent 没有返回合法 JSON：{ex.Message}");
|
||
}
|
||
}
|
||
|
||
private static CityExtraction Invalid(string reason) =>
|
||
new() { HasValidCities = false, Reason = reason };
|
||
}
|