Replace AIFunctionParameterMetadata with MethodInfo

src/Libraries/Microsoft.Extensions.AI.Abstractions/Functions/AIFunction.cs

@@ -1,20 +1,95 @@
 // Licensed to the .NET Foundation under one or more agreements.
 // The .NET Foundation licenses this file to you under the MIT license.
 
+using System;
 using System.Collections.Generic;
 using System.Diagnostics;
+using System.Diagnostics.CodeAnalysis;
+using System.Reflection;
+using System.Text.Json;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.Shared.Collections;
+using Microsoft.Shared.Diagnostics;
 
 namespace Microsoft.Extensions.AI;
 
 /// <summary>Represents a function that can be described to an AI service and invoked.</summary>
 [DebuggerDisplay("{DebuggerDisplay,nq}")]
 public abstract class AIFunction : AITool
 {
-    /// <summary>Gets metadata describing the function.</summary>
-    public abstract AIFunctionMetadata Metadata { get; }
+    /// <summary>The description of the function.</summary>
+    private readonly string _description = string.Empty;
+
+    /// <summary>The JSON schema describing the function and its input parameters.</summary>
+    private readonly JsonElement _jsonSchema = AIJsonUtilities.DefaultJsonSchema;
+
+    /// <summary>Optional additional properties in addition to the named properties already available on this class.</summary>
+    private readonly IReadOnlyDictionary<string, object?> _additionalProperties = EmptyReadOnlyDictionary<string, object?>.Instance;
+
+    /// <summary>Gets the name of the function.</summary>
+    public abstract string Name { get; }
+
+    /// <summary>Gets a description of the function, suitable for use in describing the purpose to a model.</summary>
+    [AllowNull]
+    public virtual string Description
+    {
+        get => _description;
+        init => _description = value ?? string.Empty;
+    }
+
+    /// <summary>
+    /// Gets the underlying <see cref="MethodInfo"/> that this <see cref="AIFunction"/> might be wrapping.
+    /// </summary>
+    /// <remarks>
+    /// Provides additional metadata on the function and its signature. Implementations not wrapping .NET methods may return <see langword="null"/>.
+    /// </remarks>
+    public virtual MethodInfo? UnderlyingMethod { get; init; }
+
+    /// <summary>Gets a JSON Schema describing the function and its input parameters.</summary>
+    /// <remarks>
+    /// <para>
+    /// When specified, declares a self-contained JSON schema document that describes the function and its input parameters.
+    /// A simple example of a JSON schema for a function that adds two numbers together is shown below:
+    /// </para>
+    /// <code>
+    /// {
+    ///   "title" : "addNumbers",
+    ///   "description": "A simple function that adds two numbers together.",
+    ///   "type": "object",
+    ///   "properties": {
+    ///     "a" : { "type": "number" },
+    ///     "b" : { "type": "number", "default": 1 }
+    ///   }, 
+    ///   "required" : ["a"]
+    /// }
+    /// </code>
+    /// <para>
+    /// The metadata present in the schema document plays an important role in guiding AI function invocation.
+    /// </para>
+    /// <para>
+    /// When no schema is specified, consuming chat clients should assume the "{}" or "true" schema, indicating that any JSON input is admissible.
+    /// </para>
+    /// </remarks>
+    public virtual JsonElement JsonSchema
+    {
+        get => _jsonSchema;
+        init
+        {
+            AIJsonUtilities.ValidateSchemaDocument(value);
+            _jsonSchema = value;
+        }
+    }
+
+    /// <summary>Gets any additional properties associated with the function.</summary>
+    public virtual IReadOnlyDictionary<string, object?> AdditionalProperties
+    {
+        get => _additionalProperties;
+        init => _additionalProperties = Throw.IfNull(value);
+    }
+
+    /// <summary>Gets a <see cref="JsonSerializerOptions"/> that can be used to marshal function parameters.</summary>
+    public virtual JsonSerializerOptions? JsonSerializerOptions { get; init; }
 
     /// <summary>Invokes the <see cref="AIFunction"/> and returns its result.</summary>
     /// <param name="arguments">The arguments to pass to the function's invocation.</param>
@@ -30,7 +105,7 @@ public abstract class AIFunction : AITool
     }
 
     /// <inheritdoc/>
-    public override string ToString() => Metadata.Name;
+    public override string ToString() => Name;
 
     /// <summary>Invokes the <see cref="AIFunction"/> and returns its result.</summary>
     /// <param name="arguments">The arguments to pass to the function's invocation.</param>
@@ -42,8 +117,5 @@ public abstract class AIFunction : AITool
 
     /// <summary>Gets the string to display in the debugger for this instance.</summary>
     [DebuggerBrowsable(DebuggerBrowsableState.Never)]
-    private string DebuggerDisplay =>
-        string.IsNullOrWhiteSpace(Metadata.Description) ?
-            Metadata.Name :
-            $"{Metadata.Name} ({Metadata.Description})";
+    private string DebuggerDisplay => string.IsNullOrWhiteSpace(Description) ? Name : $"{Name} ({Description})";
 }