18 lines
452 B
TypeScript
18 lines
452 B
TypeScript
import { DeepPartial } from 'ai'
|
|
import { z } from 'zod'
|
|
|
|
export const searchSchema = z.object({
|
|
query: z.string().describe('The query to search for'),
|
|
max_results: z
|
|
.number()
|
|
.max(20)
|
|
.default(5)
|
|
.describe('The maximum number of results to return'),
|
|
search_depth: z
|
|
.enum(['basic', 'advanced'])
|
|
.default('basic')
|
|
.describe('The depth of the search')
|
|
})
|
|
|
|
export type PartialInquiry = DeepPartial<typeof searchSchema>
|