48 lines
1.2 KiB
TypeScript
48 lines
1.2 KiB
TypeScript
import z from "zod/v4"
|
|
import type { MessageV2 } from "../session/message-v2"
|
|
|
|
export namespace Tool {
|
|
interface Metadata {
|
|
[key: string]: any
|
|
}
|
|
|
|
export type Context<M extends Metadata = Metadata> = {
|
|
sessionID: string
|
|
messageID: string
|
|
agent: string
|
|
abort: AbortSignal
|
|
callID?: string
|
|
extra?: { [key: string]: any }
|
|
metadata(input: { title?: string; metadata?: M }): void
|
|
}
|
|
export interface Info<Parameters extends z.ZodType = z.ZodType, M extends Metadata = Metadata> {
|
|
id: string
|
|
init: () => Promise<{
|
|
description: string
|
|
parameters: Parameters
|
|
execute(
|
|
args: z.infer<Parameters>,
|
|
ctx: Context,
|
|
): Promise<{
|
|
title: string
|
|
metadata: M
|
|
output: string
|
|
attachments?: MessageV2.FilePart[]
|
|
}>
|
|
}>
|
|
}
|
|
|
|
export function define<Parameters extends z.ZodType, Result extends Metadata>(
|
|
id: string,
|
|
init: Info<Parameters, Result>["init"] | Awaited<ReturnType<Info<Parameters, Result>["init"]>>,
|
|
): Info<Parameters, Result> {
|
|
return {
|
|
id,
|
|
init: async () => {
|
|
if (init instanceof Function) return init()
|
|
return init
|
|
},
|
|
}
|
|
}
|
|
}
|