aboutsummaryrefslogtreecommitdiffstats
path: root/packages/shared/config.ts
diff options
context:
space:
mode:
authorMohamedBassem <me@mbassem.com>2024-10-12 17:25:01 +0000
committerMohamedBassem <me@mbassem.com>2024-10-12 17:37:42 +0000
commit1b09682685f54f29957163be9b9f9fc2de3b49cc (patch)
tree7f10a7635cf984acd45147c24ec3e1d35798e8ba /packages/shared/config.ts
parentc16173ea0fdbf6cc47b13756c0a77e8399669055 (diff)
downloadkarakeep-1b09682685f54f29957163be9b9f9fc2de3b49cc.tar.zst
feature: Allow customizing the inference's context length
Diffstat (limited to 'packages/shared/config.ts')
-rw-r--r--packages/shared/config.ts2
1 files changed, 2 insertions, 0 deletions
diff --git a/packages/shared/config.ts b/packages/shared/config.ts
index 44b7e26d..325d9ffa 100644
--- a/packages/shared/config.ts
+++ b/packages/shared/config.ts
@@ -24,6 +24,7 @@ const allEnv = z.object({
INFERENCE_JOB_TIMEOUT_SEC: z.coerce.number().default(30),
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),
CRAWLER_HEADLESS_BROWSER: stringBool("true"),
BROWSER_WEB_URL: z.string().url().optional(),
BROWSER_WEBSOCKET_URL: z.string().url().optional(),
@@ -74,6 +75,7 @@ const serverConfigSchema = allEnv.transform((val) => {
textModel: val.INFERENCE_TEXT_MODEL,
imageModel: val.INFERENCE_IMAGE_MODEL,
inferredTagLang: val.INFERENCE_LANG,
+ contextLength: val.INFERENCE_CONTEXT_LENGTH,
},
crawler: {
numWorkers: val.CRAWLER_NUM_WORKERS,