mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-10-31 14:38:37 +00:00
28 lines
756 B
Ruby
28 lines
756 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
RSpec.describe DiscourseAi::AiBot::AnthropicBot do
|
||
|
|
describe "#update_with_delta" do
|
||
|
|
def bot_user
|
||
|
|
User.find(DiscourseAi::AiBot::EntryPoint::GPT4_ID)
|
||
|
|
end
|
||
|
|
|
||
|
|
subject { described_class.new(bot_user) }
|
||
|
|
|
||
|
|
describe "get_delta" do
|
||
|
|
it "can properly remove Assistant prefix" do
|
||
|
|
context = {}
|
||
|
|
reply = +""
|
||
|
|
|
||
|
|
reply << subject.get_delta({ completion: "\n\nAssist" }, context)
|
||
|
|
expect(reply).to eq("")
|
||
|
|
|
||
|
|
reply << subject.get_delta({ completion: "\n\nAssistant: test" }, context)
|
||
|
|
expect(reply).to eq("test")
|
||
|
|
|
||
|
|
reply << subject.get_delta({ completion: "\n\nAssistant: test\nworld" }, context)
|
||
|
|
expect(reply).to eq("test\nworld")
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|