aljaz/opencode
1
0
Fork 0
mirror of https://github.com/aljazceru/opencode.git synced 2025-12-30 14:14:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

add tool tests

Browse Source
This commit is contained in:
Dax Raad
2025-05-21 10:30:39 -04:00
parent f0f55bc75f
commit e01afb407c
15 changed files with 306 additions and 244 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
js/src/tool/tool.ts
View File

@@ -4,49 +4,53 @@ import { Log } from "../util/log";
const log = Log.create({ service: "tool" });
export namespace Tool {
export interface Metadata {
properties: Record<string, any>;
export interface Metadata<
Properties extends Record<string, any> = Record<string, any>,
> {
properties: Properties;
time: {
start: number;
end: number;
};
}
export function define<Params, Output>(
input: AITool<Params, { metadata?: any; output: Output }> & {
name: string;
export function define<
Params,
Output extends { metadata?: any; output: any },
Name extends string,
>(
input: AITool<Params, Output> & {
name: Name;
},
) {
return {
[input.name]: tool({
...input,
execute: async (params, opts) => {
log.info("invoking", {
id: opts.toolCallId,
name: input.name,
...params,
return tool({
...input,
execute: async (params, opts) => {
log.info("invoking", {
id: opts.toolCallId,
name: input.name,
...params,
});
try {
const start = Date.now();
const result = await input.execute!(params, opts);
const metadata: Metadata<Output["metadata"]> = {
...result.metadata,
time: {
start,
end: Date.now(),
},
};
return {
metadata,
output: result.output,
};
} catch (e: any) {
log.error("error", {
msg: e.toString(),
});
try {
const start = Date.now();
const result = await input.execute!(params, opts);
const metadata: Metadata = {
properties: result.metadata,
time: {
start,
end: Date.now(),
},
};
return {
metadata,
output: result.output,
};
} catch (e: any) {
log.error("error", {
msg: e.toString(),
});
return "An error occurred: " + e.toString();
}
},
}),
};
return "An error occurred: " + e.toString();
}
},
});
}
}