tokenlens-cli 1.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +164 -0
- package/dist/analyzer/index.d.ts +44 -0
- package/dist/analyzer/index.d.ts.map +1 -0
- package/dist/analyzer/index.js +280 -0
- package/dist/analyzer/index.js.map +1 -0
- package/dist/cli/commands.d.ts +3 -0
- package/dist/cli/commands.d.ts.map +1 -0
- package/dist/cli/commands.js +430 -0
- package/dist/cli/commands.js.map +1 -0
- package/dist/config.d.ts +29 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +106 -0
- package/dist/config.js.map +1 -0
- package/dist/dashboard/server.d.ts +12 -0
- package/dist/dashboard/server.d.ts.map +1 -0
- package/dist/dashboard/server.js +195 -0
- package/dist/dashboard/server.js.map +1 -0
- package/dist/db/schema.d.ts +34 -0
- package/dist/db/schema.d.ts.map +1 -0
- package/dist/db/schema.js +118 -0
- package/dist/db/schema.js.map +1 -0
- package/dist/db/store.d.ts +62 -0
- package/dist/db/store.d.ts.map +1 -0
- package/dist/db/store.js +110 -0
- package/dist/db/store.js.map +1 -0
- package/dist/importers/logs.d.ts +11 -0
- package/dist/importers/logs.d.ts.map +1 -0
- package/dist/importers/logs.js +243 -0
- package/dist/importers/logs.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +31 -0
- package/dist/index.js.map +1 -0
- package/dist/proxy/server.d.ts +11 -0
- package/dist/proxy/server.d.ts.map +1 -0
- package/dist/proxy/server.js +153 -0
- package/dist/proxy/server.js.map +1 -0
- package/package.json +51 -0
- package/patches/http-mitm-proxy+1.1.0.patch +13 -0
|
@@ -0,0 +1,243 @@
|
|
|
1
|
+
import { createHash } from "crypto";
|
|
2
|
+
import { createReadStream, existsSync, readdirSync, statSync } from "fs";
|
|
3
|
+
import { basename, extname, join, resolve } from "path";
|
|
4
|
+
import { createInterface } from "readline";
|
|
5
|
+
import { defaultImportPaths } from "../config.js";
|
|
6
|
+
function listDataFiles(root) {
|
|
7
|
+
if (!existsSync(root))
|
|
8
|
+
return [];
|
|
9
|
+
const stat = statSync(root);
|
|
10
|
+
if (stat.isFile())
|
|
11
|
+
return [root];
|
|
12
|
+
const files = [];
|
|
13
|
+
for (const entry of readdirSync(root, { withFileTypes: true })) {
|
|
14
|
+
const path = join(root, entry.name);
|
|
15
|
+
if (entry.isDirectory())
|
|
16
|
+
files.push(...listDataFiles(path));
|
|
17
|
+
else if ([".jsonl", ".json", ".ndjson"].includes(extname(entry.name).toLowerCase()))
|
|
18
|
+
files.push(path);
|
|
19
|
+
}
|
|
20
|
+
return files;
|
|
21
|
+
}
|
|
22
|
+
function textContent(value) {
|
|
23
|
+
if (typeof value === "string")
|
|
24
|
+
return value;
|
|
25
|
+
if (Array.isArray(value)) {
|
|
26
|
+
return value
|
|
27
|
+
.map((item) => typeof item === "string" ? item : item?.text ?? item?.content ?? "")
|
|
28
|
+
.filter(Boolean)
|
|
29
|
+
.join(" ");
|
|
30
|
+
}
|
|
31
|
+
return "";
|
|
32
|
+
}
|
|
33
|
+
function usageFrom(record) {
|
|
34
|
+
const usage = record.usage ?? record.message?.usage ?? record.response?.usage ?? record.usageMetadata;
|
|
35
|
+
if (usage) {
|
|
36
|
+
const cached = usage.cached_input_tokens ??
|
|
37
|
+
usage.cache_read_input_tokens ??
|
|
38
|
+
usage.prompt_tokens_details?.cached_tokens ??
|
|
39
|
+
0;
|
|
40
|
+
const cacheCreation = usage.cache_creation_input_tokens ?? 0;
|
|
41
|
+
const prompt = usage.prompt_tokens ??
|
|
42
|
+
usage.input_tokens ??
|
|
43
|
+
usage.promptTokenCount ??
|
|
44
|
+
0;
|
|
45
|
+
const completion = usage.completion_tokens ??
|
|
46
|
+
usage.output_tokens ??
|
|
47
|
+
usage.candidatesTokenCount ??
|
|
48
|
+
0;
|
|
49
|
+
const reasoning = usage.reasoning_output_tokens ??
|
|
50
|
+
usage.completion_tokens_details?.reasoning_tokens ??
|
|
51
|
+
0;
|
|
52
|
+
return {
|
|
53
|
+
prompt: prompt + cached + cacheCreation,
|
|
54
|
+
completion,
|
|
55
|
+
cached,
|
|
56
|
+
reasoning,
|
|
57
|
+
};
|
|
58
|
+
}
|
|
59
|
+
if (record.prompt_eval_count !== undefined || record.eval_count !== undefined) {
|
|
60
|
+
return {
|
|
61
|
+
prompt: record.prompt_eval_count ?? 0,
|
|
62
|
+
completion: record.eval_count ?? 0,
|
|
63
|
+
cached: 0,
|
|
64
|
+
reasoning: 0,
|
|
65
|
+
};
|
|
66
|
+
}
|
|
67
|
+
return null;
|
|
68
|
+
}
|
|
69
|
+
function externalId(source, path, line) {
|
|
70
|
+
return createHash("sha1").update(`${source}:${path}:${line}`).digest("hex");
|
|
71
|
+
}
|
|
72
|
+
function normalizePrompt(value) {
|
|
73
|
+
return value.toLowerCase().replace(/\s+/g, " ").trim();
|
|
74
|
+
}
|
|
75
|
+
function promptHash(value) {
|
|
76
|
+
const normalized = normalizePrompt(value);
|
|
77
|
+
if (!normalized)
|
|
78
|
+
return null;
|
|
79
|
+
return createHash("sha1").update(normalized).digest("hex");
|
|
80
|
+
}
|
|
81
|
+
function promptFamilyHash(value) {
|
|
82
|
+
const normalized = normalizePrompt(value);
|
|
83
|
+
if (!normalized)
|
|
84
|
+
return null;
|
|
85
|
+
const family = normalized.split(" ").slice(0, 40).join(" ");
|
|
86
|
+
if (!family)
|
|
87
|
+
return null;
|
|
88
|
+
return createHash("sha1").update(family).digest("hex");
|
|
89
|
+
}
|
|
90
|
+
async function importFile(store, path, sourceHint) {
|
|
91
|
+
const normalized = path.toLowerCase();
|
|
92
|
+
const source = sourceHint ??
|
|
93
|
+
(normalized.includes(`${join(".codex", "sessions").toLowerCase()}`) ? "codex" :
|
|
94
|
+
normalized.includes(`${join(".claude", "projects").toLowerCase()}`) ? "claude" :
|
|
95
|
+
"generic");
|
|
96
|
+
const sessionId = basename(path, extname(path)).replace(/^rollout-[^-]+-[^-]+-/, "");
|
|
97
|
+
const reader = createInterface({ input: createReadStream(path, { encoding: "utf8" }), crlfDelay: Infinity });
|
|
98
|
+
let imported = 0;
|
|
99
|
+
let skipped = 0;
|
|
100
|
+
let lineNumber = 0;
|
|
101
|
+
let model = source === "codex" ? "codex" : "unknown";
|
|
102
|
+
let promptText = "";
|
|
103
|
+
for await (const line of reader) {
|
|
104
|
+
lineNumber++;
|
|
105
|
+
if (!line.trim())
|
|
106
|
+
continue;
|
|
107
|
+
let record;
|
|
108
|
+
try {
|
|
109
|
+
record = JSON.parse(line);
|
|
110
|
+
}
|
|
111
|
+
catch {
|
|
112
|
+
skipped++;
|
|
113
|
+
continue;
|
|
114
|
+
}
|
|
115
|
+
if (source === "codex") {
|
|
116
|
+
if (record.type === "turn_context" && record.payload?.model)
|
|
117
|
+
model = record.payload.model;
|
|
118
|
+
if (record.type === "response_item" && record.payload?.role === "user") {
|
|
119
|
+
promptText = textContent(record.payload.content);
|
|
120
|
+
}
|
|
121
|
+
const last = record.payload?.type === "token_count"
|
|
122
|
+
? record.payload?.info?.last_token_usage
|
|
123
|
+
: null;
|
|
124
|
+
if (!last)
|
|
125
|
+
continue;
|
|
126
|
+
const currentPrompt = promptText.slice(0, 500);
|
|
127
|
+
const result = store.insertCall({
|
|
128
|
+
toolName: "Codex",
|
|
129
|
+
source,
|
|
130
|
+
provider: "OpenAI",
|
|
131
|
+
model,
|
|
132
|
+
endpoint: "local-log",
|
|
133
|
+
promptTokens: last.input_tokens ?? 0,
|
|
134
|
+
cachedTokens: last.cached_input_tokens ?? 0,
|
|
135
|
+
completionTokens: last.output_tokens ?? 0,
|
|
136
|
+
reasoningTokens: last.reasoning_output_tokens ?? 0,
|
|
137
|
+
promptPreview: currentPrompt,
|
|
138
|
+
completionPreview: "",
|
|
139
|
+
sessionId,
|
|
140
|
+
externalId: externalId(source, path, lineNumber),
|
|
141
|
+
promptHash: promptHash(promptText),
|
|
142
|
+
promptFamilyHash: promptFamilyHash(promptText),
|
|
143
|
+
createdAt: record.timestamp,
|
|
144
|
+
});
|
|
145
|
+
if (result > 0)
|
|
146
|
+
imported++;
|
|
147
|
+
else
|
|
148
|
+
skipped++;
|
|
149
|
+
continue;
|
|
150
|
+
}
|
|
151
|
+
if (source === "claude") {
|
|
152
|
+
if (record.type === "user")
|
|
153
|
+
promptText = textContent(record.message?.content);
|
|
154
|
+
if (!record.message?.usage)
|
|
155
|
+
continue;
|
|
156
|
+
const usage = usageFrom(record);
|
|
157
|
+
if (!usage || usage.prompt + usage.completion === 0)
|
|
158
|
+
continue;
|
|
159
|
+
const currentPrompt = promptText.slice(0, 500);
|
|
160
|
+
const result = store.insertCall({
|
|
161
|
+
toolName: "Claude Code",
|
|
162
|
+
source,
|
|
163
|
+
provider: "Anthropic",
|
|
164
|
+
model: record.message?.model ?? "claude",
|
|
165
|
+
endpoint: "local-log",
|
|
166
|
+
promptTokens: usage.prompt,
|
|
167
|
+
cachedTokens: usage.cached,
|
|
168
|
+
completionTokens: usage.completion,
|
|
169
|
+
reasoningTokens: usage.reasoning,
|
|
170
|
+
promptPreview: currentPrompt,
|
|
171
|
+
completionPreview: textContent(record.message?.content).slice(0, 500),
|
|
172
|
+
sessionId: record.sessionId ?? sessionId,
|
|
173
|
+
externalId: externalId(source, path, lineNumber),
|
|
174
|
+
promptHash: promptHash(promptText),
|
|
175
|
+
promptFamilyHash: promptFamilyHash(promptText),
|
|
176
|
+
createdAt: record.timestamp,
|
|
177
|
+
});
|
|
178
|
+
if (result > 0)
|
|
179
|
+
imported++;
|
|
180
|
+
else
|
|
181
|
+
skipped++;
|
|
182
|
+
continue;
|
|
183
|
+
}
|
|
184
|
+
const usage = usageFrom(record);
|
|
185
|
+
if (!usage || usage.prompt + usage.completion === 0)
|
|
186
|
+
continue;
|
|
187
|
+
const genericPrompt = textContent(record.prompt ?? record.messages ?? record.request?.messages);
|
|
188
|
+
const result = store.insertCall({
|
|
189
|
+
toolName: record.tool ?? record.provider ?? sourceHint ?? "Imported",
|
|
190
|
+
source,
|
|
191
|
+
provider: record.provider ?? record.vendor ?? sourceHint ?? "Imported",
|
|
192
|
+
model: record.model ?? record.response?.model ?? "unknown",
|
|
193
|
+
endpoint: record.endpoint ?? record.provider ?? "import",
|
|
194
|
+
promptTokens: usage.prompt,
|
|
195
|
+
cachedTokens: usage.cached,
|
|
196
|
+
completionTokens: usage.completion,
|
|
197
|
+
reasoningTokens: usage.reasoning,
|
|
198
|
+
promptPreview: genericPrompt.slice(0, 500),
|
|
199
|
+
completionPreview: textContent(record.response ?? record.message?.content).slice(0, 500),
|
|
200
|
+
sessionId: record.session_id ?? record.sessionId ?? null,
|
|
201
|
+
externalId: externalId(source, path, lineNumber),
|
|
202
|
+
promptHash: promptHash(genericPrompt),
|
|
203
|
+
promptFamilyHash: promptFamilyHash(genericPrompt),
|
|
204
|
+
createdAt: record.created_at ?? record.timestamp,
|
|
205
|
+
});
|
|
206
|
+
if (result > 0)
|
|
207
|
+
imported++;
|
|
208
|
+
else
|
|
209
|
+
skipped++;
|
|
210
|
+
}
|
|
211
|
+
return { imported, skipped, source };
|
|
212
|
+
}
|
|
213
|
+
export async function importLogs(store, paths = [], force = false, configuredDefaults = defaultImportPaths()) {
|
|
214
|
+
const defaults = configuredDefaults.length ? configuredDefaults : defaultImportPaths();
|
|
215
|
+
const roots = paths.length ? paths.map((path) => resolve(path)) : defaults;
|
|
216
|
+
const files = roots.flatMap(listDataFiles);
|
|
217
|
+
const result = {
|
|
218
|
+
filesFound: files.length,
|
|
219
|
+
filesRead: 0,
|
|
220
|
+
callsImported: 0,
|
|
221
|
+
callsSkipped: 0,
|
|
222
|
+
errors: [],
|
|
223
|
+
sources: {},
|
|
224
|
+
};
|
|
225
|
+
for (const path of files) {
|
|
226
|
+
try {
|
|
227
|
+
const stat = statSync(path);
|
|
228
|
+
if (!force && store.isFileImported(path, stat.size, Math.round(stat.mtimeMs)))
|
|
229
|
+
continue;
|
|
230
|
+
const fileResult = await importFile(store, path);
|
|
231
|
+
result.filesRead++;
|
|
232
|
+
result.callsImported += fileResult.imported;
|
|
233
|
+
result.callsSkipped += fileResult.skipped;
|
|
234
|
+
result.sources[fileResult.source] = (result.sources[fileResult.source] ?? 0) + fileResult.imported;
|
|
235
|
+
store.markFileImported(path, stat.size, Math.round(stat.mtimeMs));
|
|
236
|
+
}
|
|
237
|
+
catch (error) {
|
|
238
|
+
result.errors.push(`${path}: ${error instanceof Error ? error.message : String(error)}`);
|
|
239
|
+
}
|
|
240
|
+
}
|
|
241
|
+
return result;
|
|
242
|
+
}
|
|
243
|
+
//# sourceMappingURL=logs.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"logs.js","sourceRoot":"","sources":["../../src/importers/logs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,gBAAgB,EAAE,UAAU,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAC;AACzE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3C,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAclD,SAAS,aAAa,CAAC,IAAY;IACjC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IACjC,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC5B,IAAI,IAAI,CAAC,MAAM,EAAE;QAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjC,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QAC/D,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,KAAK,CAAC,WAAW,EAAE;YAAE,KAAK,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;aACvD,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;YAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxG,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,WAAW,CAAC,KAAc;IACjC,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5C,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK;aACT,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,IAAI,IAAI,EAAE,OAAO,IAAI,EAAE,CAAC;aAClF,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,CAAC,GAAG,CAAC,CAAC;IACf,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,SAAS,CAAC,MAAkB;IAMnC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,OAAO,EAAE,KAAK,IAAI,MAAM,CAAC,QAAQ,EAAE,KAAK,IAAI,MAAM,CAAC,aAAa,CAAC;IACtG,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,MAAM,GACV,KAAK,CAAC,mBAAmB;YACzB,KAAK,CAAC,uBAAuB;YAC7B,KAAK,CAAC,qBAAqB,EAAE,aAAa;YAC1C,CAAC,CAAC;QACJ,MAAM,aAAa,GAAG,KAAK,CAAC,2BAA2B,IAAI,CAAC,CAAC;QAC7D,MAAM,MAAM,GACV,KAAK,CAAC,aAAa;YACnB,KAAK,CAAC,YAAY;YAClB,KAAK,CAAC,gBAAgB;YACtB,CAAC,CAAC;QACJ,MAAM,UAAU,GACd,KAAK,CAAC,iBAAiB;YACvB,KAAK,CAAC,aAAa;YACnB,KAAK,CAAC,oBAAoB;YAC1B,CAAC,CAAC;QACJ,MAAM,SAAS,GACb,KAAK,CAAC,uBAAuB;YAC7B,KAAK,CAAC,yBAAyB,EAAE,gBAAgB;YACjD,CAAC,CAAC;QACJ,OAAO;YACL,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,aAAa;YACvC,UAAU;YACV,MAAM;YACN,SAAS;SACV,CAAC;IACJ,CAAC;IAED,IAAI,MAAM,CAAC,iBAAiB,KAAK,SAAS,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QAC9E,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,iBAAiB,IAAI,CAAC;YACrC,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,CAAC;YAClC,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;SACb,CAAC;IACJ,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,UAAU,CAAC,MAAc,EAAE,IAAY,EAAE,IAAY;IAC5D,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,GAAG,MAAM,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,eAAe,CAAC,KAAa;IACpC,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;AACzD,CAAC;AAED,SAAS,UAAU,CAAC,KAAa;IAC/B,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAa;IACrC,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,MAAM,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC5D,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IACzB,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACzD,CAAC;AAED,KAAK,UAAU,UAAU,CAAC,KAAiB,EAAE,IAAY,EAAE,UAAmB;IAK5E,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACtC,MAAM,MAAM,GAAG,UAAU;QACvB,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YAC7E,UAAU,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;gBAC9E,SAAS,CAAC,CAAC;IACjB,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC;IACrF,MAAM,MAAM,GAAG,eAAe,CAAC,EAAE,KAAK,EAAE,gBAAgB,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC7G,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,KAAK,GAAG,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;IACrD,IAAI,UAAU,GAAG,EAAE,CAAC;IAEpB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;QAChC,UAAU,EAAE,CAAC;QACb,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;YAAE,SAAS;QAC3B,IAAI,MAAkB,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;YACV,SAAS;QACX,CAAC;QAED,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YACvB,IAAI,MAAM,CAAC,IAAI,KAAK,cAAc,IAAI,MAAM,CAAC,OAAO,EAAE,KAAK;gBAAE,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC;YAC1F,IAAI,MAAM,CAAC,IAAI,KAAK,eAAe,IAAI,MAAM,CAAC,OAAO,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;gBACvE,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,EAAE,IAAI,KAAK,aAAa;gBACjD,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,gBAAgB;gBACxC,CAAC,CAAC,IAAI,CAAC;YACT,IAAI,CAAC,IAAI;gBAAE,SAAS;YACpB,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC;gBAC9B,QAAQ,EAAE,OAAO;gBACjB,MAAM;gBACN,QAAQ,EAAE,QAAQ;gBAClB,KAAK;gBACL,QAAQ,EAAE,WAAW;gBACrB,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,CAAC;gBACpC,YAAY,EAAE,IAAI,CAAC,mBAAmB,IAAI,CAAC;gBAC3C,gBAAgB,EAAE,IAAI,CAAC,aAAa,IAAI,CAAC;gBACzC,eAAe,EAAE,IAAI,CAAC,uBAAuB,IAAI,CAAC;gBAClD,aAAa,EAAE,aAAa;gBAC5B,iBAAiB,EAAE,EAAE;gBACrB,SAAS;gBACT,UAAU,EAAE,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC;gBAChD,UAAU,EAAE,UAAU,CAAC,UAAU,CAAC;gBAClC,gBAAgB,EAAE,gBAAgB,CAAC,UAAU,CAAC;gBAC9C,SAAS,EAAE,MAAM,CAAC,SAAS;aAC5B,CAAC,CAAC;YACH,IAAI,MAAM,GAAG,CAAC;gBAAE,QAAQ,EAAE,CAAC;;gBAAM,OAAO,EAAE,CAAC;YAC3C,SAAS;QACX,CAAC;QAED,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YACxB,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM;gBAAE,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9E,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,KAAK;gBAAE,SAAS;YACrC,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;YAChC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,UAAU,KAAK,CAAC;gBAAE,SAAS;YAC9D,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC;gBAC9B,QAAQ,EAAE,aAAa;gBACvB,MAAM;gBACN,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,KAAK,IAAI,QAAQ;gBACxC,QAAQ,EAAE,WAAW;gBACrB,YAAY,EAAE,KAAK,CAAC,MAAM;gBAC1B,YAAY,EAAE,KAAK,CAAC,MAAM;gBAC1B,gBAAgB,EAAE,KAAK,CAAC,UAAU;gBAClC,eAAe,EAAE,KAAK,CAAC,SAAS;gBAChC,aAAa,EAAE,aAAa;gBAC5B,iBAAiB,EAAE,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;gBACrE,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,SAAS;gBACxC,UAAU,EAAE,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC;gBAChD,UAAU,EAAE,UAAU,CAAC,UAAU,CAAC;gBAClC,gBAAgB,EAAE,gBAAgB,CAAC,UAAU,CAAC;gBAC9C,SAAS,EAAE,MAAM,CAAC,SAAS;aAC5B,CAAC,CAAC;YACH,IAAI,MAAM,GAAG,CAAC;gBAAE,QAAQ,EAAE,CAAC;;gBAAM,OAAO,EAAE,CAAC;YAC3C,SAAS;QACX,CAAC;QAED,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,UAAU,KAAK,CAAC;YAAE,SAAS;QAC9D,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAChG,MAAM,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC;YAC9B,QAAQ,EAAE,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,QAAQ,IAAI,UAAU,IAAI,UAAU;YACpE,MAAM;YACN,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,MAAM,IAAI,UAAU,IAAI,UAAU;YACtE,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,QAAQ,EAAE,KAAK,IAAI,SAAS;YAC1D,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,IAAI,QAAQ;YACxD,YAAY,EAAE,KAAK,CAAC,MAAM;YAC1B,YAAY,EAAE,KAAK,CAAC,MAAM;YAC1B,gBAAgB,EAAE,KAAK,CAAC,UAAU;YAClC,eAAe,EAAE,KAAK,CAAC,SAAS;YAChC,aAAa,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;YAC1C,iBAAiB,EAAE,WAAW,CAAC,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;YACxF,SAAS,EAAE,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,SAAS,IAAI,IAAI;YACxD,UAAU,EAAE,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC;YAChD,UAAU,EAAE,UAAU,CAAC,aAAa,CAAC;YACrC,gBAAgB,EAAE,gBAAgB,CAAC,aAAa,CAAC;YACjD,SAAS,EAAE,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,SAAS;SACjD,CAAC,CAAC;QACH,IAAI,MAAM,GAAG,CAAC;YAAE,QAAQ,EAAE,CAAC;;YAAM,OAAO,EAAE,CAAC;IAC7C,CAAC;IACD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AACvC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,KAAiB,EACjB,QAAkB,EAAE,EACpB,KAAK,GAAG,KAAK,EACb,qBAA+B,kBAAkB,EAAE;IAEnD,MAAM,QAAQ,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAC;IACvF,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;IAC3E,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IAC3C,MAAM,MAAM,GAAiB;QAC3B,UAAU,EAAE,KAAK,CAAC,MAAM;QACxB,SAAS,EAAE,CAAC;QACZ,aAAa,EAAE,CAAC;QAChB,YAAY,EAAE,CAAC;QACf,MAAM,EAAE,EAAE;QACV,OAAO,EAAE,EAAE;KACZ,CAAC;IAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC5B,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,cAAc,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAAE,SAAS;YACxF,MAAM,UAAU,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,CAAC,aAAa,IAAI,UAAU,CAAC,QAAQ,CAAC;YAC5C,MAAM,CAAC,YAAY,IAAI,UAAU,CAAC,OAAO,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC;YACnG,KAAK,CAAC,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;QACpE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC3F,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
package/dist/index.d.ts
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":""}
|
package/dist/index.js
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
import { loadConfig } from "./config.js";
|
|
3
|
+
import { TokenStore } from "./db/store.js";
|
|
4
|
+
import { ProxyServer } from "./proxy/server.js";
|
|
5
|
+
import { DashboardServer } from "./dashboard/server.js";
|
|
6
|
+
const args = process.argv.slice(2);
|
|
7
|
+
const isDashboardOnly = args.includes("--dashboard-only");
|
|
8
|
+
const isProxyOnly = args.includes("--proxy-only");
|
|
9
|
+
const config = loadConfig();
|
|
10
|
+
const store = new TokenStore(config.dbPath);
|
|
11
|
+
if (isDashboardOnly) {
|
|
12
|
+
const dash = new DashboardServer(store);
|
|
13
|
+
dash.start(config.dashboardPort);
|
|
14
|
+
process.on("SIGINT", () => { dash.stop(); store.close(); process.exit(0); });
|
|
15
|
+
process.on("SIGTERM", () => { dash.stop(); store.close(); process.exit(0); });
|
|
16
|
+
}
|
|
17
|
+
else if (isProxyOnly) {
|
|
18
|
+
const proxy = new ProxyServer(store);
|
|
19
|
+
proxy.start(config.proxyPort);
|
|
20
|
+
process.on("SIGINT", () => { proxy.stop(); store.close(); process.exit(0); });
|
|
21
|
+
process.on("SIGTERM", () => { proxy.stop(); store.close(); process.exit(0); });
|
|
22
|
+
}
|
|
23
|
+
else {
|
|
24
|
+
const proxy = new ProxyServer(store);
|
|
25
|
+
const dash = new DashboardServer(store);
|
|
26
|
+
proxy.start(config.proxyPort);
|
|
27
|
+
dash.start(config.dashboardPort);
|
|
28
|
+
process.on("SIGINT", () => { proxy.stop(); dash.stop(); store.close(); process.exit(0); });
|
|
29
|
+
process.on("SIGTERM", () => { proxy.stop(); dash.stop(); store.close(); process.exit(0); });
|
|
30
|
+
}
|
|
31
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AACnC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;AAC1D,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;AAElD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;AAC5B,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;AAE5C,IAAI,eAAe,EAAE,CAAC;IACpB,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;IACxC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IACjC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7E,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAChF,CAAC;KAAM,IAAI,WAAW,EAAE,CAAC;IACvB,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;IACrC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAC9B,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9E,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACjF,CAAC;KAAM,CAAC;IACN,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;IACrC,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;IAExC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAC9B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IAEjC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3F,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC9F,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { TokenStore } from "../db/store.js";
|
|
2
|
+
export declare class ProxyServer {
|
|
3
|
+
private proxy;
|
|
4
|
+
private store;
|
|
5
|
+
private toolName;
|
|
6
|
+
constructor(store: TokenStore, toolName?: string);
|
|
7
|
+
start(port: number): void;
|
|
8
|
+
stop(): void;
|
|
9
|
+
private processCall;
|
|
10
|
+
}
|
|
11
|
+
//# sourceMappingURL=server.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/proxy/server.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAoBjD,qBAAa,WAAW;IACtB,OAAO,CAAC,KAAK,CAAe;IAC5B,OAAO,CAAC,KAAK,CAAa;IAC1B,OAAO,CAAC,QAAQ,CAAS;gBAEb,KAAK,EAAE,UAAU,EAAE,QAAQ,GAAE,MAAiB;IAK1D,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAwDzB,IAAI,IAAI,IAAI;IAIZ,OAAO,CAAC,WAAW;CAgFpB"}
|
|
@@ -0,0 +1,153 @@
|
|
|
1
|
+
import { createHash } from "crypto";
|
|
2
|
+
import { Proxy } from "http-mitm-proxy";
|
|
3
|
+
import { AI_ENDPOINTS } from "../config.js";
|
|
4
|
+
function normalizePrompt(value) {
|
|
5
|
+
return value.toLowerCase().replace(/\s+/g, " ").trim();
|
|
6
|
+
}
|
|
7
|
+
function promptHash(value) {
|
|
8
|
+
const normalized = normalizePrompt(value);
|
|
9
|
+
if (!normalized)
|
|
10
|
+
return null;
|
|
11
|
+
return createHash("sha1").update(normalized).digest("hex");
|
|
12
|
+
}
|
|
13
|
+
function promptFamilyHash(value) {
|
|
14
|
+
const normalized = normalizePrompt(value);
|
|
15
|
+
if (!normalized)
|
|
16
|
+
return null;
|
|
17
|
+
const family = normalized.split(" ").slice(0, 40).join(" ");
|
|
18
|
+
if (!family)
|
|
19
|
+
return null;
|
|
20
|
+
return createHash("sha1").update(family).digest("hex");
|
|
21
|
+
}
|
|
22
|
+
export class ProxyServer {
|
|
23
|
+
proxy = new Proxy();
|
|
24
|
+
store;
|
|
25
|
+
toolName;
|
|
26
|
+
constructor(store, toolName = "system") {
|
|
27
|
+
this.store = store;
|
|
28
|
+
this.toolName = toolName;
|
|
29
|
+
}
|
|
30
|
+
start(port) {
|
|
31
|
+
this.proxy.onError((_ctx, err) => {
|
|
32
|
+
if (err?.code !== "ECONNRESET") {
|
|
33
|
+
console.error("[proxy] Error:", err?.message);
|
|
34
|
+
}
|
|
35
|
+
});
|
|
36
|
+
this.proxy.onRequest((ctx, callback) => {
|
|
37
|
+
const host = ctx.clientToProxyRequest.headers.host || "";
|
|
38
|
+
const aiEndpoint = AI_ENDPOINTS.find((e) => host.includes(e.host));
|
|
39
|
+
if (!aiEndpoint)
|
|
40
|
+
return callback();
|
|
41
|
+
const req = ctx.clientToProxyRequest;
|
|
42
|
+
const startTime = Date.now();
|
|
43
|
+
const chunks = [];
|
|
44
|
+
req.on("data", (chunk) => chunks.push(chunk));
|
|
45
|
+
req.on("end", () => {
|
|
46
|
+
ctx.userData = {
|
|
47
|
+
requestBody: Buffer.concat(chunks).toString(),
|
|
48
|
+
startTime,
|
|
49
|
+
aiEndpoint,
|
|
50
|
+
};
|
|
51
|
+
});
|
|
52
|
+
callback();
|
|
53
|
+
});
|
|
54
|
+
this.proxy.onResponse((ctx, callback) => {
|
|
55
|
+
const ud = ctx.userData;
|
|
56
|
+
if (ud?.requestBody && ud.aiEndpoint) {
|
|
57
|
+
const res = ctx.serverToProxyResponse;
|
|
58
|
+
if (!res)
|
|
59
|
+
return callback();
|
|
60
|
+
const respChunks = [];
|
|
61
|
+
res.on("data", (chunk) => respChunks.push(chunk));
|
|
62
|
+
res.on("end", () => {
|
|
63
|
+
const responseBody = Buffer.concat(respChunks).toString();
|
|
64
|
+
this.processCall(ud.requestBody, responseBody, ud.aiEndpoint);
|
|
65
|
+
});
|
|
66
|
+
}
|
|
67
|
+
callback();
|
|
68
|
+
});
|
|
69
|
+
this.proxy.listen({ port, host: "127.0.0.1", sslCaDir: undefined });
|
|
70
|
+
console.log(`[proxy] MITM proxy listening on port ${port}`);
|
|
71
|
+
}
|
|
72
|
+
stop() {
|
|
73
|
+
this.proxy.close();
|
|
74
|
+
}
|
|
75
|
+
processCall(requestBody, responseBody, endpoint) {
|
|
76
|
+
try {
|
|
77
|
+
const req = JSON.parse(requestBody);
|
|
78
|
+
const model = req.model || "unknown";
|
|
79
|
+
let promptTokens = 0;
|
|
80
|
+
let completionTokens = 0;
|
|
81
|
+
let promptPreview = "";
|
|
82
|
+
let completionPreview = "";
|
|
83
|
+
let promptText = "";
|
|
84
|
+
if (responseBody) {
|
|
85
|
+
const res = JSON.parse(responseBody);
|
|
86
|
+
// OpenAI format
|
|
87
|
+
if (res.usage) {
|
|
88
|
+
promptTokens =
|
|
89
|
+
res.usage.prompt_tokens ??
|
|
90
|
+
res.usage.input_tokens ??
|
|
91
|
+
0;
|
|
92
|
+
completionTokens =
|
|
93
|
+
res.usage.completion_tokens ??
|
|
94
|
+
res.usage.output_tokens ??
|
|
95
|
+
0;
|
|
96
|
+
}
|
|
97
|
+
// Google Gemini format
|
|
98
|
+
else if (res.usageMetadata) {
|
|
99
|
+
promptTokens = res.usageMetadata.promptTokenCount ?? 0;
|
|
100
|
+
completionTokens = res.usageMetadata.candidatesTokenCount ?? 0;
|
|
101
|
+
}
|
|
102
|
+
// Build completion preview
|
|
103
|
+
if (res.content?.length) {
|
|
104
|
+
const text = res.content
|
|
105
|
+
.map((c) => (c.type === "text" ? c.text : ""))
|
|
106
|
+
.filter(Boolean)
|
|
107
|
+
.join(" ");
|
|
108
|
+
completionPreview = text.slice(0, 200);
|
|
109
|
+
}
|
|
110
|
+
else if (res.choices?.length) {
|
|
111
|
+
completionPreview = (res.choices[0]?.message?.content ?? "").slice(0, 200);
|
|
112
|
+
}
|
|
113
|
+
else if (res.candidates?.length) {
|
|
114
|
+
completionPreview = JSON.stringify(res.candidates[0]).slice(0, 200);
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
// Build prompt preview
|
|
118
|
+
if (req.messages?.length) {
|
|
119
|
+
const last = req.messages[req.messages.length - 1];
|
|
120
|
+
promptText = typeof last.content === "string" ? last.content : JSON.stringify(last.content ?? last);
|
|
121
|
+
promptPreview = promptText.slice(0, 200);
|
|
122
|
+
}
|
|
123
|
+
else if (req.prompt) {
|
|
124
|
+
promptText = String(req.prompt);
|
|
125
|
+
promptPreview = promptText.slice(0, 200);
|
|
126
|
+
}
|
|
127
|
+
else if (req.contents) {
|
|
128
|
+
promptText = JSON.stringify(req.contents);
|
|
129
|
+
promptPreview = promptText.slice(0, 200);
|
|
130
|
+
}
|
|
131
|
+
if (promptTokens > 0 || completionTokens > 0) {
|
|
132
|
+
this.store.insertCall({
|
|
133
|
+
toolName: this.toolName,
|
|
134
|
+
source: "proxy",
|
|
135
|
+
provider: endpoint.name,
|
|
136
|
+
model,
|
|
137
|
+
endpoint: endpoint.name,
|
|
138
|
+
promptTokens,
|
|
139
|
+
completionTokens,
|
|
140
|
+
promptPreview,
|
|
141
|
+
completionPreview,
|
|
142
|
+
sessionId: null,
|
|
143
|
+
promptHash: promptHash(promptText),
|
|
144
|
+
promptFamilyHash: promptFamilyHash(promptText),
|
|
145
|
+
});
|
|
146
|
+
}
|
|
147
|
+
}
|
|
148
|
+
catch {
|
|
149
|
+
// Parsing errors expected for streaming / non-JSON
|
|
150
|
+
}
|
|
151
|
+
}
|
|
152
|
+
}
|
|
153
|
+
//# sourceMappingURL=server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/proxy/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AACxC,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAG5C,SAAS,eAAe,CAAC,KAAa;IACpC,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;AACzD,CAAC;AAED,SAAS,UAAU,CAAC,KAAa;IAC/B,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAa;IACrC,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,MAAM,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC5D,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IACzB,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACzD,CAAC;AAED,MAAM,OAAO,WAAW;IACd,KAAK,GAAG,IAAI,KAAK,EAAE,CAAC;IACpB,KAAK,CAAa;IAClB,QAAQ,CAAS;IAEzB,YAAY,KAAiB,EAAE,WAAmB,QAAQ;QACxD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,IAAY;QAChB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YAC/B,IAAK,GAAW,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;gBACxC,OAAO,CAAC,KAAK,CAAC,gBAAgB,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;YAChD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,GAAG,EAAE,QAAQ,EAAE,EAAE;YACrC,MAAM,IAAI,GAAG,GAAG,CAAC,oBAAoB,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC;YACzD,MAAM,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YAEnE,IAAI,CAAC,UAAU;gBAAE,OAAO,QAAQ,EAAE,CAAC;YAEnC,MAAM,GAAG,GAAG,GAAG,CAAC,oBAAoB,CAAC;YACrC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAa,EAAE,CAAC;YAE5B,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACtD,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBAChB,GAAW,CAAC,QAAQ,GAAG;oBACtB,WAAW,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;oBAC7C,SAAS;oBACT,UAAU;iBACX,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,GAAG,EAAE,QAAQ,EAAE,EAAE;YACtC,MAAM,EAAE,GAAI,GAAW,CAAC,QAKX,CAAC;YAEd,IAAI,EAAE,EAAE,WAAW,IAAI,EAAE,CAAC,UAAU,EAAE,CAAC;gBACrC,MAAM,GAAG,GAAG,GAAG,CAAC,qBAAqB,CAAC;gBACtC,IAAI,CAAC,GAAG;oBAAE,OAAO,QAAQ,EAAE,CAAC;gBAE5B,MAAM,UAAU,GAAa,EAAE,CAAC;gBAChC,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC1D,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBACjB,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,QAAQ,EAAE,CAAC;oBAC1D,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,WAAY,EAAE,YAAY,EAAE,EAAE,CAAC,UAAW,CAAC,CAAC;gBAClE,CAAC,CAAC,CAAC;YACL,CAAC;YAED,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;QACpE,OAAO,CAAC,GAAG,CAAC,wCAAwC,IAAI,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,IAAI;QACF,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAEO,WAAW,CACjB,WAAmB,EACnB,YAAoB,EACpB,QAAwC;QAExC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YACpC,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,IAAI,SAAS,CAAC;YACrC,IAAI,YAAY,GAAG,CAAC,CAAC;YACrB,IAAI,gBAAgB,GAAG,CAAC,CAAC;YACzB,IAAI,aAAa,GAAG,EAAE,CAAC;YACvB,IAAI,iBAAiB,GAAG,EAAE,CAAC;YAC3B,IAAI,UAAU,GAAG,EAAE,CAAC;YAEpB,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBACrC,gBAAgB;gBAChB,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;oBACd,YAAY;wBACV,GAAG,CAAC,KAAK,CAAC,aAAa;4BACvB,GAAG,CAAC,KAAK,CAAC,YAAY;4BACtB,CAAC,CAAC;oBACJ,gBAAgB;wBACd,GAAG,CAAC,KAAK,CAAC,iBAAiB;4BAC3B,GAAG,CAAC,KAAK,CAAC,aAAa;4BACvB,CAAC,CAAC;gBACN,CAAC;gBACD,uBAAuB;qBAClB,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;oBAC3B,YAAY,GAAG,GAAG,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC,CAAC;oBACvD,gBAAgB,GAAG,GAAG,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC,CAAC;gBACjE,CAAC;gBAED,2BAA2B;gBAC3B,IAAI,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;oBACxB,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO;yBACrB,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;yBAClD,MAAM,CAAC,OAAO,CAAC;yBACf,IAAI,CAAC,GAAG,CAAC,CAAC;oBACb,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;gBACzC,CAAC;qBAAM,IAAI,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;oBAC/B,iBAAiB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;gBAC7E,CAAC;qBAAM,IAAI,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;oBAClC,iBAAiB,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;YAED,uBAAuB;YACvB,IAAI,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACzB,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBACnD,UAAU,GAAG,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC;gBACpG,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAC3C,CAAC;iBAAM,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;gBACtB,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAChC,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAC3C,CAAC;iBAAM,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;gBACxB,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBAC1C,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAC3C,CAAC;YAED,IAAI,YAAY,GAAG,CAAC,IAAI,gBAAgB,GAAG,CAAC,EAAE,CAAC;gBAC7C,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC;oBACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,MAAM,EAAE,OAAO;oBACf,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,KAAK;oBACL,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,YAAY;oBACZ,gBAAgB;oBAChB,aAAa;oBACb,iBAAiB;oBACjB,SAAS,EAAE,IAAI;oBACf,UAAU,EAAE,UAAU,CAAC,UAAU,CAAC;oBAClC,gBAAgB,EAAE,gBAAgB,CAAC,UAAU,CAAC;iBAC/C,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,mDAAmD;QACrD,CAAC;IACH,CAAC;CACF"}
|
package/package.json
ADDED
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
{
|
|
2
|
+
"name": "tokenlens-cli",
|
|
3
|
+
"version": "1.0.0",
|
|
4
|
+
"description": "AI token usage monitor that captures, analyzes, and reduces token consumption across coding agents and model providers.",
|
|
5
|
+
"type": "module",
|
|
6
|
+
"files": [
|
|
7
|
+
"dist",
|
|
8
|
+
"README.md",
|
|
9
|
+
"patches"
|
|
10
|
+
],
|
|
11
|
+
"bin": {
|
|
12
|
+
"tokenlens": "dist/cli/commands.js",
|
|
13
|
+
"tl": "dist/cli/commands.js",
|
|
14
|
+
"tg": "dist/cli/commands.js"
|
|
15
|
+
},
|
|
16
|
+
"scripts": {
|
|
17
|
+
"dev": "tsx src/cli/commands.ts",
|
|
18
|
+
"cli": "tsx src/cli/commands.ts",
|
|
19
|
+
"dashboard": "tsx src/index.ts --dashboard-only",
|
|
20
|
+
"proxy": "tsx src/index.ts --proxy-only",
|
|
21
|
+
"build": "tsc",
|
|
22
|
+
"start": "node dist/cli/commands.js start",
|
|
23
|
+
"prepare": "npm run build",
|
|
24
|
+
"prepack": "npm run build",
|
|
25
|
+
"postinstall": "patch-package"
|
|
26
|
+
},
|
|
27
|
+
"keywords": [
|
|
28
|
+
"ai",
|
|
29
|
+
"tokens",
|
|
30
|
+
"usage",
|
|
31
|
+
"cli",
|
|
32
|
+
"observability",
|
|
33
|
+
"prompting"
|
|
34
|
+
],
|
|
35
|
+
"dependencies": {
|
|
36
|
+
"better-sqlite3": "^11.7.0",
|
|
37
|
+
"chalk": "^5.4.1",
|
|
38
|
+
"cli-table3": "^0.6.5",
|
|
39
|
+
"commander": "^13.1.0",
|
|
40
|
+
"express": "^4.21.0",
|
|
41
|
+
"http-mitm-proxy": "^1.1.0"
|
|
42
|
+
},
|
|
43
|
+
"devDependencies": {
|
|
44
|
+
"@types/better-sqlite3": "^7.6.12",
|
|
45
|
+
"@types/express": "^5.0.0",
|
|
46
|
+
"@types/node": "^22.10.0",
|
|
47
|
+
"patch-package": "^8.0.1",
|
|
48
|
+
"tsx": "^4.19.0",
|
|
49
|
+
"typescript": "^5.7.0"
|
|
50
|
+
}
|
|
51
|
+
}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
diff --git a/node_modules/http-mitm-proxy/dist/lib/proxy.js b/node_modules/http-mitm-proxy/dist/lib/proxy.js
|
|
2
|
+
index e896595..e94934d 100644
|
|
3
|
+
--- a/node_modules/http-mitm-proxy/dist/lib/proxy.js
|
|
4
|
+
+++ b/node_modules/http-mitm-proxy/dist/lib/proxy.js
|
|
5
|
+
@@ -361,7 +361,7 @@ class Proxy {
|
|
6
|
+
function makeConnection(port) {
|
|
7
|
+
const conn = net_1.default.connect({
|
|
8
|
+
port,
|
|
9
|
+
- host: "0.0.0.0",
|
|
10
|
+
+ host: "127.0.0.1",
|
|
11
|
+
allowHalfOpen: true,
|
|
12
|
+
}, () => {
|
|
13
|
+
const connectKey = `${conn.localPort}:${conn.remotePort}`;
|