Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 17 additions & 1 deletion XTMF2.sln
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@

Microsoft Visual Studio Solution File, Format Version 12.00
# Visual Studio Version 18
VisualStudioVersion = 18.9.12112.369 stable
VisualStudioVersion = 18.9.12112.369
MinimumVisualStudioVersion = 10.0.40219.1
Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "SolutionFiles", "SolutionFiles", "{2C9F1CF3-4FF1-4601-80EB-996F7D771178}"
ProjectSection(SolutionItems) = preProject
Expand All @@ -10,6 +10,7 @@ Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "SolutionFiles", "SolutionFi
.github\workflows\blank.yml = .github\workflows\blank.yml
..\LICENSE = ..\LICENSE
README.md = README.md
src\XTMF2.AI\README.md = src\XTMF2.AI\README.md
EndProjectSection
EndProject
Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "src", "src", "{9040CC77-712E-414D-8453-391F9F348218}"
Expand All @@ -18,6 +19,8 @@ Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "XTMF2", "src\XTMF2\XTMF2.cs
EndProject
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "XTMF2.Interfaces", "src\XTMF2.Interfaces\XTMF2.Interfaces.csproj", "{5E0E170B-5A98-41C0-A9AB-4A0ED28C5CF1}"
EndProject
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "XTMF2.AI", "src\XTMF2.AI\XTMF2.AI.csproj", "{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}"
EndProject
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "XTMF2.RunServer", "src\XTMF2.Client\XTMF2.RunServer.csproj", "{A4FC5ADC-58CA-429A-9626-EF07907AFB28}"
EndProject
Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "tests", "tests", "{77789778-93F5-46A6-9114-CCE855E5919C}"
Expand Down Expand Up @@ -64,6 +67,18 @@ Global
{5E0E170B-5A98-41C0-A9AB-4A0ED28C5CF1}.Release|x64.Build.0 = Release|Any CPU
{5E0E170B-5A98-41C0-A9AB-4A0ED28C5CF1}.Release|x86.ActiveCfg = Release|Any CPU
{5E0E170B-5A98-41C0-A9AB-4A0ED28C5CF1}.Release|x86.Build.0 = Release|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Debug|Any CPU.Build.0 = Debug|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Debug|x64.ActiveCfg = Debug|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Debug|x64.Build.0 = Debug|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Debug|x86.ActiveCfg = Debug|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Debug|x86.Build.0 = Debug|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Release|Any CPU.ActiveCfg = Release|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Release|Any CPU.Build.0 = Release|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Release|x64.ActiveCfg = Release|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Release|x64.Build.0 = Release|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Release|x86.ActiveCfg = Release|Any CPU
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB}.Release|x86.Build.0 = Release|Any CPU
{A4FC5ADC-58CA-429A-9626-EF07907AFB28}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
{A4FC5ADC-58CA-429A-9626-EF07907AFB28}.Debug|Any CPU.Build.0 = Debug|Any CPU
{A4FC5ADC-58CA-429A-9626-EF07907AFB28}.Debug|x64.ActiveCfg = Debug|Any CPU
Expand Down Expand Up @@ -131,6 +146,7 @@ Global
GlobalSection(NestedProjects) = preSolution
{57F8ED33-2002-472E-8F77-37D2C27C0ED0} = {9040CC77-712E-414D-8453-391F9F348218}
{5E0E170B-5A98-41C0-A9AB-4A0ED28C5CF1} = {9040CC77-712E-414D-8453-391F9F348218}
{D1C2B3A4-E5F6-4789-ABCD-0123456789AB} = {9040CC77-712E-414D-8453-391F9F348218}
{A4FC5ADC-58CA-429A-9626-EF07907AFB28} = {9040CC77-712E-414D-8453-391F9F348218}
{627244F8-E275-4001-937C-F2D496E72B91} = {77789778-93F5-46A6-9114-CCE855E5919C}
{A1B2C3D4-E5F6-7890-ABCD-EF1234567890} = {77789778-93F5-46A6-9114-CCE855E5919C}
Expand Down
73 changes: 73 additions & 0 deletions src/XTMF2.AI/AiAssistantService.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,73 @@
using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;

namespace XTMF2.AI;

