Dax Raad
33 lines
966 B
TypeScript
33 lines
966 B
TypeScript
import type { StandardSchemaV1 } from "@standard-schema/spec"
|
|
|
|
export namespace Tool {
|
|
interface Metadata {
|
|
[key: string]: any
|
|
}
|
|
export type Context<M extends Metadata = Metadata> = {
|
|
sessionID: string
|
|
messageID: string
|
|
abort: AbortSignal
|
|
metadata(input: { title?: string; metadata?: M }): void
|
|
}
|
|
export interface Info<Parameters extends StandardSchemaV1 = StandardSchemaV1, M extends Metadata = Metadata> {
|
|
id: string
|
|
description: string
|
|
parameters: Parameters
|
|
execute(
|
|
args: StandardSchemaV1.InferOutput<Parameters>,
|
|
ctx: Context,
|
|
): Promise<{
|
|
title: string
|
|
metadata: M
|
|
output: string
|
|
}>
|
|
}
|
|
|
|
export function define<Parameters extends StandardSchemaV1, Result extends Metadata>(
|
|
input: Info<Parameters, Result> | (() => Promise<Info<Parameters, Result>>),
|
|
): () => Promise<Info<Parameters, Result>> {
|
|
return input instanceof Function ? input : async () => input
|
|
}
|
|
}
|