Merge pull request #14 from elie222/feat/openai-key-option

feat: allow passing the openai api key as an option
lucgagan · Nov 17, 2023 · 086e9e8 · 086e9e8
2 parents 0e158c1 + 6a38a4c
commit 086e9e8
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -102,6 +102,8 @@ const options = {
   debug: boolean,
   // The OpenAI model (https://platform.openai.com/docs/models/overview)
   model: "gpt-4-1106-preview",
+  // The OpenAI API key
+  openaiApiKey: 'sk-...',
 };
 
 auto("<your prompt>", { page, test }, options);

diff --git a/src/auto.ts b/src/auto.ts
@@ -58,6 +58,7 @@ async function runTask(
       ? {
           model: options.model ?? "gpt-4-1106-preview",
           debug: options.debug ?? false,
+          openaiApiKey: options.openaiApiKey,
         }
       : undefined,
   });

diff --git a/src/completeTask.ts b/src/completeTask.ts
@@ -9,7 +9,7 @@ export const completeTask = async (
   page: Page,
   task: TaskMessage
 ): Promise<TaskResult> => {
-  const openai = new OpenAI();
+  const openai = new OpenAI({ apiKey: task.options?.openaiApiKey });
 
   let lastFunctionResult: null | { errorMessage: string } | { query: string } =
     null;

diff --git a/src/types.ts b/src/types.ts
@@ -7,6 +7,7 @@ export type Test = TestType<any, any>;
 export type StepOptions = {
   debug?: boolean;
   model?: string;
+  openaiApiKey?: string;
 };
 
 export type TaskMessage = {