diff options
Diffstat (limited to 'tests/Agent.Core.Tests')
| -rw-r--r-- | tests/Agent.Core.Tests/PromptAndRuntimeTests.cs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tests/Agent.Core.Tests/PromptAndRuntimeTests.cs b/tests/Agent.Core.Tests/PromptAndRuntimeTests.cs index 3ef5bde..fef5340 100644 --- a/tests/Agent.Core.Tests/PromptAndRuntimeTests.cs +++ b/tests/Agent.Core.Tests/PromptAndRuntimeTests.cs @@ -284,6 +284,8 @@ public sealed class PromptAndRuntimeTests Assert.Equal(AgentTaskState.Completed, completed!.State); Assert.Equal([null, false], ollama.ThinkOverrides); Assert.All(ollama.Temperatures, value => Assert.Equal(0.7, value)); + Assert.Contains(ollama.PlanningMessages[1], message => + message.Role == "system" && message.Content.Contains("Do not call or execute any Tool", StringComparison.Ordinal)); } [Fact] @@ -603,6 +605,7 @@ public sealed class PromptAndRuntimeTests { public List<bool?> ThinkOverrides { get; } = []; public List<double?> Temperatures { get; } = []; + public List<IReadOnlyList<OllamaMessage>> PlanningMessages { get; } = []; public int RunCalls { get; private set; } public Task<OllamaResult> ChatAsync(IReadOnlyList<OllamaMessage> messages, double? temperatureOverride, CancellationToken cancellationToken) => Task.FromResult(new OllamaResult("done", 1, 1)); @@ -610,6 +613,7 @@ public sealed class PromptAndRuntimeTests { ThinkOverrides.Add(thinkOverride); Temperatures.Add(temperatureOverride); + PlanningMessages.Add(messages); string content = ThinkOverrides.Count == 2 && succeedOnRetry ? """{"tasks":[{"description":"analyze","dependencies":[],"budget_class":"4K","optimistic_tokens":10,"most_likely_tokens":20,"pessimistic_tokens":30,"requires_external_effect":false}]}""" : "not-json"; |