Discource-C
/
discourse-ai
mirror of https://github.com/discourse/discourse-ai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

FEATURE: Try to generate OpenAI Summaries in current language (#146) 

* FEATURE: Try to generate OpenAI Summaries in current language

* lint
This commit is contained in:
Rafael dos Santos Silva 2023-08-21 15:40:32 -03:00 committed by GitHub
parent f9105343a6
commit ea5a443588
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lib/modules/summarization/models/open_ai.rb
View File

@ -26,7 +26,7 @@ module DiscourseAi
{ {
role: "user", role: "user",
content: content:
"Concatenate these disjoint summaries, creating a cohesive narrative:\n#{summaries.join("\n")}", "Concatenate these disjoint summaries, creating a cohesive narrative. Keep the summary in the same language used in the text below.\n#{summaries.join("\n")}",
}, },
] ]
@ -41,7 +41,8 @@ module DiscourseAi
messages << { messages << {
role: "user", role: "user",
content: "Summarize the following in 400 words:\n#{truncated_content}", content:
"Summarize the following in 400 words. Keep the summary in the same language used in the text below.\n#{truncated_content}",
} }
completion(messages, &on_partial_blk) completion(messages, &on_partial_blk)
@ -56,9 +57,9 @@ module DiscourseAi
def summarize_chunk(chunk_text, opts, &on_partial_blk) def summarize_chunk(chunk_text, opts, &on_partial_blk)
summary_instruction = summary_instruction =
if opts[:single_chunk] if opts[:single_chunk]
"Summarize the following forum discussion, creating a cohesive narrative:" "Summarize the following forum discussion, creating a cohesive narrative. Keep the summary in the same language used in the text below."
else else
"Summarize the following in 400 words:" "Summarize the following in 400 words. Keep the summary in the same language used in the text below."
end end
completion( completion(
@ -76,6 +77,7 @@ module DiscourseAi
You effectively summarise any text and reply ONLY with ONLY the summarized text. You effectively summarise any text and reply ONLY with ONLY the summarized text.
You condense it into a shorter version. You condense it into a shorter version.
You understand and generate Discourse forum Markdown. You understand and generate Discourse forum Markdown.
You format the response, including links, using markdown.
TEXT TEXT
if opts[:resource_path] if opts[:resource_path]

11
spec/lib/modules/summarization/models/open_ai_spec.rb
View File

@ -26,6 +26,7 @@ RSpec.describe DiscourseAi::Summarization::Models::OpenAi do
You effectively summarise any text and reply ONLY with ONLY the summarized text. You effectively summarise any text and reply ONLY with ONLY the summarized text.
You condense it into a shorter version. You condense it into a shorter version.
You understand and generate Discourse forum Markdown. You understand and generate Discourse forum Markdown.
You format the response, including links, using markdown.
Try generating links as well the format is #{opts[:resource_path]}. eg: [ref](#{opts[:resource_path]}/77) Try generating links as well the format is #{opts[:resource_path]}. eg: [ref](#{opts[:resource_path]}/77)
The discussion title is: #{opts[:content_title]}. The discussion title is: #{opts[:content_title]}.
TEXT TEXT
@ -37,7 +38,11 @@ RSpec.describe DiscourseAi::Summarization::Models::OpenAi do
memo += "(#{item[:id]} #{item[:poster]} said: #{item[:text]} " memo += "(#{item[:id]} #{item[:poster]} said: #{item[:text]} "
end end
messages << { role: "user", content: "Summarize the following in 400 words:\n#{text}" } messages << {
role: "user",
content:
"Summarize the following in 400 words. Keep the summary in the same language used in the text below.\n#{text}",
}
end end
describe "#summarize_in_chunks" do describe "#summarize_in_chunks" do
@ -88,7 +93,7 @@ RSpec.describe DiscourseAi::Summarization::Models::OpenAi do
{ {
role: "user", role: "user",
content: content:
"Concatenate these disjoint summaries, creating a cohesive narrative:\nsummary 1\nsummary 2", "Concatenate these disjoint summaries, creating a cohesive narrative. Keep the summary in the same language used in the text below.\nsummary 1\nsummary 2",
}, },
] ]
@ -108,7 +113,7 @@ RSpec.describe DiscourseAi::Summarization::Models::OpenAi do
truncated_version.last[ truncated_version.last[
:content :content
] = "Summarize the following in 400 words:\n(1 asd said: This is a" ] = "Summarize the following in 400 words. Keep the summary in the same language used in the text below.\n(1 asd said: This is a"
OpenAiCompletionsInferenceStubs.stub_response(truncated_version, "truncated summary") OpenAiCompletionsInferenceStubs.stub_response(truncated_version, "truncated summary")