mirror of
https://github.com/multica-ai/multica.git
synced 2026-07-23 18:17:44 +02:00
2ea778796a
Previously only Claude and Codex had log-scanning-level token usage reporting (Flow B). This adds scanners for the remaining three runtimes: - OpenCode: reads JSON message files from ~/.local/share/opencode/storage/message/ - OpenClaw: reads JSONL session files from ~/.openclaw/agents/*/sessions/ - Hermes: reads JSONL session files from ~/.hermes/sessions/ All three are registered in Scanner.Scan() and follow the same (date, provider, model) aggregation pattern as existing scanners.
94 lines
3.3 KiB
Go
94 lines
3.3 KiB
Go
package usage
|
|
|
|
import (
|
|
"log/slog"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestParseOpenClawFile(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
|
|
// Real OpenClaw session JSONL with session header, model_change, and assistant messages
|
|
content := `{"type":"session","version":3,"id":"multica-test","timestamp":"2026-04-11T13:53:05.847Z"}
|
|
{"type":"model_change","id":"03c18aae","timestamp":"2026-04-11T13:53:05.855Z","provider":"deepseek","modelId":"deepseek-chat"}
|
|
{"type":"message","id":"162ce1b7","parentId":"c90ecabe","timestamp":"2026-04-11T13:53:09.986Z","message":{"role":"assistant","content":[{"type":"text","text":"I'll start by getting the issue details."}],"api":"openai-completions","provider":"deepseek","model":"deepseek-chat","usage":{"input":133,"output":81,"cacheRead":16448,"cacheWrite":0,"totalTokens":16662}}}
|
|
{"type":"message","id":"3c063300","parentId":"50e4feb6","timestamp":"2026-04-11T13:53:14.750Z","message":{"role":"assistant","content":[{"type":"text","text":"Let me check the workspace."}],"provider":"deepseek","model":"deepseek-chat","usage":{"input":286,"output":94,"cacheRead":16448,"cacheWrite":0}}}
|
|
{"type":"message","id":"user001","timestamp":"2026-04-11T13:54:00.000Z","message":{"role":"user","content":[{"type":"text","text":"hello"}]}}
|
|
`
|
|
|
|
filePath := filepath.Join(tmp, "session.jsonl")
|
|
if err := os.WriteFile(filePath, []byte(content), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
s := NewScanner(slog.Default())
|
|
records := s.parseOpenClawFile(filePath)
|
|
|
|
if len(records) != 2 {
|
|
t.Fatalf("expected 2 records, got %d", len(records))
|
|
}
|
|
|
|
r := records[0]
|
|
if r.Provider != "openclaw" {
|
|
t.Errorf("provider = %q, want %q", r.Provider, "openclaw")
|
|
}
|
|
if r.Model != "deepseek/deepseek-chat" {
|
|
t.Errorf("model = %q, want %q", r.Model, "deepseek/deepseek-chat")
|
|
}
|
|
if r.InputTokens != 133 {
|
|
t.Errorf("input_tokens = %d, want %d", r.InputTokens, 133)
|
|
}
|
|
if r.OutputTokens != 81 {
|
|
t.Errorf("output_tokens = %d, want %d", r.OutputTokens, 81)
|
|
}
|
|
if r.CacheReadTokens != 16448 {
|
|
t.Errorf("cache_read_tokens = %d, want %d", r.CacheReadTokens, 16448)
|
|
}
|
|
if r.Date != "2026-04-11" {
|
|
t.Errorf("date = %q, want %q", r.Date, "2026-04-11")
|
|
}
|
|
}
|
|
|
|
func TestParseOpenClawFile_NoUsage(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
|
|
// Session with no assistant messages containing usage
|
|
content := `{"type":"session","version":3,"id":"empty-session","timestamp":"2026-04-11T13:53:05.847Z"}
|
|
{"type":"message","id":"user001","timestamp":"2026-04-11T13:54:00.000Z","message":{"role":"user","content":[{"type":"text","text":"hello"}]}}
|
|
`
|
|
|
|
filePath := filepath.Join(tmp, "session.jsonl")
|
|
if err := os.WriteFile(filePath, []byte(content), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
s := NewScanner(slog.Default())
|
|
records := s.parseOpenClawFile(filePath)
|
|
|
|
if len(records) != 0 {
|
|
t.Errorf("expected 0 records, got %d", len(records))
|
|
}
|
|
}
|
|
|
|
func TestNormalizeOpenClawModel(t *testing.T) {
|
|
tests := []struct {
|
|
provider string
|
|
model string
|
|
want string
|
|
}{
|
|
{"deepseek", "deepseek-chat", "deepseek/deepseek-chat"},
|
|
{"anthropic", "claude-sonnet-4-5", "anthropic/claude-sonnet-4-5"},
|
|
{"", "gpt-4o", "gpt-4o"},
|
|
{"openai", "openai/gpt-4o", "openai/gpt-4o"}, // already has /
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
got := normalizeOpenClawModel(tt.provider, tt.model)
|
|
if got != tt.want {
|
|
t.Errorf("normalizeOpenClawModel(%q, %q) = %q, want %q", tt.provider, tt.model, got, tt.want)
|
|
}
|
|
}
|
|
}
|