public sealed class AiAssistantService
{
private readonly AiProviderRegistry _providers;
private readonly IAiActionApplier _actionApplier;

public AiAssistantService(AiProviderRegistry providers, IAiActionApplier actionApplier)
{
_providers = providers ?? throw new ArgumentNullException(nameof(providers));
_actionApplier = actionApplier ?? throw new ArgumentNullException(nameof(actionApplier));
}

public IAsyncEnumerable<AiResponseChunk> ChatAsync(
string providerId,
AiChatRequest request,
CancellationToken cancellationToken = default)
{
ArgumentNullException.ThrowIfNull(request);
return _providers.GetRequired(providerId).ChatAsync(request, cancellationToken);
}

public Task<IReadOnlyList<AiModelInfo>> GetModelsAsync(
string providerId,
CancellationToken cancellationToken = default)
{
return _providers.GetModelsAsync(providerId, cancellationToken);
}

public Task<int?> GetContextSizeAsync(
string providerId,
string modelId,
CancellationToken cancellationToken = default)
{
var provider = _providers.GetRequired(providerId);
return provider is IAiModelContextInfo contextInfo
? contextInfo.GetContextSizeAsync(modelId, cancellationToken)
: Task.FromResult<int?>(null);
}

public async Task<AiActionExecutionResult> ExecuteAsync(
AiActionBatch batch,
bool approvalGranted,
bool destructiveApprovalGranted,
CancellationToken cancellationToken = default)
{
var validation = AiActionValidation.ValidateForExecution(
batch,
approvalGranted,
destructiveApprovalGranted);
if (!validation.IsValid)
{
return AiActionExecutionResult.Failure(validation.Error!);
}

return await _actionApplier.ApplyAsync(batch, cancellationToken).ConfigureAwait(false);
}

public Task<AiActionExecutionResult> ValidateAsync(
AiActionBatch batch,
CancellationToken cancellationToken = default)
{
ArgumentNullException.ThrowIfNull(batch);
return _actionApplier is IAiActionValidator validator
? validator.ValidateAsync(batch, cancellationToken)
: Task.FromResult(AiActionExecutionResult.Success(Array.Empty<string>()));
}
}
14 changes: 14 additions & 0 deletions src/XTMF2.AI/AiJson.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
using System.Text.Json;
using System.Text.Json.Serialization;

namespace XTMF2.AI;

internal static class AiJson
{
internal static readonly JsonSerializerOptions Compact = new()
{
DefaultIgnoreCondition = JsonIgnoreCondition.WhenWritingDefault,
PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
WriteIndented = false
};
}
62 changes: 62 additions & 0 deletions src/XTMF2.AI/AiProviderRegistry.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,62 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;

namespace XTMF2.AI;

public sealed class AiProviderRegistry
{
private readonly Dictionary<string, IAiProvider> _providers = new(StringComparer.OrdinalIgnoreCase);

public IReadOnlyList<AiProviderInfo> Providers =>
_providers.Values
.Select(provider => provider.Info)
.OrderBy(info => info.DisplayName, StringComparer.OrdinalIgnoreCase)
.ToArray();

public void Register(IAiProvider provider)
{
ArgumentNullException.ThrowIfNull(provider);

if (string.IsNullOrWhiteSpace(provider.Info.Id))
{
throw new ArgumentException("An AI provider must have a non-empty id.", nameof(provider));
}

if (!_providers.TryAdd(provider.Info.Id, provider))
{
throw new InvalidOperationException(
$"An AI provider with id '{provider.Info.Id}' is already registered.");
}
}

public bool TryGet(string providerId, out IAiProvider? provider)
{
if (string.IsNullOrWhiteSpace(providerId))
{
provider = null;
return false;
}

return _providers.TryGetValue(providerId, out provider);
}

public IAiProvider GetRequired(string providerId)
{
if (!TryGet(providerId, out var provider))
{
throw new KeyNotFoundException($"AI provider '{providerId}' is not registered.");
}

return provider!;
}

public Task<IReadOnlyList<AiModelInfo>> GetModelsAsync(
string providerId,
CancellationToken cancellationToken = default)
{
return GetRequired(providerId).GetModelsAsync(cancellationToken);
}
}
Loading
Loading