mika kuns
7d3d6d7b54
task_runs.tokens_in only ever held the API's uncached "input_tokens" field (off by a factor of ~400,000 on a resumed session), and tokens_out summed only the last result event instead of the whole session. TaskRunner now reads each run's cache-read/cache-write/input/output totals from the session transcript via a new ITranscriptUsageReader.ReadSessionTotalsAsync, storing the delta against prior runs on the same session so a --resume doesn't double-count. New task_runs.cache_read_tokens/cache_write_tokens columns; the Session tab now shows the raw total (what actually counts against the 5h/7d usage limit) with a breakdown tooltip.
42 lines
2.1 KiB
C#
42 lines
2.1 KiB
C#
using ClaudeDo.Data.Models;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Microsoft.EntityFrameworkCore.Metadata.Builders;
|
|
|
|
namespace ClaudeDo.Data.Configuration;
|
|
|
|
public class TaskRunEntityConfiguration : IEntityTypeConfiguration<TaskRunEntity>
|
|
{
|
|
public void Configure(EntityTypeBuilder<TaskRunEntity> builder)
|
|
{
|
|
builder.ToTable("task_runs");
|
|
|
|
builder.HasKey(r => r.Id);
|
|
builder.Property(r => r.Id).HasColumnName("id");
|
|
builder.Property(r => r.TaskId).HasColumnName("task_id").IsRequired();
|
|
builder.Property(r => r.RunNumber).HasColumnName("run_number").IsRequired();
|
|
builder.Property(r => r.SessionId).HasColumnName("session_id");
|
|
builder.Property(r => r.IsRetry).HasColumnName("is_retry").IsRequired().HasDefaultValue(false);
|
|
builder.Property(r => r.Prompt).HasColumnName("prompt").IsRequired();
|
|
builder.Property(r => r.ResultMarkdown).HasColumnName("result_markdown");
|
|
builder.Property(r => r.StructuredOutputJson).HasColumnName("structured_output");
|
|
builder.Property(r => r.ErrorMarkdown).HasColumnName("error_markdown");
|
|
builder.Property(r => r.ExitCode).HasColumnName("exit_code");
|
|
builder.Property(r => r.TurnCount).HasColumnName("turn_count");
|
|
builder.Property(r => r.TokensIn).HasColumnName("tokens_in");
|
|
builder.Property(r => r.TokensOut).HasColumnName("tokens_out");
|
|
builder.Property(r => r.CacheReadTokens).HasColumnName("cache_read_tokens");
|
|
builder.Property(r => r.CacheWriteTokens).HasColumnName("cache_write_tokens");
|
|
builder.Property(r => r.LogPath).HasColumnName("log_path");
|
|
builder.Property(r => r.StartedAt).HasColumnName("started_at");
|
|
builder.Property(r => r.FinishedAt).HasColumnName("finished_at");
|
|
builder.Property(r => r.Model).HasColumnName("model");
|
|
|
|
builder.HasOne(r => r.Task)
|
|
.WithMany(t => t.Runs)
|
|
.HasForeignKey(r => r.TaskId)
|
|
.OnDelete(DeleteBehavior.Cascade);
|
|
|
|
builder.HasIndex(r => r.TaskId).HasDatabaseName("idx_task_runs_task_id");
|
|
}
|
|
}
|