feat: Added M.E.A.I AIFunction generation

Gunpal Jain · Gunpal Jain · commit 234c61eff186 · 2025-03-10T13:44:06.000+05:30
diff --git a/CSharpToJsonSchema.sln b/CSharpToJsonSchema.sln
@@ -42,6 +42,8 @@ Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "CSharpToJsonSchema.Integrat
 EndProject
 Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "CSharpToJsonSchema.AotTests", "src\tests\CSharpToJsonSchema.AotTests\CSharpToJsonSchema.AotTests.csproj", "{6167F915-83EB-42F9-929B-AD4719A55811}"
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "CSharpToJsonSchema.MeaiTests", "src\tests\CSharpToJsonSchema.MeaiTests\CSharpToJsonSchema.MeaiTests.csproj", "{DC07C90E-A58C-44B3-82D2-E2EB8F777B92}"
+EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|Any CPU = Debug|Any CPU
@@ -76,6 +78,10 @@ Global
 		{6167F915-83EB-42F9-929B-AD4719A55811}.Debug|Any CPU.Build.0 = Debug|Any CPU
 		{6167F915-83EB-42F9-929B-AD4719A55811}.Release|Any CPU.ActiveCfg = Release|Any CPU
 		{6167F915-83EB-42F9-929B-AD4719A55811}.Release|Any CPU.Build.0 = Release|Any CPU
+		{DC07C90E-A58C-44B3-82D2-E2EB8F777B92}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{DC07C90E-A58C-44B3-82D2-E2EB8F777B92}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{DC07C90E-A58C-44B3-82D2-E2EB8F777B92}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{DC07C90E-A58C-44B3-82D2-E2EB8F777B92}.Release|Any CPU.Build.0 = Release|Any CPU
 	EndGlobalSection
 	GlobalSection(SolutionProperties) = preSolution
 		HideSolutionNode = FALSE
@@ -89,6 +95,7 @@ Global
 		{8AFCC30C-C59D-498D-BE68-9A328B3E5599} = {AAA11B78-2764-4520-A97E-46AA7089A588}
 		{247C813A-9072-4DF3-B403-B35E3688DB4B} = {AAA11B78-2764-4520-A97E-46AA7089A588}
 		{6167F915-83EB-42F9-929B-AD4719A55811} = {AAA11B78-2764-4520-A97E-46AA7089A588}
+		{DC07C90E-A58C-44B3-82D2-E2EB8F777B92} = {AAA11B78-2764-4520-A97E-46AA7089A588}
 	EndGlobalSection
 	GlobalSection(ExtensibilityGlobals) = postSolution
 		SolutionGuid = {CED9A020-DBA5-4BE6-8096-75E528648EC1}
diff --git a/src/libs/CSharpToJsonSchema.Generators/Conversion/ToModels.cs b/src/libs/CSharpToJsonSchema.Generators/Conversion/ToModels.cs
@@ -18,6 +18,8 @@ public static InterfaceData PrepareData(
                        strict;
         var generateGoogleFunctionTool = attributeData.NamedArguments.FirstOrDefault(x => x.Key == "GoogleFunctionTool").Value.Value is bool googleFunctionTool &&
                                          googleFunctionTool;
+        var meaiFunctionTool= attributeData.NamedArguments.FirstOrDefault(x => x.Key == "MeaiFunctionTool").Value.Value is bool meaift &&
+                              meaift;
         var methods = interfaceSymbol
             .GetMembers()
             .OfType<IMethodSymbol>()
@@ -46,6 +48,7 @@ public static InterfaceData PrepareData(
             Namespace: interfaceSymbol.ContainingNamespace.ToDisplayString(),
             Name: interfaceSymbol.ToDisplayString(SymbolDisplayFormat.MinimallyQualifiedFormat),
             GoogleFunctionTool:generateGoogleFunctionTool,
+            MeaiFunctionTool:meaiFunctionTool,
             Methods: methods);
     }
 
