35 lines
1.0 KiB
Ruby
35 lines
1.0 KiB
Ruby
|
#frozen_string_literal: true
|
||
|
|
|
||
|
|
RSpec.describe DiscourseAi::AiBot::Tools::CreateArtifact do
|
||
|
|
fab!(:llm_model)
|
||
|
|
let(:bot_user) { DiscourseAi::AiBot::EntryPoint.find_user_from_model(llm_model.name) }
|
||
|
|
let(:llm) { DiscourseAi::Completions::Llm.proxy("custom:#{llm_model.id}") }
|
||
|
|
|
||
|
|
before { SiteSetting.ai_bot_enabled = true }
|
||
|
|
|
||
|
|
describe "#process" do
|
||
|
|
it "can correctly handle partial updates" do
|
||
|
|
tool = described_class.new({}, bot_user: bot_user, llm: llm)
|
||
|
|
|
||
|
|
tool.parameters = { css: "a { }" }
|
||
|
|
tool.partial_invoke
|
||
|
|
|
||
|
|
expect(tool.custom_raw).to eq("### CSS\n\n```css\na { }\n```")
|
||
|
|
|
||
|
|
tool.parameters = { css: "a { }", html_body: "hello" }
|
||
|
|
tool.partial_invoke
|
||
|
|
|
||
|
|
expect(tool.custom_raw).to eq(
|
||
|
|
"### CSS\n\n```css\na { }\n```\n\n### HTML\n\n```html\nhello\n```",
|
||
|
|
)
|
||
|
|
|
||
|
|
tool.parameters = { css: "a { }", html_body: "hello world" }
|
||
|
|
tool.partial_invoke
|
||
|
|
|
||
|
|
expect(tool.custom_raw).to eq(
|
||
|
|
"### CSS\n\n```css\na { }\n```\n\n### HTML\n\n```html\nhello world\n```",
|
||
|
|
)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|