mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-09-27 00:53:37 +00:00
Keep tool results readable for the model
This commit is contained in:
parent
ad16aa8fb6
commit
38cd0c9f36
@ -1,3 +1,5 @@
|
|||||||
|
using System.Text.Encodings.Web;
|
||||||
|
using System.Text.Json;
|
||||||
using System.Text.Json.Nodes;
|
using System.Text.Json.Nodes;
|
||||||
|
|
||||||
using AIStudio.Provider;
|
using AIStudio.Provider;
|
||||||
@ -7,6 +9,18 @@ namespace AIStudio.Tools.ToolCallingSystem;
|
|||||||
|
|
||||||
public sealed class ToolExecutionResult
|
public sealed class ToolExecutionResult
|
||||||
{
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// How a JSON result is written for the model.
|
||||||
|
/// </summary>
|
||||||
|
/// <remarks>
|
||||||
|
/// The result goes into the request as a string, and the request is serialized once more on its
|
||||||
|
/// way to the provider, so the model reads whatever this escapes as the escape itself. The
|
||||||
|
/// default encoder escapes every character outside ASCII and those HTML treats specially, for
|
||||||
|
/// JSON embedded in a web page, which this never is: a German document would reach the model
|
||||||
|
/// with every umlaut as six characters. The relaxed encoder escapes only what JSON requires.
|
||||||
|
/// </remarks>
|
||||||
|
private static readonly JsonSerializerOptions MODEL_CONTENT_OPTIONS = new() { Encoder = JavaScriptEncoder.UnsafeRelaxedJsonEscaping };
|
||||||
|
|
||||||
public string? TextContent { get; init; }
|
public string? TextContent { get; init; }
|
||||||
|
|
||||||
public JsonNode? JsonContent { get; init; }
|
public JsonNode? JsonContent { get; init; }
|
||||||
@ -29,7 +43,7 @@ public sealed class ToolExecutionResult
|
|||||||
public string ToModelContent()
|
public string ToModelContent()
|
||||||
{
|
{
|
||||||
if (this.JsonContent is not null)
|
if (this.JsonContent is not null)
|
||||||
return this.JsonContent.ToJsonString();
|
return this.JsonContent.ToJsonString(MODEL_CONTENT_OPTIONS);
|
||||||
|
|
||||||
return this.TextContent ?? string.Empty;
|
return this.TextContent ?? string.Empty;
|
||||||
}
|
}
|
||||||
|
|||||||
39
app/Tests/Tools/ToolCalling/ToolExecutionResultTests.cs
Normal file
39
app/Tests/Tools/ToolCalling/ToolExecutionResultTests.cs
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
using System.Text.Json.Nodes;
|
||||||
|
|
||||||
|
using AIStudio.Tools.ToolCallingSystem;
|
||||||
|
|
||||||
|
namespace AIStudio.Tests.Tools.ToolCalling;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Checks what a model reads of a tool result.
|
||||||
|
/// </summary>
|
||||||
|
/// <remarks>
|
||||||
|
/// The result goes into the request as a string, and the request is serialized once more on its way
|
||||||
|
/// to the provider. Whatever the first serialization escapes therefore reaches the model as the
|
||||||
|
/// escape itself: a German document would arrive with every umlaut spelled out as six characters,
|
||||||
|
/// and a piece of code with every angle bracket. That costs tokens, the budget of all tool results
|
||||||
|
/// counts it, and a model quoting a name from it may quote the escape.
|
||||||
|
/// </remarks>
|
||||||
|
[TestFixture]
|
||||||
|
public sealed class ToolExecutionResultTests
|
||||||
|
{
|
||||||
|
[TestCase("Größe der Übersicht")]
|
||||||
|
[TestCase("if (a < b && c > d) return 'x';")]
|
||||||
|
[TestCase("日本語のテキスト")]
|
||||||
|
public void TextReachesTheModelAsWritten(string text)
|
||||||
|
{
|
||||||
|
var result = new ToolExecutionResult { JsonContent = new JsonObject { ["text_content"] = text } };
|
||||||
|
Assert.That(result.ToModelContent(), Does.Contain(text));
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void TheResultStaysValidJson()
|
||||||
|
{
|
||||||
|
const string TEXT = "A \"quoted\" line\nand a backslash \\ at its end.";
|
||||||
|
var result = new ToolExecutionResult { JsonContent = new JsonObject { ["text_content"] = TEXT } };
|
||||||
|
|
||||||
|
var readBack = JsonSerializer.Deserialize<JsonObject>(result.ToModelContent());
|
||||||
|
Assert.That(readBack?["text_content"]?.GetValue<string>(), Is.EqualTo(TEXT), "What JSON itself has to escape, it still escapes.");
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user