Add hook after link creation to maybe summarize link

backend/lynx/lynx.go

@@ -10,7 +10,10 @@ import (
 	"github.com/pocketbase/pocketbase/apis"
 	"github.com/pocketbase/pocketbase/core"
 	"github.com/pocketbase/pocketbase/models"
+	"github.com/pocketbase/pocketbase/tools/routine"
 	"github.com/pocketbase/pocketbase/tools/security"
+
+	"main/lynx/summarizer"
 )
 
 var parseUrlHandlerFunc = handleParseURL
@@ -73,6 +76,13 @@ func InitializePocketbase(app core.App) {
 
 		return nil
 	})
+
+	app.OnModelAfterCreate("links").Add(func(e *core.ModelEvent) error {
+		routine.FireAndForget(func() {
+			summarizer.MaybeSummarizeLink(app, e.Model.GetId())
+		})
+		return nil
+	})
 }
 
 func handleGenerateAPIKey(app core.App, c echo.Context) error {

backend/lynx/summarizer/summarizer.go

@@ -1 +1,84 @@
-package summarizer
+package summarizer
+
+import (
+	"github.com/pocketbase/dbx"
+	"github.com/pocketbase/pocketbase/core"
+)
+
+func MaybeSummarizeLink(app core.App, linkID string) {
+	logger := app.Logger().With("action", "summarizeLink", "linkID", linkID)
+
+	link, err := app.Dao().FindRecordById("links", linkID)
+	if err != nil {
+		logger.Error("Summarization failed, failed to find link", "error", err)
+		return
+	}
+
+	if link.GetString("summary") != "" {
+		logger.Info("Summarization skipped, link already has a summary")
+		return
+	}
+
+	userID := link.GetString("user")
+	userSettings, err := app.Dao().FindFirstRecordByFilter("user_settings", "user = {:user}",
+		dbx.Params{
+			"user": userID,
+		})
+	if err != nil {
+		logger.Info("Summarization skipped, user_settings not found", "userID", userID)
+		return
+	}
+
+	if !userSettings.GetBool("automatically_summarize_new_links") {
+		logger.Info("Summarization skipped, Automatic summarization is disabled for user", "userID", userID)
+		return
+	}
+
+	summarizationModel := userSettings.GetString("summarization_model")
+	if summarizationModel == "" {
+		logger.Info("Summarization skipped, summarization model not set for user", "userID", userID)
+		return
+	}
+
+	var summary string
+	var apiKey string
+
+	switch summarizationModel {
+	case string(GPT4o), string(GPT4Mini):
+		apiKey = userSettings.GetString("openai_api_key")
+		if apiKey == "" {
+			logger.Error("Summarization failed, OpenAI API key not set for user", "userID", userID)
+			return
+		}
+
+		summarizer := NewOpenAISummarizer()
+		summary, err = summarizer.SummarizeText(link.GetString("raw_text_content"), apiKey, OpenAIModel(summarizationModel))
+
+	case string(Claude3Opus), string(Claude3Sonnet), string(Claude3Haiku):
+		apiKey = userSettings.GetString("anthropic_api_key")
+		if apiKey == "" {
+			logger.Error("Summarization failed, Anthropic API key not set for user", "userID", userID)
+			return
+		}
+
+		summarizer := NewAnthropicSummarizer()
+		summary, err = summarizer.SummarizeText(link.GetString("raw_text_content"), apiKey, AnthropicModel(summarizationModel))
+
+	default:
+		logger.Error("Summarization failed, Unsupported summarization model", "model", summarizationModel)
+		return
+	}
+
+	if err != nil {
+		logger.Error("Summarization failed", "error", err)
+		return
+	}
+
+	link.Set("summary", summary)
+	if err := app.Dao().SaveRecord(link); err != nil {
+		logger.Error("Summarization failed, failed to save updated link", "error", err)
+		return
+	}
+
+	logger.Info("Successfully summarized link", "model", summarizationModel)
+}