David Blass
112 lines
3.8 KiB
TypeScript
112 lines
3.8 KiB
TypeScript
import { query, type SDKMessage } from "@anthropic-ai/claude-agent-sdk";
|
|
import { log } from "../utils/cli.ts";
|
|
import { instructions } from "./shared.ts";
|
|
import type { Agent } from "./types.ts";
|
|
|
|
export const claude: Agent = {
|
|
run: async ({ prompt, mcpServers, apiKey }) => {
|
|
process.env.ANTHROPIC_API_KEY = apiKey;
|
|
// Create the query with SDK options
|
|
const queryInstance = query({
|
|
prompt: `${instructions}\n\n${prompt}`,
|
|
options: {
|
|
permissionMode: "bypassPermissions",
|
|
mcpServers,
|
|
},
|
|
});
|
|
|
|
// Stream the results
|
|
for await (const message of queryInstance) {
|
|
const handler = messageHandlers[message.type];
|
|
await handler(message as never);
|
|
}
|
|
|
|
return {
|
|
success: true,
|
|
output: "",
|
|
};
|
|
},
|
|
};
|
|
|
|
type SDKMessageType = SDKMessage["type"];
|
|
|
|
type SDKMessageHandler<type extends SDKMessageType = SDKMessageType> = (
|
|
data: Extract<SDKMessage, { type: type }>
|
|
) => void | Promise<void>;
|
|
|
|
type SDKMessageHandlers = {
|
|
[type in SDKMessageType]: SDKMessageHandler<type>;
|
|
};
|
|
|
|
const messageHandlers: SDKMessageHandlers = {
|
|
assistant: (data) => {
|
|
if (data.message?.content) {
|
|
for (const content of data.message.content) {
|
|
if (content.type === "text" && content.text?.trim()) {
|
|
log.box(content.text.trim(), { title: "Claude" });
|
|
} else if (content.type === "tool_use") {
|
|
log.info(`→ ${content.name}`);
|
|
|
|
if (content.input) {
|
|
const input = content.input as any;
|
|
if (input.description) log.info(` └─ ${input.description}`);
|
|
if (input.command) log.info(` └─ command: ${input.command}`);
|
|
if (input.file_path) log.info(` └─ file: ${input.file_path}`);
|
|
if (input.content) {
|
|
const preview =
|
|
input.content.length > 100
|
|
? `${input.content.substring(0, 100)}...`
|
|
: input.content;
|
|
log.info(` └─ content: ${preview}`);
|
|
}
|
|
if (input.query) log.info(` └─ query: ${input.query}`);
|
|
if (input.pattern) log.info(` └─ pattern: ${input.pattern}`);
|
|
if (input.url) log.info(` └─ url: ${input.url}`);
|
|
if (input.edits && Array.isArray(input.edits)) {
|
|
log.info(` └─ edits: ${input.edits.length} changes`);
|
|
input.edits.forEach((edit: any, index: number) => {
|
|
if (edit.file_path) log.info(` ${index + 1}. ${edit.file_path}`);
|
|
});
|
|
}
|
|
if (input.task) log.info(` └─ task: ${input.task}`);
|
|
if (input.bash_command) log.info(` └─ bash_command: ${input.bash_command}`);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
},
|
|
user: (data) => {
|
|
if (data.message?.content) {
|
|
for (const content of data.message.content) {
|
|
if (content.type === "tool_result" && content.is_error) {
|
|
log.warning(`Tool error: ${content.content}`);
|
|
}
|
|
}
|
|
}
|
|
},
|
|
result: async (data) => {
|
|
if (data.subtype === "success") {
|
|
await log.summaryTable([
|
|
[
|
|
{ data: "Cost", header: true },
|
|
{ data: "Input Tokens", header: true },
|
|
{ data: "Output Tokens", header: true },
|
|
],
|
|
[
|
|
`$${data.total_cost_usd?.toFixed(4) || "0.0000"}`,
|
|
String(data.usage?.input_tokens || 0),
|
|
String(data.usage?.output_tokens || 0),
|
|
],
|
|
]);
|
|
} else if (data.subtype === "error_max_turns") {
|
|
log.error(`Max turns reached: ${JSON.stringify(data)}`);
|
|
} else if (data.subtype === "error_during_execution") {
|
|
log.error(`Execution error: ${JSON.stringify(data)}`);
|
|
} else {
|
|
log.error(`Failed: ${JSON.stringify(data)}`);
|
|
}
|
|
},
|
|
system: () => {},
|
|
stream_event: () => {},
|
|
};
|