(sess *session.Session, a *agent.Agent, content, modelID string, usage *chat.Usage, cost float64, events EventSink)
| 343 | } |
| 344 | |
| 345 | func (r *LocalRuntime) recordHarnessAssistantMessage(sess *session.Session, a *agent.Agent, content, modelID string, usage *chat.Usage, cost float64, events EventSink) { |
| 346 | if strings.TrimSpace(content) == "" && usage == nil { |
| 347 | return |
| 348 | } |
| 349 | |
| 350 | msg := chat.Message{ |
| 351 | Role: chat.MessageRoleAssistant, |
| 352 | Content: content, |
| 353 | CreatedAt: r.now().Format(time.RFC3339), |
| 354 | Usage: usage, |
| 355 | Model: modelID, |
| 356 | Cost: cost, |
| 357 | FinishReason: chat.FinishReasonStop, |
| 358 | } |
| 359 | addAgentMessage(sess, a, &msg, events) |
| 360 | |
| 361 | if usage == nil { |
| 362 | return |
| 363 | } |
| 364 | input := usage.InputTokens + usage.CachedInputTokens + usage.CacheWriteTokens |
| 365 | sess.SetUsage(input, usage.OutputTokens) |
| 366 | msgUsage := &MessageUsage{ |
| 367 | Usage: *usage, |
| 368 | Cost: cost, |
| 369 | Model: modelID, |
| 370 | FinishReason: chat.FinishReasonStop, |
| 371 | } |
| 372 | usageEvent := SessionUsage(sess, 0) |
| 373 | usageEvent.LastMessage = msgUsage |
| 374 | events.Emit(NewTokenUsageEvent(sess.ID, a.Name(), usageEvent)) |
| 375 | } |
| 376 | |
| 377 | func harnessPromptFromMessages(messages []chat.Message) string { |
| 378 | var b strings.Builder |
no test coverage detected