fix(schema): make content field optional in messageSchema to allow fl…

…exibility in input feat(schema): add refinement to messageSchema to ensure either content or content_path is provided for validation
0xJord4n · Dec 24, 2024 · a8d31a4 · a8d31a4
1 parent 56b441f
commit a8d31a4
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 7 deletions.
diff --git a/dist/index.js b/dist/index.js
@@ -143,10 +143,15 @@ exports.providerOptionsSchema = zod.object({
     base_url: zod.string().optional(),
     headers: zod.record(zod.string(), zod.string()).optional(),
 });
-exports.messageSchema = zod.object({
+exports.messageSchema = zod
+    .object({
     role: zod.nativeEnum(types_1.Role),
-    content: zod.string(),
+    content: zod.string().optional(),
     content_path: zod.string().optional(),
+})
+    .refine((data) => data.content || data.content_path, {
+    message: "Either content or content_path must be provided",
+    path: ["content", "content_path"],
 });
 exports.inputSchema = zod.object({
     provider: zod.nativeEnum(types_1.Provider),

diff --git a/src/input.ts b/src/input.ts
@@ -7,11 +7,16 @@ export const providerOptionsSchema = zod.object({
   headers: zod.record(zod.string(), zod.string()).optional(),
 });
 
-export const messageSchema = zod.object({
-  role: zod.nativeEnum(Role),
-  content: zod.string(),
-  content_path: zod.string().optional(),
-});
+export const messageSchema = zod
+  .object({
+    role: zod.nativeEnum(Role),
+    content: zod.string().optional(),
+    content_path: zod.string().optional(),
+  })
+  .refine((data) => data.content || data.content_path, {
+    message: "Either content or content_path must be provided",
+    path: ["content", "content_path"],
+  });
 
 export const inputSchema = zod.object({
   provider: zod.nativeEnum(Provider),