From b8bd7d7eb27aaaadae728599f64a0874f66196ea Mon Sep 17 00:00:00 2001
From: Mohamed Bassem <me@mbassem.com>
Date: Sun, 12 Jan 2025 15:31:30 +0000
Subject: feat: Support customizing the summarization prompt. Fixes #731

---
 packages/trpc/routers/bookmarks.ts | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'packages/trpc')

diff --git a/packages/trpc/routers/bookmarks.ts b/packages/trpc/routers/bookmarks.ts
index 026bd322..f7f8e6b7 100644
--- a/packages/trpc/routers/bookmarks.ts
+++ b/packages/trpc/routers/bookmarks.ts
@@ -19,6 +19,7 @@ import {
   bookmarksInLists,
   bookmarkTags,
   bookmarkTexts,
+  customPrompts,
   rssFeedImportsTable,
   tagsOnBookmarks,
 } from "@hoarder/db/schema";
@@ -1191,8 +1192,19 @@ Description: ${bookmark.description ?? ""}
 Content: ${bookmark.content ?? ""}
 `;
 
+      const prompts = await ctx.db.query.customPrompts.findMany({
+        where: and(
+          eq(customPrompts.userId, ctx.user.id),
+          eq(customPrompts.appliesTo, "summary"),
+        ),
+        columns: {
+          text: true,
+        },
+      });
+
       const summaryPrompt = buildSummaryPrompt(
         serverConfig.inference.inferredTagLang,
+        prompts.map((p) => p.text),
         bookmarkDetails,
         serverConfig.inference.contextLength,
       );
-- 
cgit v1.2.3-70-g09d2