Add ITool + ToolRegistry + GetCurrentTimeTool
This commit is contained in:
@@ -0,0 +1,29 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
using backend.Tools;
|
||||||
|
using FluentAssertions;
|
||||||
|
using Xunit;
|
||||||
|
|
||||||
|
namespace backend.tests;
|
||||||
|
|
||||||
|
public class GetCurrentTimeToolTests
|
||||||
|
{
|
||||||
|
[Fact]
|
||||||
|
public async Task Returns_now_field_with_iso_utc_timestamp()
|
||||||
|
{
|
||||||
|
var tool = new GetCurrentTimeTool();
|
||||||
|
var args = JsonDocument.Parse("{}").RootElement;
|
||||||
|
var ctx = new DeviceContext(Guid.NewGuid(), Guid.NewGuid(), new NoopChannel());
|
||||||
|
|
||||||
|
var res = await tool.ExecuteAsync(new ToolInvocation("call_1", args), ctx, default);
|
||||||
|
|
||||||
|
res.Output.TryGetProperty("now", out var now).Should().BeTrue();
|
||||||
|
now.GetString().Should().MatchRegex(@"^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}");
|
||||||
|
now.GetString()!.Should().EndWith("Z");
|
||||||
|
}
|
||||||
|
|
||||||
|
private class NoopChannel : IDeviceChannel
|
||||||
|
{
|
||||||
|
public Task<JsonElement> CallPiToolAsync(string name, JsonElement args, CancellationToken ct)
|
||||||
|
=> throw new NotSupportedException();
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,29 @@
|
|||||||
|
using backend.Tools;
|
||||||
|
using FluentAssertions;
|
||||||
|
using Xunit;
|
||||||
|
|
||||||
|
namespace backend.tests;
|
||||||
|
|
||||||
|
public class ToolRegistryTests
|
||||||
|
{
|
||||||
|
private readonly ITool[] _all = new ITool[]
|
||||||
|
{
|
||||||
|
new GetCurrentTimeTool(),
|
||||||
|
};
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public void Get_returns_tool_by_exact_name()
|
||||||
|
{
|
||||||
|
var reg = new ToolRegistry(_all);
|
||||||
|
reg.Get("get_current_time").Should().NotBeNull();
|
||||||
|
reg.Get("does_not_exist").Should().BeNull();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public void Filter_returns_only_enabled_tools_preserving_order()
|
||||||
|
{
|
||||||
|
var reg = new ToolRegistry(_all);
|
||||||
|
var enabled = new HashSet<string> { "get_current_time", "ghost" };
|
||||||
|
reg.EnabledFor(enabled).Select(t => t.Name).Should().Equal("get_current_time");
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,20 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
|
||||||
|
namespace backend.Tools;
|
||||||
|
|
||||||
|
public class GetCurrentTimeTool : ITool
|
||||||
|
{
|
||||||
|
public string Name => "get_current_time";
|
||||||
|
public string Description => "Returns the current UTC time as an ISO-8601 string.";
|
||||||
|
public bool RunsDuringResponse => false;
|
||||||
|
|
||||||
|
public JsonElement ParameterSchema { get; } =
|
||||||
|
JsonDocument.Parse("""{"type":"object","properties":{},"required":[]}""").RootElement;
|
||||||
|
|
||||||
|
public Task<ToolResult> ExecuteAsync(ToolInvocation invocation, DeviceContext device, CancellationToken ct)
|
||||||
|
{
|
||||||
|
var now = DateTimeOffset.UtcNow.ToString("yyyy-MM-ddTHH:mm:ssZ");
|
||||||
|
var output = JsonDocument.Parse($$"""{"now":"{{now}}"}""").RootElement;
|
||||||
|
return Task.FromResult(new ToolResult(output));
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,8 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
|
||||||
|
namespace backend.Tools;
|
||||||
|
|
||||||
|
public interface IDeviceChannel
|
||||||
|
{
|
||||||
|
Task<JsonElement> CallPiToolAsync(string name, JsonElement args, CancellationToken ct);
|
||||||
|
}
|
||||||
@@ -0,0 +1,19 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
|
||||||
|
namespace backend.Tools;
|
||||||
|
|
||||||
|
public record DeviceContext(Guid DeviceId, Guid ConversationId, IDeviceChannel Channel);
|
||||||
|
public record ToolInvocation(string CallId, JsonElement Arguments);
|
||||||
|
public record ToolResult(JsonElement Output);
|
||||||
|
|
||||||
|
public interface ITool
|
||||||
|
{
|
||||||
|
string Name { get; }
|
||||||
|
string Description { get; }
|
||||||
|
JsonElement ParameterSchema { get; }
|
||||||
|
bool RunsDuringResponse { get; }
|
||||||
|
Task<ToolResult> ExecuteAsync(
|
||||||
|
ToolInvocation invocation,
|
||||||
|
DeviceContext device,
|
||||||
|
CancellationToken ct);
|
||||||
|
}
|
||||||
@@ -0,0 +1,20 @@
|
|||||||
|
namespace backend.Tools;
|
||||||
|
|
||||||
|
public class ToolRegistry
|
||||||
|
{
|
||||||
|
private readonly Dictionary<string, ITool> _byName;
|
||||||
|
private readonly List<ITool> _all;
|
||||||
|
|
||||||
|
public ToolRegistry(IEnumerable<ITool> tools)
|
||||||
|
{
|
||||||
|
_all = tools.ToList();
|
||||||
|
_byName = _all.ToDictionary(t => t.Name);
|
||||||
|
}
|
||||||
|
|
||||||
|
public IReadOnlyList<ITool> All => _all;
|
||||||
|
|
||||||
|
public ITool? Get(string name) => _byName.GetValueOrDefault(name);
|
||||||
|
|
||||||
|
public IEnumerable<ITool> EnabledFor(IReadOnlySet<string> enabled)
|
||||||
|
=> _all.Where(t => enabled.Contains(t.Name));
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user