discourse/discourse-ai
1
0
Fork 0
mirror of https://github.com/discourse/discourse-ai.git synced 2025-06-26 01:22:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

FIX: Correct prompt format for img2text used in our AI Bot PDF Rag pipeline (#1323)

Browse Source
This commit is contained in:
Rafael dos Santos Silva 2025-05-07 16:48:56 -03:00 committed by GitHub
parent f090065405
commit 1b71330cea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lib/utils/image_to_text.rb
View File

@ -127,14 +127,15 @@ class DiscourseAi::Utils::ImageToText
messages = [ messages = [
{ {
type: :user, type: :user,
content: content: [
"The following text was extracted from an image using OCR. Please enhance, correct, and structure this content while maintaining the original text:\n\n#{raw_text}", "The following text was extracted from an image using OCR. Please enhance, correct, and structure this content while maintaining the original text:\n\n#{raw_text}",
upload_ids: [page.id], { upload_id: page.id },
],
}, },
] ]
else else
messages = [ messages = [
{ type: :user, content: "Please OCR the content in the image.", upload_ids: [page.id] }, { type: :user, content: ["Please OCR the content in the image.", { upload_id: page.id }] },
] ]
end end
prompt = DiscourseAi::Completions::Prompt.new(system_message, messages: messages) prompt = DiscourseAi::Completions::Prompt.new(system_message, messages: messages)