@@ -60,6 +63,7 @@ public static InterfaceData PrepareMethodData(
         List<MethodData> methodList = new();
         List<string> namespaces = new();
         bool generateGoogleFunctionTools = false;
+        bool meaiFunctionTools = false;
         foreach (var l in list)
         {
             var (interfaceSymbol, attributeData) = l;
@@ -70,6 +74,11 @@ public static InterfaceData PrepareMethodData(
             if(ggft)
                 generateGoogleFunctionTools = true;
             
+            var meai = attributeData.NamedArguments.FirstOrDefault(x => x.Key == "MeaiFunctionTool").Value.Value is bool meaift &&
+                       meaift;
+            if(meai)
+                meaiFunctionTools = true;
+            
             var x = interfaceSymbol;
             var parameters = x.Parameters
                 //.Where(static x => x.Type.MetadataName != "CancellationToken")
@@ -94,6 +103,7 @@ public static InterfaceData PrepareMethodData(
             Namespace: GetCommonRootNamespace(namespaces)??namespaceName,
             Name: className,
             GoogleFunctionTool: generateGoogleFunctionTools,
+            MeaiFunctionTool:meaiFunctionTools,
             Methods: methodList.ToArray());
     }
     public static string? GetCommonRootNamespace(IEnumerable<string> namespaces)
diff --git a/src/libs/CSharpToJsonSchema.Generators/JsonSchemaGenerator.cs b/src/libs/CSharpToJsonSchema.Generators/JsonSchemaGenerator.cs
@@ -50,6 +50,9 @@ private void ProcessMethods(IncrementalGeneratorInitializationContext context)
         attributes
             .SelectAndReportExceptions(AsGoogleFunctionToolsForMethods, context, Id)
             .AddSource(context);
+        attributes
+            .SelectAndReportExceptions(AsMeaiFunctionToolsForMethods, context, Id)
+            .AddSource(context);
         
         var generator = new JsonSourceGenerator();
         generator.InitializeForFunctionTools(context);
@@ -73,6 +76,9 @@ private void ProcessInterfaces(IncrementalGeneratorInitializationContext context
         attributes
             .SelectAndReportExceptions(AsGoogleFunctionToolsForInterface, context, Id)
             .AddSource(context);
+        attributes
+            .SelectAndReportExceptions(AsMeaiFunctionToolsForInterface, context, Id)
+            .AddSource(context);
         
         var generator = new JsonSourceGenerator();
         generator.Initialize2(context);
@@ -119,14 +125,28 @@ private static FileWithName AsFunctionCalls(InterfaceData @interface)
             Name: $"{@interface.Name}.FunctionCalls.generated.cs",
             Text: Sources.GenerateFunctionCalls(@interface));
     }
-
+    
     private static FileWithName AsGoogleFunctionToolsForMethods(InterfaceData @interface)
     {
         return new FileWithName(
             Name: $"{@interface.Name}.GoogleFunctionTools.generated.cs",
             Text: Sources.GenerateGoogleFunctionToolForMethods(@interface));
     }
     
+    private static FileWithName AsMeaiFunctionToolsForMethods(InterfaceData @interface)
+    {
+        return new FileWithName(
+            Name: $"{@interface.Name}.MeaiTools.generated.cs",
+            Text: Sources.GenerateMeaiFunctionToolForMethods(@interface));
+    }
+    
+    private static FileWithName AsMeaiFunctionToolsForInterface(InterfaceData @interface)
+    {
+        return new FileWithName(
+            Name: $"{@interface.Name}.MeaiToolsExtensions.generated.cs",
+            Text: Sources.GenerateMeaiFunctionToolForInterface(@interface));
+    }
+    
     private static FileWithName AsGoogleFunctionToolsForInterface(InterfaceData @interface)
     {
         return new FileWithName(
diff --git a/src/libs/CSharpToJsonSchema.Generators/Models/InterfaceData.cs b/src/libs/CSharpToJsonSchema.Generators/Models/InterfaceData.cs
@@ -4,4 +4,5 @@ public readonly record struct InterfaceData(
     string Namespace,
     string Name,
     bool GoogleFunctionTool,
+    bool MeaiFunctionTool,
     IReadOnlyCollection<MethodData> Methods);
diff --git a/src/libs/CSharpToJsonSchema.Generators/Sources.Method.Calls.cs b/src/libs/CSharpToJsonSchema.Generators/Sources.Method.Calls.cs
@@ -17,6 +17,14 @@ public static string GenerateFunctionCalls(InterfaceData @interface)
 
 namespace {@interface.Namespace}
 {{
+
+
+
+
+
+
+
+
 {@interface.Methods.Select(static method => $@"
     public class {method.Name}Args
     {{         
diff --git a/src/libs/CSharpToJsonSchema.Generators/Sources.Method.GoogleFunctionTools.cs b/src/libs/CSharpToJsonSchema.Generators/Sources.Method.GoogleFunctionTools.cs
@@ -42,9 +42,9 @@ public static string GenerateGoogleFunctionToolForInterface(InterfaceData @inter
 
 namespace {@interface.Namespace}
 {{
-    public partial class {extensionsClassName}
+    public static partial class {extensionsClassName}
     {{    
-        public global::GenerativeAI.Core.IFunctionTool AsGoogleFunctionTool(this {@interface.Name} service)
+        public static global::GenerativeAI.Core.IFunctionTool AsGoogleFunctionTool(this {@interface.Name} service)
         {{
             return new global::GenerativeAI.Tools.GenericFunctionTool(service.AsTools(), service.AsCalls());
         }}
diff --git a/src/libs/CSharpToJsonSchema.Generators/Sources.Method.MeaiTools.cs b/src/libs/CSharpToJsonSchema.Generators/Sources.Method.MeaiTools.cs
@@ -0,0 +1,69 @@
+﻿using CSharpToJsonSchema.Generators.Models;
+
+namespace CSharpToJsonSchema.Generators;
+
+internal static partial class Sources
+{
+    public static string GenerateMeaiFunctionToolForMethods(InterfaceData @interface)
+    {
+        if(@interface.Methods.Count == 0 || !@interface.MeaiFunctionTool)
+            return string.Empty;
+        var extensionsClassName = @interface.Name;
+       
+        return @$"
+#nullable enable
+
+namespace {@interface.Namespace}
+{{
+    public partial class {extensionsClassName}
+    {{               
+        public static implicit operator global::System.Collections.Generic.List<global::Microsoft.Extensions.AI.AITool>? ({@interface.Namespace}.{extensionsClassName} tools)
+        {{
+            return tools.AsMeaiTools();
+        }}             
+
+        public global::System.Collections.Generic.List<global::Microsoft.Extensions.AI.AITool> AsMeaiTools()
+        {{
+            var lst = new global::System.Collections.Generic.List<global::Microsoft.Extensions.AI.AITool>();
+            var tools = this.AsTools();
+            var calls = this.AsCalls();
+            foreach (var tool in tools)
+            {{
+                var call = calls[tool.Name];
+                lst.Add(new global::CSharpToJsonSchema.MeaiFunction(tool, call));
+            }}
+            return lst;
+        }}
+    }}
+}}";
+    }
+    
+    public static string GenerateMeaiFunctionToolForInterface(InterfaceData @interface)
+    {
+        if(!@interface.MeaiFunctionTool)
+            return string.Empty;
+        var extensionsClassName = @interface.Name.Substring(startIndex: 1) + "Extensions";
+       
+        return @$"
+#nullable enable
+
+namespace {@interface.Namespace}
+{{
+    public partial class {extensionsClassName}
+    {{    
+        public static global::System.Collections.Generic.List<global::Microsoft.Extensions.AI.AITool> AsMeaiTools(this {@interface.Name} service)
+        {{
+            var lst = new global::System.Collections.Generic.List<global::Microsoft.Extensions.AI.AITool>();
+            var tools = service.AsTools();
+            var calls = service.AsCalls();
+            foreach (var tool in tools)
+            {{
+                var call = calls[tool.Name];
+                lst.Add(new global::CSharpToJsonSchema.MeaiFunction(tool, call));
+            }}
+            return lst;
+        }}
+    }}
+}}";
+    }
+}
diff --git a/src/libs/CSharpToJsonSchema/CSharpToJsonSchema.csproj b/src/libs/CSharpToJsonSchema/CSharpToJsonSchema.csproj
@@ -17,6 +17,7 @@
   </PropertyGroup>
   
   <ItemGroup>
+    <PackageReference Include="Microsoft.Extensions.AI.Abstractions" Version="9.3.0-preview.1.25114.11" />
     <PackageReference Include="System.Text.Json" Version="9.0.0" />
   </ItemGroup>
 
diff --git a/src/libs/CSharpToJsonSchema/FunctionToolAttribute.cs b/src/libs/CSharpToJsonSchema/FunctionToolAttribute.cs
@@ -16,4 +16,9 @@ public sealed class FunctionToolAttribute : Attribute
     /// Generate Google Function Tools extensions for Google_GenerativeAI SDK
     /// </summary>
     public bool GoogleFunctionTool { get; set; }
+    
+    /// <summary>
+    /// Generate Microsoft.Extension.AI compatible function tools
+    /// </summary>
+    public bool MeaiFunctionTool { get; set; }
 }
diff --git a/src/libs/CSharpToJsonSchema/GenerateJsonSchemaAttribute.cs b/src/libs/CSharpToJsonSchema/GenerateJsonSchemaAttribute.cs
@@ -17,4 +17,9 @@ public sealed class GenerateJsonSchemaAttribute : Attribute
     /// Generate Google Function Tools extensions for Google_GenerativeAI SDK
     /// </summary>
     public bool GoogleFunctionTool { get; set; }
+    
+    /// <summary>
+    /// Generate Microsoft.Extension.AI compatible function tools
+    /// </summary>
+    public bool MeaiFunctionTool { get; set; }
 }
diff --git a/src/libs/CSharpToJsonSchema/MeaiFunction.cs b/src/libs/CSharpToJsonSchema/MeaiFunction.cs
@@ -0,0 +1,61 @@
+﻿using System.Text.Json;
+using System.Text.Json.Nodes;
+using Microsoft.Extensions.AI;
+
+namespace CSharpToJsonSchema;
+
+public partial class MeaiFunction : AIFunction
+{
+    private readonly Tool _tool;
+    private readonly Func<string, CancellationToken, Task<string>> _call;
+    
+    private JsonElement _jsonSchema;
+    public override JsonElement JsonSchema => _jsonSchema;
+    public override string Name => _tool.Name;
+    public override string Description => _tool.Description;
+    
+    public MeaiFunction(Tool tool, Func<string, CancellationToken, Task<string>> call)
+    {
+        this._tool = tool;
+        this._call = call;
+        
+        _jsonSchema = JsonSerializer.Deserialize<JsonElement>(JsonSerializer.Serialize(tool.Parameters, OpenApiSchemaJsonContext.Default.OpenApiSchema),OpenApiSchemaJsonContext.Default.JsonElement);
+    }
+    protected override async Task<object?> InvokeCoreAsync(IEnumerable<KeyValuePair<string, object?>> arguments,
+        CancellationToken cancellationToken)
+    {
+        try
+        {
+            var json = GetArgsString(arguments);
+
+            var call = await _call(json, cancellationToken);
+            
+            return JsonSerializer.Deserialize(call, OpenApiSchemaJsonContext.Default.JsonElement);
+        }catch(Exception e)
+        {
+            return e.Message;
+        }
+    }
+
+    private string GetArgsString(IEnumerable<KeyValuePair<string, object?>> arguments)
+    {
+        var jsonObject = new JsonObject();
+
+        foreach (var args in arguments)
+        {
+            if (args.Value is JsonElement element)
+            {
+                if(element.ValueKind == JsonValueKind.Array)
+                    jsonObject[args.Key] = JsonArray.Create(element);
+                else if(element.ValueKind == JsonValueKind.Object)
+                    jsonObject[args.Key] = JsonObject.Create(element);
+            }
+            else if (args.Value is JsonNode node)
+            {
+                jsonObject[args.Key] = node;
+            }
+        }
+
+        return jsonObject.ToJsonString();
+    }
+}
diff --git a/src/libs/CSharpToJsonSchema/OpenApiSchemaJsonContext.cs b/src/libs/CSharpToJsonSchema/OpenApiSchemaJsonContext.cs
@@ -7,7 +7,8 @@ namespace CSharpToJsonSchema;
 [JsonSerializable(typeof(IDictionary<string, string>))]
 [JsonSerializable(typeof(Tool))]
 [JsonSerializable(typeof(List<Tool>))]
-[JsonSourceGenerationOptions(NumberHandling = JsonNumberHandling.Strict, DefaultIgnoreCondition = JsonIgnoreCondition.WhenWritingNull)]
+[JsonSerializable(typeof(JsonElement))]
+[JsonSourceGenerationOptions(NumberHandling = JsonNumberHandling.Strict, DefaultIgnoreCondition = JsonIgnoreCondition.WhenWritingNull,WriteIndented = false)]
 public partial class OpenApiSchemaJsonContext:JsonSerializerContext
 {
     
diff --git a/src/tests/CSharpToJsonSchema.MeaiTests/CSharpToJsonSchema.MeaiTests.csproj b/src/tests/CSharpToJsonSchema.MeaiTests/CSharpToJsonSchema.MeaiTests.csproj
@@ -0,0 +1,42 @@
+﻿<Project Sdk="Microsoft.NET.Sdk">
+
+    <PropertyGroup>
+        <TargetFramework>net9.0</TargetFramework>
+    </PropertyGroup>
+
+    <ItemGroup Label="Base packages">
+        <PackageReference Include="GitHubActionsTestLogger" Version="2.4.1">
+            <PrivateAssets>all</PrivateAssets>
+            <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
+        </PackageReference>
+        <PackageReference Include="H.Generators.Tests.Extensions" Version="1.24.2" />
+        <PackageReference Include="H.Resources.Generator" Version="1.8.0">
+            <PrivateAssets>all</PrivateAssets>
+            <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
+        </PackageReference>
+        <PackageReference Include="Microsoft.Extensions.AI" Version="9.3.0-preview.1.25114.11" />
+        <PackageReference Include="Microsoft.Extensions.AI.OpenAI" Version="9.3.0-preview.1.25114.11" />
+        <PackageReference Include="MSTest" Version="3.8.2" />
+        <PackageReference Include="FluentAssertions" Version="8.1.1" />
+    </ItemGroup>    
+
+    <ItemGroup>
+        <PackageReference Include="Microsoft.CodeAnalysis.CSharp.SourceGenerators.Testing.MSTest" Version="1.1.2" />
+        <PackageReference Include="Microsoft.CodeAnalysis.CSharp.Workspaces" Version="4.13.0" />
+        <PackageReference Include="Verify.MSTest" Version="28.13.0" />
+        <PackageReference Include="Verify.SourceGenerators" Version="2.5.0" />
+    </ItemGroup>
+
+    <ItemGroup Label="GlobalUsings">
+        <Using Include="Microsoft.VisualStudio.TestTools.UnitTesting" />
+        <Using Include="FluentAssertions" />
+    </ItemGroup>
+
+    <ItemGroup>
+        <ProjectReference Include="..\..\libs\CSharpToJsonSchema.Generators\CSharpToJsonSchema.Generators.csproj" OutputItemType="analyzer" />
+        <ProjectReference Include="..\..\libs\CSharpToJsonSchema\CSharpToJsonSchema.csproj" />
+    </ItemGroup>
+
+    
+
+</Project>
diff --git a/src/tests/CSharpToJsonSchema.MeaiTests/Meai_Tests.cs b/src/tests/CSharpToJsonSchema.MeaiTests/Meai_Tests.cs
diff --git a/src/tests/CSharpToJsonSchema.MeaiTests/Services/BookService.cs b/src/tests/CSharpToJsonSchema.MeaiTests/Services/BookService.cs
diff --git a/src/tests/CSharpToJsonSchema.MeaiTests/Services/StudentRecordService.cs b/src/tests/CSharpToJsonSchema.MeaiTests/Services/StudentRecordService.cs
diff --git a/src/tests/CSharpToJsonSchema.SnapshotTests/Snapshots/ToolTests.VariousTypes#IVariousTypesTools.g.received.cs b/src/tests/CSharpToJsonSchema.SnapshotTests/Snapshots/ToolTests.VariousTypes#IVariousTypesTools.g.received.cs
diff --git a/src/tests/CSharpToJsonSchema.SnapshotTests/Snapshots/ToolTests.Weather#IWeatherTools.g.received.cs b/src/tests/CSharpToJsonSchema.SnapshotTests/Snapshots/ToolTests.Weather#IWeatherTools.g.received.cs
