feat(config): add rag schema

Signed-off-by: Braden Mars <[email protected]>
CrisisCleanup · May 15, 2024 · 313cabb · 313cabb
1 parent bebd93d
commit 313cabb
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.envrc b/.envrc
diff --git a/packages/config/src/schema.ts b/packages/config/src/schema.ts
@@ -154,6 +154,12 @@ const langchainSchema = z.object({
 	project: z.string().default('crisiscleanup-3-api'),
 })
 
+const ragSchema = z.object({
+	chatModel: z.string().default('gpt-4o'),
+	documentEmbedModel: z.string().default('text-embedding-3-large'),
+	queryEmbedModel: z.string().default('text-embedding-3-small'),
+})
+
 const langchainSecretsSchema = z.object({
 	apiKey: z.string().default('').optional(),
 })
@@ -168,6 +174,7 @@ const apiAppConfigSchema = z.object({
 	connect: connectSchema.default({}),
 	phone: phoneSchema.default({}),
 	langchain: langchainSchema.default({}),
+	rag: ragSchema.default({}),
 })
 export interface ApiAppConfig extends z.infer<typeof apiAppConfigSchema> {}