sdk: update default fs

Author: charleslien

sdk/src/run-state.ts

@@ -23,6 +23,7 @@ import type {
   CustomToolDefinitions,
   FileTreeNode,
 } from '@codebuff/common/util/file'
+import type * as fsType from 'fs'
 
 export type RunState = {
   sessionState: SessionState
@@ -149,13 +150,13 @@ async function discoverProjectFiles(params: {
     )
   }
 
-  const projectFilesRecord: Record<string, string> = {}
+  const projectFilesResolved: Record<string, string> = {}
   for (const [filePath, contentPromise] of Object.entries(
     projectFilePromises,
   )) {
-    projectFilesRecord[filePath] = await contentPromise
+    projectFilesResolved[filePath] = await contentPromise
   }
-  return projectFilesRecord
+  return projectFilesResolved
 }
 
 /**
@@ -180,37 +181,48 @@ function deriveKnowledgeFiles(
 export async function initialSessionState(
   params: InitialSessionStateOptions,
 ): Promise<SessionState> {
-  const withDefaults = {
-    agentDefinitions: [],
-    customToolDefinitions: [],
-    logger: {
-      debug: () => {},
-      info: () => {},
-      warn: () => {},
-      error: () => {},
-    },
-    ...params,
-  }
-  const {
-    cwd,
+  const { cwd, maxAgentSteps } = params
+  let {
     agentDefinitions,
     customToolDefinitions,
-    maxAgentSteps,
+    projectFiles,
+    knowledgeFiles,
+    fs,
     logger,
-  } = withDefaults
-  let { projectFiles, knowledgeFiles, fs } = withDefaults
-
+  } = params
+  if (!agentDefinitions) {
+    agentDefinitions = []
+  }
+  if (!customToolDefinitions) {
+    customToolDefinitions = []
+  }
+  if (!projectFiles) {
+    projectFiles = {}
+  }
+  if (!knowledgeFiles) {
+    knowledgeFiles = {}
+  }
   if (!fs) {
-    fs = (await import('fs')) as unknown as CodebuffFileSystem
+    fs = (require('fs') as typeof fsType).promises
+  }
+  if (!logger) {
+    logger = {
+      debug: () => {},
+      info: () => {},
+      warn: () => {},
+      error: () => {},
+    }
   }
 
   // Auto-discover project files if not provided and cwd is available
   if (projectFiles === undefined && cwd) {
     projectFiles = await discoverProjectFiles({ cwd, fs, logger })
   }
+  logger.info({ projectFiles }, 'asdf')
   if (knowledgeFiles === undefined) {
     knowledgeFiles = projectFiles ? deriveKnowledgeFiles(projectFiles) : {}
   }
+  logger.info({ knowledgeFiles }, 'asdf')
 
   const processedAgentTemplates = processAgentDefinitions(agentDefinitions)
   const processedCustomToolDefinitions = processCustomToolDefinitions(

sdk/src/run.ts

@@ -120,7 +120,7 @@ export async function run({
   overrideTools,
   customToolDefinitions,
 
-  fsSource = () => require('fs'),
+  fsSource = () => require('fs').promises,
   logger,
 
   agent,