lazily compute agent runtime impl

Author: charleslien

sdk/src/impl/agent-runtime.ts

@@ -12,8 +12,12 @@ import type {
   AgentRuntimeDeps,
   AgentRuntimeScopedDeps,
 } from '@codebuff/common/types/contracts/agent-runtime'
+import type { Logger } from '@codebuff/common/types/contracts/logger'
 
-export const CLI_AGENT_RUNTIME_IMPL: Omit<
+export function getAgentRuntimeImpl(params: {
+  logger?: Logger
+  apiKey: string
+}): Omit<
   AgentRuntimeDeps & AgentRuntimeScopedDeps,
   | 'addAgentStep'
   | 'promptAiSdkStream'
@@ -26,50 +30,54 @@ export const CLI_AGENT_RUNTIME_IMPL: Omit<
   | 'requestOptionalFile'
   | 'sendAction'
   | 'sendSubagentChunk'
-> = {
-  // Database
-  getUserInfoFromApiKey,
-  fetchAgentFromDatabase,
-  startAgentRun,
-  finishAgentRun,
-  // addAgentStep: AddAgentStepFn
+> {
+  const { logger, apiKey } = params
+
+  return {
+    // Database
+    getUserInfoFromApiKey,
+    fetchAgentFromDatabase,
+    startAgentRun,
+    finishAgentRun,
+    // addAgentStep: AddAgentStepFn
 
-  // Billing
-  consumeCreditsWithFallback: async () =>
-    success({
-      chargedToOrganization: false,
-    }),
+    // Billing
+    consumeCreditsWithFallback: async () =>
+      success({
+        chargedToOrganization: false,
+      }),
 
-  // LLM
-  // promptAiSdkStream: PromptAiSdkStreamFn,
-  // promptAiSdk: PromptAiSdkFn,
-  // promptAiSdkStructured: PromptAiSdkStructuredFn,
+    // LLM
+    // promptAiSdkStream: PromptAiSdkStreamFn,
+    // promptAiSdk: PromptAiSdkFn,
+    // promptAiSdkStructured: PromptAiSdkStructuredFn,
 
-  // Mutable State
-  databaseAgentCache: new Map(),
-  liveUserInputRecord: {},
-  sessionConnections: {},
+    // Mutable State
+    databaseAgentCache: new Map(),
+    liveUserInputRecord: {},
+    sessionConnections: {},
 
-  // Analytics
-  trackEvent,
+    // Analytics
+    trackEvent,
 
-  // Other
-  logger: {
-    info: () => {},
-    debug: () => {},
-    warn: () => {},
-    error: () => {},
-  },
-  fetch: globalThis.fetch,
+    // Other
+    logger: logger ?? {
+      info: () => {},
+      debug: () => {},
+      warn: () => {},
+      error: () => {},
+    },
+    fetch: globalThis.fetch,
 
-  // Client (WebSocket)
-  // handleStepsLogChunk: HandleStepsLogChunkFn,
-  // requestToolCall: RequestToolCallFn,
-  // requestMcpToolData: RequestMcpToolDataFn,
-  // requestFiles: RequestFilesFn,
-  // requestOptionalFile: RequestOptionalFileFn,
-  // sendAction: SendActionFn,
-  // sendSubagentChunk: SendSubagentChunkFn,
+    // Client (WebSocket)
+    // handleStepsLogChunk: HandleStepsLogChunkFn,
+    // requestToolCall: RequestToolCallFn,
+    // requestMcpToolData: RequestMcpToolDataFn,
+    // requestFiles: RequestFilesFn,
+    // requestOptionalFile: RequestOptionalFileFn,
+    // sendAction: SendActionFn,
+    // sendSubagentChunk: SendSubagentChunkFn,
 
-  apiKey: process.env.CODEBUFF_API_KEY ?? '',
+    apiKey,
+  }
 }