From ed3a68807b6351cfae67acc7098a91bf587eba9c Mon Sep 17 00:00:00 2001 From: markwallace-microsoft <127216156+markwallace-microsoft@users.noreply.github.com> Date: Fri, 4 Apr 2025 12:04:29 +0100 Subject: [PATCH] Include loading .yml files --- .../Yaml/Plugins/CreateKernelPluginYamlTests.cs | 2 +- .../Functions/Functions.Yaml/PromptYamlKernelExtensions.cs | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/dotnet/src/Functions/Functions.UnitTests/Yaml/Plugins/CreateKernelPluginYamlTests.cs b/dotnet/src/Functions/Functions.UnitTests/Yaml/Plugins/CreateKernelPluginYamlTests.cs index c8a636e3c7b9..b6db05a3aaf7 100644 --- a/dotnet/src/Functions/Functions.UnitTests/Yaml/Plugins/CreateKernelPluginYamlTests.cs +++ b/dotnet/src/Functions/Functions.UnitTests/Yaml/Plugins/CreateKernelPluginYamlTests.cs @@ -37,7 +37,7 @@ public PromptYamlKernelExtensionsTests() string yamlFile1Path = Path.Combine(plugin1Directory, $"{nameof(YAML)}.yaml"); string yamlFile2Path = Path.Combine(plugin1Directory, $"{nameof(YAMLWithCustomSettings)}.yaml"); - string yamlFile3Path = Path.Combine(plugin2Directory, $"{nameof(YAMLNoExecutionSettings)}.yaml"); + string yamlFile3Path = Path.Combine(plugin2Directory, $"{nameof(YAMLNoExecutionSettings)}.yml"); File.WriteAllText(yamlFile1Path, YAML); File.WriteAllText(yamlFile2Path, YAMLWithCustomSettings); diff --git a/dotnet/src/Functions/Functions.Yaml/PromptYamlKernelExtensions.cs b/dotnet/src/Functions/Functions.Yaml/PromptYamlKernelExtensions.cs index 8a61cfc99ab5..3feccc183e2b 100644 --- a/dotnet/src/Functions/Functions.Yaml/PromptYamlKernelExtensions.cs +++ b/dotnet/src/Functions/Functions.Yaml/PromptYamlKernelExtensions.cs @@ -4,6 +4,7 @@ using System.Collections.Generic; using System.Diagnostics.CodeAnalysis; using System.IO; +using System.Linq; using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.Logging; using Microsoft.Extensions.Logging.Abstractions; @@ -87,6 +88,7 @@ private static KernelPlugin CreatePluginFromPromptDirectoryYaml( IServiceProvider? services = null) { const string YamlFilePattern = "*.yaml"; + const string YmlFilePattern = "*.yml"; Verify.DirectoryExists(pluginDirectory); pluginName ??= new DirectoryInfo(pluginDirectory).Name; @@ -96,7 +98,8 @@ private static KernelPlugin CreatePluginFromPromptDirectoryYaml( var functions = new List(); ILogger logger = loggerFactory.CreateLogger(typeof(Kernel)) ?? NullLogger.Instance; - foreach (string functionFile in Directory.GetFiles(pluginDirectory, YamlFilePattern)) + var yamlFiles = Directory.GetFiles(pluginDirectory, YamlFilePattern).Concat(Directory.GetFiles(pluginDirectory, YmlFilePattern)); + foreach (string functionFile in yamlFiles) { var functionName = Path.GetFileName(functionFile); var functionYaml = File.ReadAllText(functionFile);