57 lines
1.7 KiB
Ruby
57 lines
1.7 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
class SummarizationStubs
|
||
|
|
class << self
|
||
|
|
def test_summary
|
||
|
|
"This is a summary"
|
||
|
|
end
|
||
|
|
|
||
|
|
def openai_response(content)
|
||
|
|
{
|
||
|
|
id: "chatcmpl-6sZfAb30Rnv9Q7ufzFwvQsMpjZh8S",
|
||
|
|
object: "chat.completion",
|
||
|
|
created: 1_678_464_820,
|
||
|
|
model: "gpt-3.5-turbo-0301",
|
||
|
|
usage: {
|
||
|
|
prompt_tokens: 337,
|
||
|
|
completion_tokens: 162,
|
||
|
|
total_tokens: 499,
|
||
|
|
},
|
||
|
|
choices: [
|
||
|
|
{ message: { role: "assistant", content: content }, finish_reason: "stop", index: 0 },
|
||
|
|
],
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
def openai_chat_summarization_stub(chat_messages)
|
||
|
|
prompt_messages =
|
||
|
|
chat_messages
|
||
|
|
.sort_by(&:created_at)
|
||
|
|
.map { |m| "#{m.user.username_lower}: #{m.message}" }
|
||
|
|
.join("\n")
|
||
|
|
|
||
|
|
summary_prompt = [{ role: "system", content: <<~TEXT }]
|
||
|
|
Summarize the following article:\n\n#{prompt_messages}
|
||
|
|
TEXT
|
||
|
|
|
||
|
|
WebMock
|
||
|
|
.stub_request(:post, "https://api.openai.com/v1/chat/completions")
|
||
|
|
.with(body: { model: "gpt-4", messages: summary_prompt }.to_json)
|
||
|
|
.to_return(status: 200, body: JSON.dump(openai_response(test_summary)))
|
||
|
|
end
|
||
|
|
|
||
|
|
def openai_topic_summarization_stub(topic, user)
|
||
|
|
prompt_posts = TopicView.new(topic, user, { filter: "summary" }).posts.map(&:raw).join("\n")
|
||
|
|
|
||
|
|
summary_prompt = [{ role: "system", content: <<~TEXT }]
|
||
|
|
Summarize the following article:\n\n#{prompt_posts}
|
||
|
|
TEXT
|
||
|
|
|
||
|
|
WebMock
|
||
|
|
.stub_request(:post, "https://api.openai.com/v1/chat/completions")
|
||
|
|
.with(body: { model: "gpt-4", messages: summary_prompt }.to_json)
|
||
|
|
.to_return(status: 200, body: JSON.dump(openai_response(test_summary)))
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|