From c73dde43cc816b8bf8eee74051d4fdac402cc710 Mon Sep 17 00:00:00 2001
From: MohamedBassem <me@mbassem.com>
Date: Tue, 2 Apr 2024 11:08:53 +0100
Subject: feat(workers): Allow configuring the language in which the tags are
 generated. Fixes #68

---
 packages/shared/config.ts | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'packages/shared/config.ts')

diff --git a/packages/shared/config.ts b/packages/shared/config.ts
index 21bcba5a..c1cc371a 100644
--- a/packages/shared/config.ts
+++ b/packages/shared/config.ts
@@ -28,6 +28,7 @@ const allEnv = z.object({
   DEMO_MODE_PASSWORD: z.string().optional(),
   DATA_DIR: z.string().default(""),
   MAX_ASSET_SIZE_MB: z.coerce.number().default(4),
+  INFERENCE_LANG: z.string().default("english"),
 });
 
 const serverConfigSchema = allEnv.transform((val) => {
@@ -42,6 +43,7 @@ const serverConfigSchema = allEnv.transform((val) => {
       ollamaBaseUrl: val.OLLAMA_BASE_URL,
       textModel: val.INFERENCE_TEXT_MODEL,
       imageModel: val.INFERENCE_IMAGE_MODEL,
+      inferredTagLang: val.INFERENCE_LANG,
     },
     bullMQ: {
       redisHost: val.REDIS_HOST,
-- 
cgit v1.2.3-70-g09d2