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",
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 << {
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)
@ -56,9 +57,9 @@ module DiscourseAi
def summarize_chunk(chunk_text, opts, &on_partial_blk)
summary_instruction =
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
"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
completion(
@ -76,6 +77,7 @@ module DiscourseAi
You effectively summarise any text and reply ONLY with ONLY the summarized text.
You condense it into a shorter version.
You understand and generate Discourse forum Markdown.
You format the response, including links, using markdown.
TEXT
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 condense it into a shorter version.
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)
The discussion title is: #{opts[:content_title]}.
TEXT
@ -37,7 +38,11 @@ RSpec.describe DiscourseAi::Summarization::Models::OpenAi do
memo += "(#{item[:id]} #{item[:poster]} said: #{item[:text]} "
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
describe "#summarize_in_chunks" do
@ -88,7 +93,7 @@ RSpec.describe DiscourseAi::Summarization::Models::OpenAi do
{
role: "user",
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[
: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")