feat: Added M.E.A.I. AIFunction Generation support (#24)

* fix: Json Schema Generation, used System.Text.Json JsonSchema generation.

* fix: warning suppressions

* feat: Added FunctionTool attribute, which can be used to convert individual methods to Tools.

* feat: Added GoogleFunctionTool optional parameters in GenerateJsonSchemaAttribute.cs and FunctionToolAttribute.cs

* Updated README.md

* fix: schema generation

* removed uncessary variable

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

* Simplify description retrieval in SchemaSubsetHelper.

Replaced direct dictionary access with TryGetValue to prevent potential KeyNotFound exceptions. This ensures safer and more robust handling of missing keys when fetching "mainFunction_Desc".

* Refactor MeaiFunction.cs for improved readability

Removed unnecessary try-catch block around async call invocation and adjusted formatting for better code clarity. Minor whitespace and indentation improvements were also made to enhance consistency and maintainability.

* Refactor MeaiFunction to enhance clarity and functionality.

Added detailed XML documentation for MeaiFunction class and its members, improving code readability and maintainability. Introduced support for strict mode and updated method implementations to enhance functionality. These changes provide better structure and guidance for future development.

* Update GoogleFunctionTool type references in generator

Replaced GenericFunctionTool with GoogleFunctionTool to align with the updated namespace and class structure. This ensures compatibility with the latest library changes and maintains consistency in functionality.

---------

Co-authored-by: Gunpal Jain <[email protected]>

Author: gunpal5

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}

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)

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(

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);

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
     {{         

src/libs/CSharpToJsonSchema.Generators/Sources.Method.GoogleFunctionTools.cs

@@ -18,12 +18,12 @@ namespace {@interface.Namespace}
 {{
     public partial class {extensionsClassName}
     {{               
-        public static implicit operator global::GenerativeAI.Tools.GenericFunctionTool ({@interface.Namespace}.{extensionsClassName} tools)
+        public static implicit operator global::GenerativeAI.Core.GoogleFunctionTool ({@interface.Namespace}.{extensionsClassName} tools)
         {{
             return tools.AsGoogleFunctionTool();
         }}   
 
-        public global::GenerativeAI.Tools.GenericFunctionTool AsGoogleFunctionTool()
+        public global::GenerativeAI.Core.GoogleFunctionTool AsGoogleFunctionTool()
         {{
             return new global::GenerativeAI.Tools.GenericFunctionTool(this.AsTools(), this.AsCalls());
         }}
@@ -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());
         }}

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;
+        }}
+    }}
+}}";
+    }
+}

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>
 

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; }
 }

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; }
 }