diff options
| -rw-r--r-- | packages/opencode/src/session/index.ts | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/packages/opencode/src/session/index.ts b/packages/opencode/src/session/index.ts index 13496a775..d77da209e 100644 --- a/packages/opencode/src/session/index.ts +++ b/packages/opencode/src/session/index.ts @@ -379,9 +379,12 @@ ${app.git ? await ListTool.execute({ path: app.path.cwd }, { sessionID: input.se let text: Message.TextPart | undefined const result = streamText({ onStepFinish: async (step) => { + log.info("step finish", { + finishReason: step.finishReason, + }) const assistant = next.metadata!.assistant! const usage = getUsage(step.usage, model.info) - assistant.cost = usage.cost + assistant.cost += usage.cost assistant.tokens = usage.tokens await updateMessage(next) if (text) { @@ -465,6 +468,12 @@ ${app.git ? await ListTool.execute({ path: app.path.cwd }, { sessionID: input.se } await updateMessage(next) }, + async onFinish(input) { + const assistant = next.metadata!.assistant! + const usage = getUsage(input.totalUsage, model.info) + assistant.cost = usage.cost + await updateMessage(next) + }, onError(input) { if (input.error instanceof Error) { next.metadata.error = input.error.toString() |