mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-11-14 21:48:27 +00:00
Added the Model Comparison assistant for blind comparisons of two models (#1009)
Co-authored-by: Thorsten Sommer <SommerEngineering@users.noreply.github.com>
This commit is contained in:
1 parent
12f26445ba
commit
b7255694cd
221 files changed
+19989
-90
No files matched your search
@@ -0,0 +1,113 @@
|
||||
using System.Runtime.CompilerServices;
|
||||
|
||||
using AIStudio.Chat;
|
||||
using AIStudio.Provider;
|
||||
using AIStudio.Settings;
|
||||
using AIStudio.Tools.Security;
|
||||
|
||||
namespace AIStudio.Assistants.ModelComparison;
|
||||
|
||||
/// <summary>
|
||||
/// Connects a provider of the app to the model comparison.
|
||||
/// </summary>
|
||||
public static class ModelComparisonProviderAdapter
|
||||
{
|
||||
/// <summary>
|
||||
/// Creates the participant for a provider.
|
||||
/// </summary>
|
||||
/// <param name="providerSettings">The provider, with the model to ask.</param>
|
||||
/// <param name="settingsManager">The settings the provider needs for its request.</param>
|
||||
/// <param name="component">The component whose confidence requirements the provider has to meet.</param>
|
||||
/// <param name="title">The title of the assistant, as the name of the throwaway chat.</param>
|
||||
/// <returns>The participant.</returns>
|
||||
/// <remarks>
|
||||
/// The provider is built once here, not once per run: it holds nothing but its own
|
||||
/// <c>HttpClient</c>, so every run of a batch reuses the same connection instead of paying for a
|
||||
/// fresh one every time.
|
||||
/// </remarks>
|
||||
public static ModelComparisonParticipant CreateParticipant(AIStudio.Settings.Provider providerSettings, SettingsManager settingsManager, Tools.Components component, string title)
|
||||
{
|
||||
var provider = providerSettings.CreateProvider();
|
||||
return new()
|
||||
{
|
||||
Label = providerSettings.ToString(),
|
||||
Ask = (prompt, token) => StreamAnswer(provider, providerSettings, settingsManager, component, title, prompt, token),
|
||||
};
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Creates the filter which the answers of both models pass before the judge reads them.
|
||||
/// </summary>
|
||||
/// <param name="guard">The prompt injection filter of the app.</param>
|
||||
/// <param name="reportingScope">The batch the answers belong to, so the user hears once, when it is over, what the filter removed.</param>
|
||||
/// <param name="title">The title of the assistant, as the source the report names.</param>
|
||||
/// <returns>The filter, for ModelComparisonRunner.</returns>
|
||||
public static Func<IReadOnlyList<string>, Task<IReadOnlyList<string>>> CreateJudgeInputFilter(PromptInjectionGuardService guard, PromptInjectionReportingScope reportingScope, string title) =>
|
||||
answers => guard.SanitizeAsync([.. answers.Select(answer => new PromptInjectionText(answer, PromptInjectionSource.ModelAnswer(title)))], reportingScope);
|
||||
|
||||
/// <summary>
|
||||
/// Asks the model straight through the provider, without <c>ContentText.CreateFromProviderAsync</c>.
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// Two reasons. The stream of chunks is what the first-token time is measured on, and the
|
||||
/// streaming event of a <c>ContentText</c> only fires every few seconds in the energy saving
|
||||
/// mode. And that method turns a provider which is not allowed into an empty answer without a
|
||||
/// word, which would let the model lose every vote. Here it is an error, so the run reports it
|
||||
/// as a failure of that model.
|
||||
///
|
||||
/// The request is what the model comparison promises: no profile, no system prompt of its own,
|
||||
/// no tools and no data sources, only the prompt.
|
||||
/// </remarks>
|
||||
private static async IAsyncEnumerable<string> StreamAnswer(
|
||||
IProvider provider,
|
||||
AIStudio.Settings.Provider providerSettings,
|
||||
SettingsManager settingsManager,
|
||||
Tools.Components component,
|
||||
string title,
|
||||
string prompt,
|
||||
[EnumeratorCancellation] CancellationToken token)
|
||||
{
|
||||
// Fails for a provider the organization prohibited as well, see SettingsManager.IsProviderUsable:
|
||||
if (!settingsManager.IsProviderConfident(providerSettings, component))
|
||||
throw new InvalidOperationException($"The provider '{providerSettings.InstanceName}' may not be used by the model comparison: the organization prohibited it, or it does not meet the confidence requirements.");
|
||||
|
||||
var chatThread = new ChatThread
|
||||
{
|
||||
IncludeDateTime = false,
|
||||
SelectedProvider = providerSettings.Id,
|
||||
SelectedProfile = Profile.NO_PROFILE.Id,
|
||||
SelectedToolIds = [],
|
||||
SystemPrompt = string.Empty,
|
||||
WorkspaceId = Guid.Empty,
|
||||
ChatId = Guid.NewGuid(),
|
||||
Name = title,
|
||||
Blocks = [],
|
||||
RuntimeComponent = component,
|
||||
RuntimeSelectedToolIds = [],
|
||||
RuntimeToolsAreAssistantManaged = true,
|
||||
};
|
||||
|
||||
chatThread.Blocks.Add(new ContentBlock
|
||||
{
|
||||
Time = DateTimeOffset.Now,
|
||||
ContentType = ContentType.TEXT,
|
||||
Role = ChatRole.USER,
|
||||
Content = new ContentText
|
||||
{
|
||||
Text = prompt,
|
||||
},
|
||||
});
|
||||
|
||||
// The provider writes tool traces into the last block of the AI, so the thread has one:
|
||||
chatThread.Blocks.Add(new ContentBlock
|
||||
{
|
||||
Time = DateTimeOffset.Now,
|
||||
ContentType = ContentType.TEXT,
|
||||
Role = ChatRole.AI,
|
||||
Content = new ContentText(),
|
||||
});
|
||||
|
||||
await foreach (var chunk in provider.StreamChatCompletion(providerSettings.Model, chatThread, settingsManager, token).ConfigureAwait(false))
|
||||
yield return chunk.Content;
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user