Mirrored-Projects/paperless-gpt
1
0
Fork 0
mirror of https://github.com/icereed/paperless-gpt.git synced 2025-03-12 21:08:00 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Make the existing document title available in prompt template. (#158)

Browse source 
This allow a prompt to feed the LLM with the title as determined by
Paperless to improve its suggestion.

This also fixes the case when `getSuggestedTags` or
`getSuggestedCorrespondent` are called, but `GenerateTitles` is off (in
which case an empty title was used).

Fixes #137
This commit is contained in:
Ingmar Stein 2025-01-27 09:10:53 +01:00 committed by GitHub
parent 27b56ee1d3
commit 8b6041a93f
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
app_llm.go
View file

@ -191,7 +191,7 @@ func (app *App) doOCRViaLLM(ctx context.Context, jpegBytes []byte, logger *logru
} }
// getSuggestedTitle generates a suggested title for a document using the LLM // getSuggestedTitle generates a suggested title for a document using the LLM
func (app *App) getSuggestedTitle(ctx context.Context, content string, logger *logrus.Entry) (string, error) { func (app *App) getSuggestedTitle(ctx context.Context, content string, suggestedTitle string, logger *logrus.Entry) (string, error) {
likelyLanguage := getLikelyLanguage() likelyLanguage := getLikelyLanguage()
templateMutex.RLock() templateMutex.RLock()
@ -201,6 +201,7 @@ func (app *App) getSuggestedTitle(ctx context.Context, content string, logger *l
err := titleTemplate.Execute(&promptBuffer, map[string]interface{}{ err := titleTemplate.Execute(&promptBuffer, map[string]interface{}{
"Language": likelyLanguage, "Language": likelyLanguage,
"Content": content, "Content": content,
"Title": suggestedTitle,
}) })
if err != nil { if err != nil {
return "", fmt.Errorf("error executing title template: %v", err) return "", fmt.Errorf("error executing title template: %v", err)
@ -276,12 +277,12 @@ func (app *App) generateDocumentSuggestions(ctx context.Context, suggestionReque
content = content[:5000] content = content[:5000]
} }
var suggestedTitle string suggestedTitle := doc.Title
var suggestedTags []string var suggestedTags []string
var suggestedCorrespondent string var suggestedCorrespondent string
if suggestionRequest.GenerateTitles { if suggestionRequest.GenerateTitles {
suggestedTitle, err = app.getSuggestedTitle(ctx, content, docLogger) suggestedTitle, err = app.getSuggestedTitle(ctx, content, suggestedTitle, docLogger)
if err != nil { if err != nil {
mu.Lock() mu.Lock()
errorsList = append(errorsList, fmt.Errorf("Document %d: %v", documentID, err)) errorsList = append(errorsList, fmt.Errorf("Document %d: %v", documentID, err))