fix: clean up (#1)

Author: Pradumnasaraf

.gitignore

@@ -43,9 +43,6 @@ node_modules/
 jspm_packages/
 
 
-#
-src/
-
 # Snowpack dependency directory (https://snowpack.dev/)
 web_modules/
 
@@ -93,6 +90,7 @@ out
 
 # Nuxt.js build / generate output
 .nuxt
+local-test/
 
 
 # Gatsby files

action.yml

@@ -1,5 +1,5 @@
-name: 'Hello World JS Action'
-description: 'Greet someone and record the time'
+name: 'PullPrompt'
+description: 'A GitHub Actions that comments on a pull request with a user given prompt and uses the Google Gemini API to generate a response'
 author: "Pradumna Saraf"
 
 inputs:
@@ -14,7 +14,6 @@ inputs:
     description: 'Your Gemini API Key'
     required: true
 
-
 runs:
   using: 'node20'
   main: 'dist/index.js'

src/index.js

@@ -0,0 +1,45 @@
+import { getInput, setFailed } from "@actions/core";
+import { context, getOctokit } from "@actions/github";
+import { GoogleGenerativeAI } from "@google/generative-ai";
+
+async function run() {
+  try {
+    const token = getInput("github-token", { required: true });
+    const geminiApiKey = getInput("gemini-api-key", { required: true });
+    const prompt = getInput("user-prompt");
+
+    const octokit = getOctokit(token);
+
+    if (context.payload.pull_request == null) {
+      throw new Error("This action can only be run on pull_request events");
+    }
+
+    const { owner, repo } = context.repo;
+    const { number } = context.payload.pull_request;
+    const commentBody = await geminiCall(geminiApiKey, prompt);
+    await octokit.rest.issues.createComment({
+      owner,
+      repo,
+      issue_number: number,
+      body: commentBody,
+    });
+  } catch (error) {
+    setFailed(error.message);
+  }
+}
+
+async function geminiCall(key, prompt) {
+  try {
+    const genAI = new GoogleGenerativeAI(key);
+    const model = genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
+
+    const result = await model.generateContent(prompt);
+
+    return result.response.text();
+  } catch (error) {
+    setFailed("Error generating content with Gemini:", error);
+    return "An error occurred while generating content.";
+  }
+}
+
+run();