refactor config file

presubmit · Nov 21, 2024 · f4d1a04 · f4d1a04
1 parent 33ada0b
commit f4d1a04
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/ai.ts b/src/ai.ts
@@ -46,7 +46,7 @@ export async function runPrompt({
     throw new Error(`Unknown LLM model: ${config.llmModel}`);
   }
 
-  const llm = model.createAi({ apiKey: process.env.LLM_API_KEY });
+  const llm = model.createAi({ apiKey: config.llmApiKey });
   const { object, usage } = await generateObject({
     model: llm(model.name),
     prompt,

diff --git a/src/config.ts b/src/config.ts
@@ -1,12 +1,25 @@
 import { getInput } from "@actions/core";
 
 export class Config {
+  public llmApiKey: string | undefined;
   public llmModel: string | undefined;
   public githubToken: string | undefined;
 
   constructor() {
     this.githubToken = process.env.GITHUB_TOKEN;
+    if (!this.githubToken) {
+      throw new Error("GITHUB_TOKEN is not set");
+    }
+
+    this.llmApiKey = process.env.LLM_API_KEY;
+    if (!this.llmApiKey) {
+      throw new Error("LLM_API_KEY is not set");
+    }
+
     this.llmModel = process.env.LLM_MODEL;
+    if (!this.llmModel) {
+      throw new Error("LLM_MODEL is not set");
+    }
   }
 
   public loadInputs() {