diff --git a/DevProxy.Abstractions/LanguageModel/BaseLanguageModelClient.cs b/DevProxy.Abstractions/LanguageModel/BaseLanguageModelClient.cs index 731ab5ca..93a2574a 100644 --- a/DevProxy.Abstractions/LanguageModel/BaseLanguageModelClient.cs +++ b/DevProxy.Abstractions/LanguageModel/BaseLanguageModelClient.cs @@ -83,7 +83,7 @@ public async Task IsEnabledAsync(CancellationToken cancellationToken) return _lmAvailable.Value; } - protected abstract IEnumerable ConvertMessages(ChatMessage[] messages); + protected abstract IEnumerable ConvertMessages(IEnumerable messages); protected abstract Task GenerateChatCompletionCoreAsync(IEnumerable messages, CompletionOptions? options, CancellationToken cancellationToken); @@ -105,8 +105,8 @@ public async Task IsEnabledAsync(CancellationToken cancellationToken) var promptContents = File.ReadAllText(filePath); var prompty = Prompt.FromMarkdown(promptContents); - if (prompty.Prepare(parameters) is not ChatMessage[] promptyMessages || - promptyMessages.Length == 0) + if (prompty.Prepare(parameters) is not IEnumerable promptyMessages || + !promptyMessages.Any()) { Logger.LogError("No messages found in the prompt file: {FilePath}", filePath); return (null, null); diff --git a/DevProxy.Abstractions/LanguageModel/OllamaLanguageModelClient.cs b/DevProxy.Abstractions/LanguageModel/OllamaLanguageModelClient.cs index 917c4a15..055ade6f 100644 --- a/DevProxy.Abstractions/LanguageModel/OllamaLanguageModelClient.cs +++ b/DevProxy.Abstractions/LanguageModel/OllamaLanguageModelClient.cs @@ -79,7 +79,7 @@ public sealed class OllamaLanguageModelClient( } } - protected override IEnumerable ConvertMessages(ChatMessage[] messages) + protected override IEnumerable ConvertMessages(IEnumerable messages) { return messages.Select(m => new OllamaLanguageModelChatCompletionMessage { diff --git a/DevProxy.Abstractions/LanguageModel/OpenAILanguageModelClient.cs b/DevProxy.Abstractions/LanguageModel/OpenAILanguageModelClient.cs index 71af4530..7043f145 100644 --- a/DevProxy.Abstractions/LanguageModel/OpenAILanguageModelClient.cs +++ b/DevProxy.Abstractions/LanguageModel/OpenAILanguageModelClient.cs @@ -87,7 +87,7 @@ public sealed class OpenAILanguageModelClient( } } - protected override IEnumerable ConvertMessages(ChatMessage[] messages) + protected override IEnumerable ConvertMessages(IEnumerable messages) { return messages.Select(m => new OpenAIChatCompletionMessage {