Compare commits

...

3 Commits

Author SHA1 Message Date
David Blass 9fb6135fd2 bump 2025-10-17 22:27:59 -04:00
David Blass bb78e5f94b update lockfile 2025-10-17 22:27:24 -04:00
David Blass c668578c6f refactor mcp and add instructions prefix 2025-10-17 22:26:24 -04:00
13 changed files with 99 additions and 101 deletions
+20 -41
View File
@@ -3,6 +3,7 @@ import * as core from "@actions/core";
import { createMcpConfig } from "../mcp/config.ts"; import { createMcpConfig } from "../mcp/config.ts";
import { spawn } from "../utils/subprocess.ts"; import { spawn } from "../utils/subprocess.ts";
import { boxString, tableString } from "../utils/table.ts"; import { boxString, tableString } from "../utils/table.ts";
import { instructions } from "./shared.ts";
import type { Agent, AgentConfig, AgentResult } from "./types.ts"; import type { Agent, AgentConfig, AgentResult } from "./types.ts";
/** /**
@@ -10,7 +11,7 @@ import type { Agent, AgentConfig, AgentResult } from "./types.ts";
*/ */
export class ClaudeAgent implements Agent { export class ClaudeAgent implements Agent {
private apiKey: string; private apiKey: string;
private githubInstallationToken?: string; private githubInstallationToken: string;
public runStats = { public runStats = {
toolsUsed: 0, toolsUsed: 0,
turns: 0, turns: 0,
@@ -18,9 +19,6 @@ export class ClaudeAgent implements Agent {
}; };
constructor(config: AgentConfig) { constructor(config: AgentConfig) {
if (!config.apiKey) {
throw new Error("Claude agent requires an API key");
}
this.apiKey = config.apiKey; this.apiKey = config.apiKey;
this.githubInstallationToken = config.githubInstallationToken; this.githubInstallationToken = config.githubInstallationToken;
} }
@@ -51,10 +49,7 @@ export class ClaudeAgent implements Agent {
try { try {
const result = await spawn({ const result = await spawn({
cmd: "bash", cmd: "bash",
args: [ args: ["-c", "curl -fsSL https://claude.ai/install.sh | bash -s 1.0.93"],
"-c",
"curl -fsSL https://claude.ai/install.sh | bash -s 1.0.93",
],
env: { ANTHROPIC_API_KEY: this.apiKey }, env: { ANTHROPIC_API_KEY: this.apiKey },
timeout: 120000, // 2 minute timeout timeout: 120000, // 2 minute timeout
onStdout: () => {}, onStdout: () => {},
@@ -62,9 +57,7 @@ export class ClaudeAgent implements Agent {
}); });
if (result.exitCode !== 0) { if (result.exitCode !== 0) {
throw new Error( throw new Error(`Installation failed with exit code ${result.exitCode}: ${result.stderr}`);
`Installation failed with exit code ${result.exitCode}: ${result.stderr}`
);
} }
core.info("Claude Code installed successfully"); core.info("Claude Code installed successfully");
@@ -81,7 +74,16 @@ export class ClaudeAgent implements Agent {
try { try {
const claudePath = `${process.env.HOME}/.local/bin/claude`; const claudePath = `${process.env.HOME}/.local/bin/claude`;
const env = {
ANTHROPIC_API_KEY: this.apiKey,
};
console.log(boxString(prompt, { title: "Prompt" })); console.log(boxString(prompt, { title: "Prompt" }));
const mcpConfig = createMcpConfig(this.githubInstallationToken);
console.log("📋 MCP Config:", mcpConfig);
const args = [ const args = [
"--print", "--print",
"--output-format", "--output-format",
@@ -90,22 +92,10 @@ export class ClaudeAgent implements Agent {
"--debug", "--debug",
"--permission-mode", "--permission-mode",
"bypassPermissions", "bypassPermissions",
"--mcp-config",
mcpConfig,
]; ];
if (!this.githubInstallationToken) {
throw new Error(
"GITHUB_INSTALLATION_TOKEN is required for GitHub integration"
);
}
const mcpConfig = createMcpConfig(this.githubInstallationToken);
console.log("📋 MCP Config:", mcpConfig);
args.push("--mcp-config", mcpConfig);
const env = {
ANTHROPIC_API_KEY: this.apiKey,
};
core.startGroup("🔄 Run details"); core.startGroup("🔄 Run details");
this.runStats = { this.runStats = {
@@ -121,7 +111,7 @@ export class ClaudeAgent implements Agent {
cmd: claudePath, cmd: claudePath,
args, args,
env, env,
input: prompt, input: `${instructions} ${prompt}`,
timeout: 10 * 60 * 1000, // 10 minutes timeout: 10 * 60 * 1000, // 10 minutes
onStdout: (_chunk) => { onStdout: (_chunk) => {
processJSONChunk(_chunk, this); processJSONChunk(_chunk, this);
@@ -161,8 +151,7 @@ export class ClaudeAgent implements Agent {
try { try {
core.endGroup(); core.endGroup();
} catch {} } catch {}
const errorMessage = const errorMessage = error instanceof Error ? error.message : "Unknown error";
error instanceof Error ? error.message : "Unknown error";
return { return {
success: false, success: false,
error: `Failed to execute Claude Code: ${errorMessage}`, error: `Failed to execute Claude Code: ${errorMessage}`,
@@ -188,12 +177,7 @@ function processJSONChunk(chunk: string, agent?: ClaudeAgent): void {
["model", parsedChunk.model], ["model", parsedChunk.model],
["cwd", parsedChunk.cwd], ["cwd", parsedChunk.cwd],
["permission_mode", parsedChunk.permissionMode], ["permission_mode", parsedChunk.permissionMode],
[ ["tools", parsedChunk.tools?.length ? `${parsedChunk.tools.length} tools` : "none"],
"tools",
parsedChunk.tools?.length
? `${parsedChunk.tools.length} tools`
: "none",
],
[ [
"mcp_servers", "mcp_servers",
parsedChunk.mcp_servers?.length parsedChunk.mcp_servers?.length
@@ -220,9 +204,7 @@ function processJSONChunk(chunk: string, agent?: ClaudeAgent): void {
for (const content of parsedChunk.message.content) { for (const content of parsedChunk.message.content) {
if (content.type === "text") { if (content.type === "text") {
if (content.text.trim()) { if (content.text.trim()) {
core.info( core.info(boxString(content.text.trim(), { title: "Claude Code" }));
boxString(content.text.trim(), { title: "Claude Code" })
);
} }
} else if (content.type === "tool_use") { } else if (content.type === "tool_use") {
if (agent) { if (agent) {
@@ -307,10 +289,7 @@ function processJSONChunk(chunk: string, agent?: ClaudeAgent): void {
if (parsedChunk.subtype === "success") { if (parsedChunk.subtype === "success") {
core.info( core.info(
tableString([ tableString([
[ ["Cost", `$${parsedChunk.total_cost_usd?.toFixed(4) || "0.0000"}`],
"Cost",
`$${parsedChunk.total_cost_usd?.toFixed(4) || "0.0000"}`,
],
["Input Tokens", parsedChunk.usage?.input_tokens || 0], ["Input Tokens", parsedChunk.usage?.input_tokens || 0],
["Output Tokens", parsedChunk.usage?.output_tokens || 0], ["Output Tokens", parsedChunk.usage?.output_tokens || 0],
["Duration", `${parsedChunk.duration_ms}ms`], ["Duration", `${parsedChunk.duration_ms}ms`],
+6
View File
@@ -0,0 +1,6 @@
import { mcpServerName } from "../mcp/config.ts";
export const instructions = `- use the ${mcpServerName} MCP server to interact with github
- if ${mcpServerName} is not available or doesn't include the functionality you need, describe why and bail
- do not under any circumstances use the gh cli
`;
+2 -2
View File
@@ -29,6 +29,6 @@ export interface AgentResult {
* Configuration for agent creation * Configuration for agent creation
*/ */
export interface AgentConfig { export interface AgentConfig {
apiKey?: string; apiKey: string;
[key: string]: any; githubInstallationToken: string;
} }
+1 -3
View File
@@ -1,3 +1 @@
Use the MCP GitHub comment tool to add a comment containing your best frog joke to GitHub issue https://github.com/pullfrogai/scratch/issues/2. add a comment containing your best frog joke to GitHub issue https://github.com/pullfrogai/scratch/issues/2
Do not use the gh cli. If the mcp tool does not work, bail.
+1
View File
@@ -27,6 +27,7 @@ export async function main(inputs: Inputs): Promise<MainResult> {
apiKey: inputs.anthropic_api_key!, apiKey: inputs.anthropic_api_key!,
githubInstallationToken, githubInstallationToken,
}); });
await agent.install(); await agent.install();
const result = await agent.execute(inputs.prompt); const result = await agent.execute(inputs.prompt);
+6 -18
View File
@@ -1,7 +1,5 @@
import { Octokit } from "@octokit/rest";
import { type } from "arktype"; import { type } from "arktype";
import { resolveRepoContext } from "../utils/repo-context.ts"; import { getMcpContext, tool } from "./shared.ts";
import { tool } from "./shared.ts";
export const Comment = type({ export const Comment = type({
issueNumber: type.number.describe("the issue number to comment on"), issueNumber: type.number.describe("the issue number to comment on"),
@@ -12,22 +10,12 @@ export const CommentTool = tool({
name: "create_issue_comment", name: "create_issue_comment",
description: "Create a comment on a GitHub issue", description: "Create a comment on a GitHub issue",
parameters: Comment, parameters: Comment,
execute: async ({ issueNumber, body }: { issueNumber: number; body: string }) => { execute: async ({ issueNumber, body }) => {
const ctx = getMcpContext();
try { try {
const githubInstallationToken = process.env.GITHUB_INSTALLATION_TOKEN; const result = await ctx.octokit.rest.issues.createComment({
if (!githubInstallationToken) { owner: ctx.owner,
throw new Error("GITHUB_INSTALLATION_TOKEN environment variable is required"); repo: ctx.name,
}
const repoContext = resolveRepoContext();
const octokit = new Octokit({
auth: githubInstallationToken,
});
const result = await octokit.rest.issues.createComment({
owner: repoContext.owner,
repo: repoContext.name,
issue_number: issueNumber, issue_number: issueNumber,
body: body, body: body,
}); });
+3 -1
View File
@@ -5,6 +5,8 @@ import { fromHere } from "@ark/fs";
const actionPath = fromHere(".."); const actionPath = fromHere("..");
export const mcpServerName = "gh-pullfrog";
export function createMcpConfig(githubInstallationToken: string) { export function createMcpConfig(githubInstallationToken: string) {
const githubRepository = process.env.GITHUB_REPOSITORY; const githubRepository = process.env.GITHUB_REPOSITORY;
if (!githubRepository) { if (!githubRepository) {
@@ -16,7 +18,7 @@ export function createMcpConfig(githubInstallationToken: string) {
return JSON.stringify( return JSON.stringify(
{ {
mcpServers: { mcpServers: {
minimal_github_comment: { [mcpServerName]: {
command: "node", command: "node",
args: [`${actionPath}/mcp/server.ts`], args: [`${actionPath}/mcp/server.ts`],
env: { env: {
+1 -1
View File
@@ -5,7 +5,7 @@ import { CommentTool } from "./comment.ts";
import { addTools } from "./shared.ts"; import { addTools } from "./shared.ts";
const server = new FastMCP({ const server = new FastMCP({
name: "Minimal GitHub Issue Comment Server", name: "gh-pullfrog",
version: "0.0.1", version: "0.0.1",
}); });
+22 -1
View File
@@ -1,7 +1,28 @@
import { cached } from "@ark/util";
import { Octokit } from "@octokit/rest";
import type { StandardSchemaV1 } from "@standard-schema/spec"; import type { StandardSchemaV1 } from "@standard-schema/spec";
import type { FastMCP, Tool } from "fastmcp"; import type { FastMCP, Tool } from "fastmcp";
import { parseRepoContext, type RepoContext } from "../utils/github.ts";
export const tool = <const params>(tool: Tool<{}, StandardSchemaV1<params>>) => tool; export const getMcpContext = cached((): McpContext => {
const githubInstallationToken = process.env.GITHUB_INSTALLATION_TOKEN;
if (!githubInstallationToken) {
throw new Error("GITHUB_INSTALLATION_TOKEN environment variable is required");
}
return {
...parseRepoContext(),
octokit: new Octokit({
auth: githubInstallationToken,
}),
};
});
export interface McpContext extends RepoContext {
octokit: Octokit;
}
export const tool = <const params>(tool: Tool<any, StandardSchemaV1<params>>) => tool;
export const addTools = (server: FastMCP, tools: Tool<any, any>[]) => { export const addTools = (server: FastMCP, tools: Tool<any, any>[]) => {
for (const tool of tools) { for (const tool of tools) {
+3 -2
View File
@@ -1,6 +1,6 @@
{ {
"name": "@pullfrog/action", "name": "@pullfrog/action",
"version": "0.0.50", "version": "0.0.52",
"type": "module", "type": "module",
"files": [ "files": [
"index.js", "index.js",
@@ -21,7 +21,8 @@
}, },
"dependencies": { "dependencies": {
"@actions/core": "^1.11.1", "@actions/core": "^1.11.1",
"@ark/fs": "0.49.0", "@ark/fs": "0.50.0",
"@ark/util": "0.50.0",
"@octokit/rest": "^22.0.0", "@octokit/rest": "^22.0.0",
"@octokit/webhooks-types": "^7.6.1", "@octokit/webhooks-types": "^7.6.1",
"@standard-schema/spec": "1.0.0", "@standard-schema/spec": "1.0.0",
+8 -5
View File
@@ -12,8 +12,11 @@ importers:
specifier: ^1.11.1 specifier: ^1.11.1
version: 1.11.1 version: 1.11.1
'@ark/fs': '@ark/fs':
specifier: 0.49.0 specifier: 0.50.0
version: 0.49.0 version: 0.50.0
'@ark/util':
specifier: 0.50.0
version: 0.50.0
'@octokit/rest': '@octokit/rest':
specifier: ^22.0.0 specifier: ^22.0.0
version: 22.0.0 version: 22.0.0
@@ -63,8 +66,8 @@ packages:
'@actions/io@1.1.3': '@actions/io@1.1.3':
resolution: {integrity: sha512-wi9JjgKLYS7U/z8PPbco+PvTb/nRWjeoFlJ1Qer83k/3C5PHQi28hiVdeE2kHXmIL99mQFawx8qt/JPjZilJ8Q==} resolution: {integrity: sha512-wi9JjgKLYS7U/z8PPbco+PvTb/nRWjeoFlJ1Qer83k/3C5PHQi28hiVdeE2kHXmIL99mQFawx8qt/JPjZilJ8Q==}
'@ark/fs@0.49.0': '@ark/fs@0.50.0':
resolution: {integrity: sha512-AEjAQS/bu1CGIRiKK/XLaQ73cSJHixfexq28wNt+kBpQ0h1RwVIVzaGsn/+5IWw6DEbR7LB+3hil5gzrzEeyZQ==} resolution: {integrity: sha512-6OrxNt2T+/pL4RUMZK/aiVRLIS3acNs5uSpHDyZAP6+OXwXchFSQ1lJTH+uuGBlDWeQxPD7VmymYXLF5s1Eyhw==}
'@ark/regex@0.0.0': '@ark/regex@0.0.0':
resolution: {integrity: sha512-p4vsWnd/LRGOdGQglbwOguIVhPmCAf5UzquvnDoxqhhPWTP84wWgi1INea8MgJ4SnI2gp37f13oA4Waz9vwNYg==} resolution: {integrity: sha512-p4vsWnd/LRGOdGQglbwOguIVhPmCAf5UzquvnDoxqhhPWTP84wWgi1INea8MgJ4SnI2gp37f13oA4Waz9vwNYg==}
@@ -792,7 +795,7 @@ snapshots:
'@actions/io@1.1.3': {} '@actions/io@1.1.3': {}
'@ark/fs@0.49.0': {} '@ark/fs@0.50.0': {}
'@ark/regex@0.0.0': '@ark/regex@0.0.0':
dependencies: dependencies:
+26 -5
View File
@@ -1,6 +1,5 @@
import { createSign } from "node:crypto"; import { createSign } from "node:crypto";
import * as core from "@actions/core"; import * as core from "@actions/core";
import { resolveRepoContext } from "./repo-context.ts";
export interface InstallationToken { export interface InstallationToken {
token: string; token: string;
@@ -55,7 +54,7 @@ function isGitHubActionsEnvironment(): boolean {
async function acquireTokenViaOIDC(): Promise<string> { async function acquireTokenViaOIDC(): Promise<string> {
core.info("Generating OIDC token..."); core.info("Generating OIDC token...");
const oidcToken = await core.getIDToken("pullfrog-api"); const oidcToken = await core.getIDToken("pullfrog-api");
core.info("OIDC token generated successfully"); core.info("OIDC token generated successfully");
@@ -208,7 +207,7 @@ const findInstallationId = async (
}; };
async function acquireTokenViaGitHubApp(): Promise<string> { async function acquireTokenViaGitHubApp(): Promise<string> {
const repoContext = resolveRepoContext(); const repoContext = parseRepoContext();
const config: GitHubAppConfig = { const config: GitHubAppConfig = {
appId: process.env.GITHUB_APP_ID!, appId: process.env.GITHUB_APP_ID!,
@@ -244,9 +243,31 @@ export async function setupGitHubInstallationToken(): Promise<string> {
} }
const token = await acquireNewToken(); const token = await acquireNewToken();
core.setSecret(token); core.setSecret(token);
process.env.GITHUB_INSTALLATION_TOKEN = token; process.env.GITHUB_INSTALLATION_TOKEN = token;
return token; return token;
} }
export interface RepoContext {
owner: string;
name: string;
}
/**
* Parse repository context from GITHUB_REPOSITORY environment variable.
*/
export function parseRepoContext(): RepoContext {
const githubRepo = process.env.GITHUB_REPOSITORY;
if (!githubRepo) {
throw new Error("GITHUB_REPOSITORY environment variable is required");
}
const [owner, name] = githubRepo.split("/");
if (!owner || !name) {
throw new Error(`Invalid GITHUB_REPOSITORY format: ${githubRepo}. Expected 'owner/repo'`);
}
return { owner, name };
}
-22
View File
@@ -1,22 +0,0 @@
export interface RepoContext {
owner: string;
name: string;
}
/**
* Resolve repository context from GITHUB_REPOSITORY environment variable.
* Throws if not available.
*/
export function resolveRepoContext(): RepoContext {
const githubRepo = process.env.GITHUB_REPOSITORY;
if (!githubRepo) {
throw new Error('GITHUB_REPOSITORY environment variable is required');
}
const [owner, name] = githubRepo.split('/');
if (!owner || !name) {
throw new Error(`Invalid GITHUB_REPOSITORY format: ${githubRepo}. Expected 'owner/repo'`);
}
return { owner, name };
}