ASP.NET Extensions 中自定义 AIContent 类型的 Native AOT 序列化问题解析

在开发基于 ASP.NET Extensions 的 AI 抽象层实现时，特别是针对 Gemini 服务的集成过程中，开发者可能会遇到自定义 AIContent 派生类型在 Native AOT 编译环境下无法正确序列化的问题。本文将深入分析这一问题的技术背景、解决方案及其实现原理。

问题背景

当开发者尝试扩展 Microsoft.Extensions.AI.Abstractions 功能时，通常会创建自定义的 AIContent 派生类型。例如，为 Gemini 服务实现代码解释器工具支持时，需要创建 ExecutableCodeContent 和 CodeExecutionContent 等类型来封装代码执行结果。

在常规的 CLR 环境下，这些自定义类型能够通过反射机制正常序列化。然而，当应用采用 Native AOT 编译或进行程序集裁剪时，反射机制不可用，必须依赖源生成（Source Generation）技术来实现 JSON 序列化。

技术挑战

Native AOT 环境下的主要挑战在于：

1. 默认的 AIJsonUtilities.DefaultOptions 会根据运行时环境自动切换行为
2. 在 AOT 环境下，它使用仅限于当前程序集内类型的源生成契约
3. 任何扩展类型都需要显式进行源生成并添加到解析器链中

解决方案演进

初始方案

开发者最初尝试的方案是：

var options = new JsonSerializerOptions(AIJsonUtilities.DefaultOptions)
{
    TypeInfoResolver = JsonTypeInfoResolver.Combine(
        JsonContext.Default,
        AIJsonUtilities.DefaultOptions.TypeInfoResolver
    )
};

这种方案在理论上应该可行，但在实践中遇到了类型解析顺序问题。

正确实现

经过验证，正确的实现方式需要注意以下几点：

1. 必须先配置 TypeInfoResolver 链
2. 然后再注册自定义 AIContent 类型
3. 必须确保源生成上下文包含所有自定义类型

最终的工作代码如下：

var options = new JsonSerializerOptions(AIJsonUtilities.DefaultOptions);

// 先配置解析器链
options.TypeInfoResolver = JsonTypeInfoResolver.Combine(
    JsonContext.Default,
    AIJsonUtilities.DefaultOptions.TypeInfoResolver
);

// 后注册自定义类型
options.AddAIContentType<ExecutableCodeContent>("executable_code");
options.AddAIContentType<CodeExecutionContent>("code_execution");

// 源生成上下文定义
[JsonSerializable(typeof(ExecutableCodeContent))]
[JsonSerializable(typeof(CodeExecutionContent))]
partial class JsonContext : JsonSerializerContext;

技术原理

这一解决方案背后的关键技术点包括：

1. 解析器链顺序：JsonTypeInfoResolver.Combine 创建的是不可变实例，避免了循环引用问题
2. 源生成机制：在 AOT 环境下为特定类型生成静态序列化代码
3. 多态类型处理：通过 AddAIContentType 注册派生类型与鉴别器的映射关系

最佳实践建议

基于这一问题的解决经验，我们建议开发者在处理类似场景时：

1. 始终先配置解析器链再注册自定义类型
2. 为所有自定义 AIContent 派生类型创建源生成上下文
3. 在 Native AOT 环境下进行充分测试
4. 注意解析器链的组合顺序可能影响类型解析结果

通过遵循这些实践，开发者可以确保自定义 AI 内容类型在各种运行时环境下都能正确序列化，为 AI 服务集成提供可靠的基础设施支持。