40 lines
1.2 KiB
TypeScript
40 lines
1.2 KiB
TypeScript
import { z } from 'zod';
|
|
import { queryFilterSchema } from '@morten-olsen/stash-query-dsl';
|
|
|
|
import { createListResultSchema, queryDSLSchema } from '../../utils/utils.schema.js';
|
|
|
|
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([queryDSLSchema, queryFilterSchema]).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' });
|
|
|
|
type DocumentChunksFindResult = z.infer<typeof documentChunksFindResultSchema>;
|
|
|
|
export type { DocumentChunk, DocumentChunkFilter, DocumentChunksFindResult };
|
|
export { documentChunkSchema, documentChunkFilterSchema, documentChunksFindResultSchema };
|