chore: improved schema

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

@@ -1,31 +1,37 @@
 import { z } from 'zod';
 import { queryFilterSchema } from '@morten-olsen/stash-query-dsl';
 
-import { createListResultSchema } from '#root/utils/utils.schema.js';
+import { createListResultSchema, queryDSLSchema } from '#root/utils/utils.schema.js';
 
-const documentChunkSchema = z.object({
-  id: z.string(),
-  owner: z.string(),
-  content: z.string(),
-  metadata: z.unknown(),
-});
+const documentChunkSchema = z
+  .object({
+    id: z.guid(),
+    owner: z.string(),
+    content: z.string(),
+    metadata: z.unknown(),
+  })
+  .meta({ id: 'DocumentChunk' });
 
 type DocumentChunk = z.infer<typeof documentChunkSchema>;
 
-const documentChunkFilterSchema = z.object({
-  limit: z.number().default(20),
-  offset: z.number().default(0),
-  semanticText: z.string().optional(),
-  conditions: z.union([queryFilterSchema, z.string()]).optional(),
-});
+const documentChunkFilterSchema = z
+  .object({
+    limit: z.number().default(20),
+    offset: z.number().default(0),
+    semanticText: z.string().optional(),
+    conditions: z.union([queryDSLSchema, queryFilterSchema]).optional(),
+  })
+  .meta({ id: 'DocumentChunkFilter' });
 
 type DocumentChunkFilter = z.infer<typeof documentChunkFilterSchema>;
 
 const documentChunksFindResultSchema = createListResultSchema(
-  documentChunkSchema.extend({
-    distance: z.number().optional(),
-  }),
-);
+  documentChunkSchema
+    .extend({
+      distance: z.number().optional(),
+    })
+    .meta({ id: 'DocumentChunkWithDistance' }),
+).meta({ id: 'DocumentChunkFindResult' });
 
 type DocumentChunksFindResult = z.infer<typeof documentChunksFindResultSchema>;