auggie.ts

import * as core from "@actions/core";
import { Auggie } from "@augmentcode/auggie-sdk";
import type { GithubPullRequest } from "./types";
import { AGENT_SYSTEM_PROMPT, generateContextPrompt } from "./utils/prompt";

/**
 * Options for running Auggie agent
 */
export type RunAuggieOptions = {
  userPrompt: string;
  apiKey?: string;
  apiUrl?: string;
  workspaceRoot?: string;
  context?: GithubPullRequest;
  /** The body of the comment that triggered this action, if triggered by a comment */
  commentBody?: string;
  /** The ID of the comment that triggered this action, if triggered by a comment */
  commentId?: number;
};

/**
 * Run Auggie agent with the given prompt and return the response
 */
export async function runAuggie(options: RunAuggieOptions): Promise<string> {
  const {
    userPrompt,
    apiKey,
    apiUrl,
    workspaceRoot,
    context,
    commentBody,
    commentId,
  } = options;

  const workspace = workspaceRoot || process.cwd();

  core.info("🤖 Initializing Auggie agent...");
  core.info(`📁 Workspace root: ${workspace}`);

  let client: Auggie | null = null;

  try {
    // Set GITHUB_API_TOKEN environment variable from GITHUB_TOKEN if available
    const githubToken = process.env.GITHUB_TOKEN;
    if (githubToken) {
      process.env.GITHUB_API_TOKEN = githubToken;
      core.info("✅ GITHUB_API_TOKEN environment variable set from GITHUB_TOKEN");
    }

    // Create Auggie client
    client = await Auggie.create({
      model: "sonnet4.5",
      apiKey,
      apiUrl,
      workspaceRoot: workspace,
      allowIndexing: true,
    });

    core.info("📝 Running Auggie agent ...");

    // Build the full prompt by combining system prompt, context, and user prompt
    let fullPrompt = AGENT_SYSTEM_PROMPT;

    // Add context information if provided
    if (context) {
      const contextPrompt = generateContextPrompt(context);
      fullPrompt = `${fullPrompt}\n\n## PR Context:\n${contextPrompt}`;
    }

    // Add user comment request if triggered by a comment
    if (commentBody) {
      fullPrompt = `${fullPrompt}\n\n## User Specific Request:\n${commentBody}`;
      core.info("📨 User comment included in prompt");
    }

    // Add triggering comment ID if available
    if (commentId) {
      fullPrompt = `${fullPrompt}\n\n## Triggering Comment ID:\n${commentId}`;
      core.info(`📍 Triggering comment ID: ${commentId}`);
    }

    // Add user prompt
    fullPrompt = `${fullPrompt}\n\n${userPrompt}`;

    const response = await client.prompt(fullPrompt, { isAnswerOnly: true });

    core.info("✅ Auggie agent completed successfully");

    return response;
  } catch (error) {
    const errorMessage = error instanceof Error ? error.message : String(error);
    core.error(`Failed to run Auggie agent: ${errorMessage}`);
    throw error;
  } finally {
    // Always close the client
    if (client) {
      await client.close();
    }
  }
}