fix: re-add missing schemas

old-agentic-v1^2
Philipp Burckhardt 2023-06-19 13:30:54 -04:00
rodzic 3a12d71fce
commit eeb2315879
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: A2C3BCA4F31D1DDD
1 zmienionych plików z 28 dodań i 2 usunięć

Wyświetl plik

@ -1,7 +1,33 @@
import { z } from 'zod'
import * as metaphor from '@/services/metaphor'
import * as types from '@/types'
import { BaseTask } from '@/task'
export const MetaphorInputSchema = z.object({
query: z.string(),
numResults: z.number().optional(),
useQueryExpansion: z.boolean().optional(),
includeDomains: z.array(z.string()).optional(),
excludeDomains: z.array(z.string()).optional(),
startCrawlDate: z.string().optional(),
endCrawlDate: z.string().optional(),
startPublishedDate: z.string().optional(),
endPublishedDate: z.string().optional()
})
export const MetaphorOutputSchema = z.object({
results: z.array(
z.object({
author: z.string().nullable(),
publishedDate: z.string().nullable(),
title: z.string().nullable(),
score: z.number(),
url: z.string()
})
)
})
export class MetaphorSearchTool extends BaseTask<
metaphor.MetaphorSearchInput,
metaphor.MetaphorSearchOutput
@ -21,11 +47,11 @@ export class MetaphorSearchTool extends BaseTask<
}
public override get inputSchema() {
return metaphor.MetaphorSearchInputSchema
return MetaphorInputSchema
}
public override get outputSchema() {
return metaphor.MetaphorSearchOutputSchema
return MetaphorOutputSchema
}
public override get nameForModel(): string {