sdk: implement addAgentStep

Author: charleslien

common/src/types/contracts/database.ts

@@ -77,6 +77,7 @@ export type FinishAgentRunFn = (params: {
 }) => Promise<void>
 
 export type AddAgentStepFn = (params: {
+  apiKey: string
   userId: string | undefined
   agentRunId: string
   stepNumber: number
@@ -87,6 +88,6 @@ export type AddAgentStepFn = (params: {
   errorMessage?: string
   startTime: Date
   logger: Logger
-}) => Promise<string>
+}) => Promise<string | null>
 
 export type DatabaseAgentCache = Map<string, AgentTemplate | null>

packages/agent-runtime/src/run-agent-step.ts

@@ -450,6 +450,20 @@ export async function loopAgentSteps(
     ParamsExcluding<
       FinishAgentRunFn,
       'runId' | 'status' | 'totalSteps' | 'directCredits' | 'totalCredits'
+    > &
+    ParamsExcluding<
+      typeof runAgentStep,
+      'agentState' | 'prompt' | 'spawnParams' | 'system'
+    > &
+    ParamsExcluding<
+      AddAgentStepFn,
+      | 'agentRunId'
+      | 'stepNumber'
+      | 'credits'
+      | 'childRunIds'
+      | 'messageId'
+      | 'status'
+      | 'startTime'
     >,
 ): Promise<{
   agentState: AgentState
@@ -677,14 +691,6 @@ export async function loopAgentSteps(
         messageId,
       } = await runAgentStep({
         ...params,
-        userId,
-        userInputId,
-        clientSessionId,
-        fingerprintId,
-        onResponseChunk,
-        localAgentTemplates,
-        agentType,
-        fileContext,
         agentState: currentAgentState,
         prompt: currentPrompt,
         spawnParams: currentParams,
@@ -693,15 +699,14 @@ export async function loopAgentSteps(
 
       if (newAgentState.runId) {
         await addAgentStep({
-          userId,
+          ...params,
           agentRunId: newAgentState.runId,
           stepNumber: totalSteps,
           credits: newAgentState.directCreditsUsed - creditsBefore,
           childRunIds: newAgentState.childRunIds.slice(childrenBefore),
           messageId,
           status: 'completed',
           startTime,
-          logger,
         })
       } else {
         logger.error('No runId found for agent state after finishing agent run')

sdk/src/impl/database.ts

@@ -3,6 +3,7 @@ import { getErrorObject } from '@codebuff/common/util/error'
 import { WEBSITE_URL } from '../constants'
 
 import type {
+  AddAgentStepFn,
   FetchAgentFromDatabaseFn,
   FinishAgentRunFn,
   GetUserInfoFromApiKeyInput,
@@ -155,15 +156,62 @@ export async function finishAgentRun(
   }
 }
 
-console.log(
-  await finishAgentRun({
-    apiKey: '12345',
-    status: 'completed',
-    totalSteps: 5,
-    userId: undefined,
-    runId: 'e7a129b2-feb5-40ac-b2cd-0d7c115962f5',
-    directCredits: 12,
-    totalCredits: 34,
-    logger: console,
-  }),
-)
+export async function addAgentStep(
+  params: ParamsOf<AddAgentStepFn>,
+): ReturnType<AddAgentStepFn> {
+  const {
+    apiKey,
+    agentRunId,
+    stepNumber,
+    credits,
+    childRunIds,
+    messageId,
+    status = 'completed',
+    errorMessage,
+    startTime,
+    logger,
+  } = params
+
+  const url = new URL(`/api/v1/agent-runs/${agentRunId}/steps`, WEBSITE_URL)
+
+  try {
+    const response = await fetch(url, {
+      method: 'POST',
+      headers: {
+        Authorization: `Bearer ${apiKey}`,
+      },
+      body: JSON.stringify({
+        stepNumber,
+        credits,
+        childRunIds,
+        messageId,
+        status,
+        errorMessage,
+        startTime,
+      }),
+    })
+
+    if (!response.ok) {
+      logger.error({ response }, 'addAgentStep request failed')
+      return null
+    }
+
+    return response.json()
+  } catch (error) {
+    logger.error(
+      {
+        error: getErrorObject(error),
+        agentRunId,
+        stepNumber,
+        credits,
+        childRunIds,
+        messageId,
+        status,
+        errorMessage,
+        startTime,
+      },
+      'addAgentStep error',
+    )
+    return null
+  }
+}