a4df3b8688
- implement `Prompt` - adjust Tool approach - update readme for tool calling - document usage of `Prompt`
51 lines
1.5 KiB
Ruby
51 lines
1.5 KiB
Ruby
# typed: strict
|
|
# frozen_string_literal: true
|
|
|
|
require "test_helper"
|
|
|
|
module ModelContextProtocol
|
|
class PromptTest < ActiveSupport::TestCase
|
|
test "#template validates arguments" do
|
|
prompt = Prompt.new(
|
|
name: "test_prompt",
|
|
description: "Test prompt",
|
|
arguments: [
|
|
Prompt::Argument.new(name: "test_argument", description: "Test argument", required: true),
|
|
],
|
|
)
|
|
assert_raises(ArgumentError) do
|
|
prompt.template({})
|
|
end
|
|
end
|
|
|
|
test "#template returns a Result with description and messages when arguments are valid" do
|
|
prompt = Prompt.new(
|
|
name: "test_prompt",
|
|
description: "Test prompt",
|
|
arguments: [
|
|
Prompt::Argument.new(name: "test_argument", description: "Test argument", required: true),
|
|
],
|
|
) do |_|
|
|
Prompt::Result.new(
|
|
description: "Hello, world!",
|
|
messages: [
|
|
Prompt::Message.new(role: "user", content: Content::Text.new("Hello, world!")),
|
|
Prompt::Message.new(role: "assistant", content: Content::Text.new("Hello, friend!")),
|
|
],
|
|
)
|
|
end
|
|
|
|
expected = {
|
|
description: "Hello, world!",
|
|
messages: [
|
|
{ role: "user", content: { text: "Hello, world!" } },
|
|
{ role: "assistant", content: { text: "Hello, friend!" } },
|
|
],
|
|
}
|
|
|
|
result = prompt.template({ "test_argument" => "Hello, friend!" })
|
|
|
|
assert_equal expected, result.to_h
|
|
end
|
|
end
|
|
end
|