在dotnet/extensions项目中实现自定义AIContent类型的Native AOT序列化

背景介绍

在dotnet/extensions项目中，Microsoft.Extensions.AI.Abstractions提供了一套用于人工智能交互的抽象接口。当开发者需要实现自定义的AIContent派生类型时，特别是在Native AOT编译环境下，会遇到JSON序列化方面的挑战。

问题核心

开发者在为Gemini实现AI扩展时，创建了两种自定义AIContent派生类型：ExecutableCodeContent和CodeExecutionContent。在常规CLR环境下，这些类型能够正常通过LoggingChatClient进行JSON序列化记录。但在Native AOT编译环境下，系统会抛出"JsonTypeInfo metadata not provided"异常。

技术分析

问题的根源在于Native AOT环境下System.Text.Json的行为变化：

1. 在常规CLR环境下，AIJsonUtilities.DefaultOptions使用反射机制进行序列化
2. 在Native AOT或裁剪环境下，它会切换到仅针对程序集内类型的源生成契约

解决方案

要解决这个问题，需要以下步骤：

1. 为自定义类型创建JsonSerializerContext
2. 将自定义上下文的TypeInfoResolver与AIJsonUtilities.DefaultOptions的解析器链合并
3. 确保在设置TypeInfoResolver后再调用AddAIContentType方法

具体实现代码如下：

// 创建自定义JSON序列化上下文
[JsonSerializable(typeof(ExecutableCodeContent))]
[JsonSerializable(typeof(CodeExecutionContent))]
partial class JsonContext : JsonSerializerContext;

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

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

注意事项

1. 顺序很重要：必须先配置TypeInfoResolver，再调用AddAIContentType
2. 使用JsonTypeInfoResolver.Combine创建不可变实例，避免潜在的循环引用问题
3. 在Native AOT环境下，所有需要序列化的类型都必须显式声明在JsonSerializerContext中

潜在问题

开发者可能会遇到堆栈溢出问题，这通常是由于解析器链配置不当导致的。正确的做法是使用JsonTypeInfoResolver.Combine来合并解析器，而不是直接修改TypeInfoResolverChain属性。

结论

在dotnet/extensions项目中实现自定义AIContent类型的Native AOT序列化需要特别注意JSON序列化上下文的配置。通过正确组合源生成解析器并遵循特定的配置顺序，可以确保自定义类型在Native AOT环境下能够正常序列化。这种解决方案不仅适用于当前场景，也为其他需要在Native AOT环境下使用System.Text.Json进行复杂类型序列化的场景提供了参考。