aboutsummaryrefslogtreecommitdiffstats
path: root/packages/shared
diff options
context:
space:
mode:
Diffstat (limited to 'packages/shared')
-rw-r--r--packages/shared/config.ts4
-rw-r--r--packages/shared/queues.ts5
-rw-r--r--packages/shared/types/bookmarks.ts1
3 files changed, 8 insertions, 2 deletions
diff --git a/packages/shared/config.ts b/packages/shared/config.ts
index 046583c6..a4548348 100644
--- a/packages/shared/config.ts
+++ b/packages/shared/config.ts
@@ -40,6 +40,8 @@ const allEnv = z.object({
INFERENCE_OUTPUT_SCHEMA: z
.enum(["structured", "json", "plain"])
.default("structured"),
+ INFERENCE_ENABLE_AUTO_TAGGING: stringBool("true"),
+ INFERENCE_ENABLE_AUTO_SUMMARIZATION: stringBool("false"),
OCR_CACHE_DIR: z.string().optional(),
OCR_LANGS: z
.string()
@@ -120,6 +122,8 @@ const serverConfigSchema = allEnv.transform((val) => {
? ("structured" as const)
: ("plain" as const)
: val.INFERENCE_OUTPUT_SCHEMA,
+ enableAutoTagging: val.INFERENCE_ENABLE_AUTO_TAGGING,
+ enableAutoSummarization: val.INFERENCE_ENABLE_AUTO_SUMMARIZATION,
},
embedding: {
textModel: val.EMBEDDING_TEXT_MODEL,
diff --git a/packages/shared/queues.ts b/packages/shared/queues.ts
index 571df568..bbf69428 100644
--- a/packages/shared/queues.ts
+++ b/packages/shared/queues.ts
@@ -32,9 +32,10 @@ export const LinkCrawlerQueue = new SqliteQueue<ZCrawlLinkRequest>(
},
);
-// OpenAI Worker
+// Inference Worker
export const zOpenAIRequestSchema = z.object({
bookmarkId: z.string(),
+ type: z.enum(["summarize", "tag"]).default("tag"),
});
export type ZOpenAIRequest = z.infer<typeof zOpenAIRequestSchema>;
@@ -195,7 +196,7 @@ export async function triggerWebhook(
});
}
-// RuleEgine worker
+// RuleEngine worker
export const zRuleEngineRequestSchema = z.object({
bookmarkId: z.string(),
events: z.array(zRuleEngineEventSchema),
diff --git a/packages/shared/types/bookmarks.ts b/packages/shared/types/bookmarks.ts
index 709fd431..5fe77278 100644
--- a/packages/shared/types/bookmarks.ts
+++ b/packages/shared/types/bookmarks.ts
@@ -87,6 +87,7 @@ export const zBareBookmarkSchema = z.object({
archived: z.boolean(),
favourited: z.boolean(),
taggingStatus: z.enum(["success", "failure", "pending"]).nullable(),
+ summarizationStatus: z.enum(["success", "failure", "pending"]).nullable(),
note: z.string().nullish(),
summary: z.string().nullish(),
});