mirror of
https://github.com/langgenius/dify.git
synced 2026-07-22 03:08:33 +08:00
32 lines
1.1 KiB
TypeScript
32 lines
1.1 KiB
TypeScript
import * as z from 'zod'
|
|
|
|
const arrayStringSchemaParttern = z.array(z.string())
|
|
const arrayNumberSchemaParttern = z.array(z.number())
|
|
|
|
// # jsonSchema from https://zod.dev/?id=json-type
|
|
const literalSchema = z.union([z.string(), z.number(), z.boolean(), z.null()])
|
|
type Literal = z.infer<typeof literalSchema>
|
|
type Json = Literal | { [key: string]: Json } | Json[]
|
|
const jsonSchema: z.ZodType<Json> = z.lazy(() =>
|
|
z.union([literalSchema, z.array(jsonSchema), z.record(z.string(), jsonSchema)]),
|
|
)
|
|
const arrayJsonSchema: z.ZodType<Json[]> = z.lazy(() => z.array(jsonSchema))
|
|
|
|
export const validateJSONSchema = (schema: any, type: string) => {
|
|
if (type === 'array[string]') {
|
|
const result = arrayStringSchemaParttern.safeParse(schema)
|
|
return result
|
|
} else if (type === 'array[number]') {
|
|
const result = arrayNumberSchemaParttern.safeParse(schema)
|
|
return result
|
|
} else if (type === 'object') {
|
|
const result = jsonSchema.safeParse(schema)
|
|
return result
|
|
} else if (type === 'array[object]') {
|
|
const result = arrayJsonSchema.safeParse(schema)
|
|
return result
|
|
} else {
|
|
return { success: true } as any
|
|
}
|
|
}
|