diff options
| author | Mohamed Bassem <me@mbassem.com> | 2024-10-20 21:06:58 +0000 |
|---|---|---|
| committer | Mohamed Bassem <me@mbassem.com> | 2024-10-20 21:06:58 +0000 |
| commit | 019b5d2f5ea0a78cb6c44be26b1eba60b2a4e88d (patch) | |
| tree | 2b2f99dc9efa90372e277fbc6fe7c3371aafc785 /packages/shared/config.ts | |
| parent | f793646b0daa007137e2b0bb908be0219c9cfbe8 (diff) | |
| download | karakeep-019b5d2f5ea0a78cb6c44be26b1eba60b2a4e88d.tar.zst | |
feature: Add OCR support for images. Fixes #296
Diffstat (limited to 'packages/shared/config.ts')
| -rw-r--r-- | packages/shared/config.ts | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/packages/shared/config.ts b/packages/shared/config.ts index 325d9ffa..41430566 100644 --- a/packages/shared/config.ts +++ b/packages/shared/config.ts @@ -25,6 +25,12 @@ const allEnv = z.object({ INFERENCE_TEXT_MODEL: z.string().default("gpt-4o-mini"), INFERENCE_IMAGE_MODEL: z.string().default("gpt-4o-mini"), INFERENCE_CONTEXT_LENGTH: z.coerce.number().default(2048), + OCR_CACHE_DIR: z.string().optional(), + OCR_LANGS: z + .string() + .default("eng") + .transform((val) => val.split(",")), + OCR_CONFIDENCE_THRESHOLD: z.coerce.number().default(50), CRAWLER_HEADLESS_BROWSER: stringBool("true"), BROWSER_WEB_URL: z.string().url().optional(), BROWSER_WEBSOCKET_URL: z.string().url().optional(), @@ -90,6 +96,11 @@ const serverConfigSchema = allEnv.transform((val) => { fullPageScreenshot: val.CRAWLER_FULL_PAGE_SCREENSHOT, fullPageArchive: val.CRAWLER_FULL_PAGE_ARCHIVE, }, + ocr: { + langs: val.OCR_LANGS, + cacheDir: val.OCR_CACHE_DIR, + confidenceThreshold: val.OCR_CONFIDENCE_THRESHOLD, + }, meilisearch: val.MEILI_ADDR ? { address: val.MEILI_ADDR, |