summaryrefslogtreecommitdiffhomepage
path: root/packages
diff options
context:
space:
mode:
Diffstat (limited to 'packages')
-rw-r--r--packages/tui/internal/components/core/status.go6
-rw-r--r--packages/tui/pkg/client/gen/openapi.json18
-rw-r--r--packages/tui/pkg/client/generated-client.go4
3 files changed, 26 insertions, 2 deletions
diff --git a/packages/tui/internal/components/core/status.go b/packages/tui/internal/components/core/status.go
index e9b913642..9540c6c2b 100644
--- a/packages/tui/internal/components/core/status.go
+++ b/packages/tui/internal/components/core/status.go
@@ -102,7 +102,11 @@ func (m statusComponent) View() string {
cost += message.Metadata.Assistant.Cost
usage := message.Metadata.Assistant.Tokens
if usage.Output > 0 {
- tokens = (usage.Input + usage.Output + usage.Reasoning)
+ tokens = (usage.Input +
+ usage.Cache.Write +
+ usage.Cache.Read +
+ usage.Output +
+ usage.Reasoning)
}
}
}
diff --git a/packages/tui/pkg/client/gen/openapi.json b/packages/tui/pkg/client/gen/openapi.json
index c062192ec..9cbfc70e2 100644
--- a/packages/tui/pkg/client/gen/openapi.json
+++ b/packages/tui/pkg/client/gen/openapi.json
@@ -782,12 +782,28 @@
},
"reasoning": {
"type": "number"
+ },
+ "cache": {
+ "type": "object",
+ "properties": {
+ "read": {
+ "type": "number"
+ },
+ "write": {
+ "type": "number"
+ }
+ },
+ "required": [
+ "read",
+ "write"
+ ]
}
},
"required": [
"input",
"output",
- "reasoning"
+ "reasoning",
+ "cache"
]
}
},
diff --git a/packages/tui/pkg/client/generated-client.go b/packages/tui/pkg/client/generated-client.go
index 4e8736d65..59332f0a9 100644
--- a/packages/tui/pkg/client/generated-client.go
+++ b/packages/tui/pkg/client/generated-client.go
@@ -126,6 +126,10 @@ type MessageInfo struct {
Summary *bool `json:"summary,omitempty"`
System []string `json:"system"`
Tokens struct {
+ Cache struct {
+ Read float32 `json:"read"`
+ Write float32 `json:"write"`
+ } `json:"cache"`
Input float32 `json:"input"`
Output float32 `json:"output"`
Reasoning float32 `json:"reasoning"`