feat: improved zod handling

packages/runtime/src/services/document-chunks/document-chunks.schemas.ts

@@ -1,7 +1,7 @@
 import { z } from 'zod';
-import { queryFilterSchema } from '@morten-olsen/stash-query-dsl';
+import { querySchema } from '@morten-olsen/stash-query-dsl';
 
-import { createListResultSchema, queryDSLSchema } from '../../utils/utils.schema.js';
+import { createListResultSchema } from '../../utils/utils.schema.js';
 
 const documentChunkSchema = z
   .object({
@@ -19,21 +19,28 @@ const documentChunkFilterSchema = z
     limit: z.number().default(20),
     offset: z.number().default(0),
     semanticText: z.string().optional(),
-    conditions: z.union([queryDSLSchema, queryFilterSchema]).optional(),
+    conditions: querySchema.optional(),
   })
   .meta({ id: 'DocumentChunkFilter' });
 
 type DocumentChunkFilter = z.infer<typeof documentChunkFilterSchema>;
 
-const documentChunksFindResultSchema = createListResultSchema(
-  documentChunkSchema
-    .extend({
-      distance: z.number().optional(),
-    })
-    .meta({ id: 'DocumentChunkWithDistance' }),
-).meta({ id: 'DocumentChunkFindResult' });
+const documentChunkWithDistanceSchema = documentChunkSchema
+  .extend({
+    distance: z.number().optional(),
+  })
+  .meta({ id: 'DocumentChunkWithDistance' });
+
+const documentChunksFindResultSchema = createListResultSchema(documentChunkWithDistanceSchema).meta({
+  id: 'DocumentChunkFindResult',
+});
 
 type DocumentChunksFindResult = z.infer<typeof documentChunksFindResultSchema>;
 
 export type { DocumentChunk, DocumentChunkFilter, DocumentChunksFindResult };
-export { documentChunkSchema, documentChunkFilterSchema, documentChunksFindResultSchema };
+export {
+  documentChunkSchema,
+  documentChunkFilterSchema,
+  documentChunksFindResultSchema,
+  documentChunkWithDistanceSchema,
+};