only allow api key through env

Author: charleslien

sdk/src/client.ts

@@ -13,22 +13,21 @@ import type {
 } from './types'
 
 export class CodebuffClient {
-  private authToken: string
-
   public cwd: string
 
-  constructor({ apiKey, cwd }: CodebuffClientOptions) {
+  constructor({ cwd }: CodebuffClientOptions) {
     // TODO: download binary automatically
     if (execFileSync('which', [CODEBUFF_BINARY]).toString().trim() === '') {
       throw new Error(
         'Codebuff binary not found. Please run "npm i -g codebuff"',
       )
     }
+    if (!process.env[API_KEY_ENV_VAR]) {
+      throw new Error(
+        `Codebuff API key not found. Please set the ${API_KEY_ENV_VAR} environment variable.`,
+      )
+    }
 
-    this.authToken =
-      apiKey.type === 'string'
-        ? apiKey.value
-        : process.env[API_KEY_ENV_VAR] ?? ''
     this.cwd = cwd
   }
 
@@ -51,7 +50,6 @@ export class CodebuffClient {
 
     await processStream({
       codebuffArgs: args,
-      authToken: this.authToken,
       handleEvent,
     })
 
@@ -82,7 +80,6 @@ export class CodebuffClient {
 
     await processStream({
       codebuffArgs: args,
-      authToken: this.authToken,
       handleEvent,
     })
 

sdk/src/process-stream.ts

@@ -6,11 +6,9 @@ import type { PrintModeEvent } from '@codebuff/common/types/print-mode'
 
 export function processStream({
   codebuffArgs,
-  authToken,
   handleEvent,
 }: {
   codebuffArgs: string[]
-  authToken?: string
   handleEvent: (event: PrintModeEvent) => void
 }): Promise<void> {
   let buffer = ''
@@ -29,9 +27,6 @@ export function processStream({
   }
 
   const env = { ...process.env }
-  if (authToken) {
-    env.CODEBUFF_API_KEY = authToken
-  }
   const child = spawn(CODEBUFF_BINARY, codebuffArgs, {
     stdio: 'pipe',
     env,

sdk/src/types.ts

@@ -2,7 +2,6 @@ import type { PrintModeEvent } from '@codebuff/common/types/print-mode'
 import type { AgentTemplateType } from '@codebuff/common/types/session-state'
 
 export type CodebuffClientOptions = {
-  apiKey: { type: 'string'; value: string } | { type: 'env' }
   cwd: string
 }