@tech-leads-club/harness-toolkit 0.3.6 → 0.4.0

package/dist/price-lookup.mjs

@@ -1,2 +1,2 @@
-import{createRequire as E}from"node:module";var H=E(import.meta.url);function M(P){let x=[],J=!1;for(let G of P){if(G==="--json"){J=!0;continue}x.push(G)}return{json:J,rest:x}}function _(P,x=N){x(`${JSON.stringify(P)}
-`)}function N(P){process.stdout.write(P)}import{existsSync as D,readFileSync as z}from"node:fs";import{join as k}from"node:path";import{homedir as S}from"node:os";import{join as T}from"node:path";function b(){return T(S(),".tlc","harness")}function L(P=process.env){return P.TLC_HOME??b()}var f="litellm",j={"cursor-grok-4.5":"grok-4.5",auto:"auto-cost"};function F(P){if(!P)return{};let{_meta:x,...J}=P;return J}function A(P){return P.trim().toLowerCase().replace(/\[([^\]]+)\]\([^)]*\)/g,"$1").replace(/[[\]]/g,"").replace(/[^a-z0-9.+]+/g,"-").replace(/^-+|-+$/g,"")}function p(P,x){let J=P.trim(),G=[],Q=($)=>{if($&&!G.includes($))G.push($)};if(Q(J),Q(x[J]),Q(A(J)),Q(x[A(J)]),J.includes("/"))Q(J.slice(J.lastIndexOf("/")+1));let Z=J.replace(/-(high|medium|low|max|fast|thinking)$/i,"");if(Z!==J)Q(Z),Q(x[Z]),Q(A(Z));return G}function U(P,x){let J=P[x];if(J)return{key:x,entry:J};for(let[G,Q]of Object.entries(P))if(x.startsWith(G)||G.startsWith(x))return{key:G,entry:Q};return}function v(){return k(L(),"model-prices.json")}function g(){return k(L(),"model-prices.local.json")}var R=new Map;function O(P){if(!D(P))return R.delete(P),null;let x;try{x=z(P,"utf8")}catch{return R.delete(P),null}let J=R.get(P);if(J&&J.text===x)return J.value;let G;try{G=JSON.parse(x)}catch{return R.delete(P),null}return R.set(P,{text:x,value:G}),G}function y(){return O(v())??{}}function C(P,x){return F(P.planes?.[x]??null)}function W(P,x){let J=x.trim();if(!J)return;let G=y(),Q=F(O(g())),Z=P?C(G,P):{},$=C(G,f),B=p(J,j);for(let Y of B){let q=Q[Y];if(q)return{entry:q,key:Y,source:"override"}}for(let Y of B){let q=Z[Y];if(q)return{entry:q,key:Y,source:"provider"}}for(let Y of B){let q=$[Y];if(q)return{entry:q,key:Y,source:"litellm"}}let X=A(J),I=U(Q,X);if(I)return{...I,source:"override"};let K=U(Z,X);if(K)return{...K,source:"provider"};let V=U($,X);if(V)return{...V,source:"litellm"};return}function w(P,x,J){let G=J.inputTokens??0,Q=J.outputTokens??0,Z=J.cacheReadTokens??0,$=J.cacheWriteTokens??0;if(!x||G<=0&&Q<=0&&Z<=0&&$<=0)return{costUsd:null,billing:"unknown",pool:"unknown",source:"missing"};let B=W(P,x);if(!B)return{costUsd:null,billing:"unknown",pool:"unknown",source:"missing"};let{entry:X,key:I,source:K}=B,V=X.pool??"unknown";if(X.billing==="included")return{costUsd:null,billing:"included",pool:V,source:K,catalogKey:I};if(typeof X.promptPer1M!=="number"||typeof X.completionPer1M!=="number")return{costUsd:null,billing:X.billing??"unknown",pool:V,source:K,catalogKey:I};let Y=G/1e6*X.promptPer1M+Q/1e6*X.completionPer1M;if(typeof X.cacheReadPer1M==="number"&&Z>0)Y+=Z/1e6*X.cacheReadPer1M;if(typeof X.cacheWritePer1M==="number"&&$>0)Y+=$/1e6*X.cacheWritePer1M;return{costUsd:Y,billing:"metered",pool:V,source:K,catalogKey:I}}function c(P){let x=P[0];if(!x)return null;return{model:x,provider:P[1]??""}}function m(P){let x=W(P.provider,P.model),J=w(P.provider,P.model,{inputTokens:1e6,outputTokens:1e6});return{model:P.model,provider:P.provider,resolved:x,per1M:J}}function u(){let{json:P,rest:x}=M(process.argv.slice(2)),J=c(x);if(!J)console.error("usage: tlc harness prices lookup <model-id> [provider]"),console.error("   or: node --experimental-strip-types tools/price-lookup.ts <model-id> [provider]  (dev)"),process.exit(1);let G=m(J);if(P)_(G);else console.log(JSON.stringify(G,null,2));if(!G.resolved)process.exit(2)}if(H.main==H.module)u();export{c as parsePriceLookupArgs,m as lookupPrice};
+import{createRequire as T}from"node:module";var k=T(import.meta.url);function M(P){let G=[],J=!1;for(let Q of P){if(Q==="--json"){J=!0;continue}G.push(Q)}return{json:J,rest:G}}function _(P,G=w){G(`${JSON.stringify(P)}
+`)}function w(P){process.stdout.write(P)}import{existsSync as z,readFileSync as f}from"node:fs";import{join as E}from"node:path";import{homedir as S}from"node:os";import{delimiter as o,join as b,resolve as n}from"node:path";function C(){return b(S(),".tlc","harness")}function D(P=process.env){return P.TLC_HOME??C()}function R(P=process.env){return P.TLC_HOME_FROM_ENV==="0"?C():D(P)}var p="litellm",j={"cursor-grok-4.5":"grok-4.5",auto:"auto-cost"};function F(P){if(!P)return{};let{_meta:G,...J}=P;return J}function L(P){return P.trim().toLowerCase().replace(/\[([^\]]+)\]\([^)]*\)/g,"$1").replace(/[[\]]/g,"").replace(/[^a-z0-9.+]+/g,"-").replace(/^-+|-+$/g,"")}function v(P,G){let J=P.trim(),Q=[],X=(q)=>{if(q&&!Q.includes(q))Q.push(q)};if(X(J),X(G[J]),X(L(J)),X(G[L(J)]),J.includes("/"))X(J.slice(J.lastIndexOf("/")+1));let $=J.replace(/-(high|medium|low|max|fast|thinking)$/i,"");if($!==J)X($),X(G[$]),X(L($));return Q}function U(P,G){let J=P[G];if(J)return{key:G,entry:J};for(let[Q,X]of Object.entries(P))if(G.startsWith(Q)||Q.startsWith(G))return{key:Q,entry:X};return}function g(){return E(R(),"model-prices.json")}function y(){return E(R(),"model-prices.local.json")}var A=new Map;function O(P){if(!z(P))return A.delete(P),null;let G;try{G=f(P,"utf8")}catch{return A.delete(P),null}let J=A.get(P);if(J&&J.text===G)return J.value;let Q;try{Q=JSON.parse(G)}catch{return A.delete(P),null}return A.set(P,{text:G,value:Q}),Q}function c(){return O(g())??{}}function H(P,G){return F(P.planes?.[G]??null)}function W(P,G){let J=G.trim();if(!J)return;let Q=c(),X=F(O(y())),$=P?H(Q,P):{},q=H(Q,p),B=v(J,j);for(let Z of B){let x=X[Z];if(x)return{entry:x,key:Z,source:"override"}}for(let Z of B){let x=$[Z];if(x)return{entry:x,key:Z,source:"provider"}}for(let Z of B){let x=q[Z];if(x)return{entry:x,key:Z,source:"litellm"}}let Y=L(J),I=U(X,Y);if(I)return{...I,source:"override"};let K=U($,Y);if(K)return{...K,source:"provider"};let V=U(q,Y);if(V)return{...V,source:"litellm"};return}function N(P,G,J){let Q=J.inputTokens??0,X=J.outputTokens??0,$=J.cacheReadTokens??0,q=J.cacheWriteTokens??0;if(!G||Q<=0&&X<=0&&$<=0&&q<=0)return{costUsd:null,billing:"unknown",pool:"unknown",source:"missing"};let B=W(P,G);if(!B)return{costUsd:null,billing:"unknown",pool:"unknown",source:"missing"};let{entry:Y,key:I,source:K}=B,V=Y.pool??"unknown";if(Y.billing==="included")return{costUsd:null,billing:"included",pool:V,source:K,catalogKey:I};if(typeof Y.promptPer1M!=="number"||typeof Y.completionPer1M!=="number")return{costUsd:null,billing:Y.billing??"unknown",pool:V,source:K,catalogKey:I};let Z=Q/1e6*Y.promptPer1M+X/1e6*Y.completionPer1M;if(typeof Y.cacheReadPer1M==="number"&&$>0)Z+=$/1e6*Y.cacheReadPer1M;if(typeof Y.cacheWritePer1M==="number"&&q>0)Z+=q/1e6*Y.cacheWritePer1M;return{costUsd:Z,billing:"metered",pool:V,source:K,catalogKey:I}}function m(P){let G=P[0];if(!G)return null;return{model:G,provider:P[1]??""}}function u(P){let G=W(P.provider,P.model),J=N(P.provider,P.model,{inputTokens:1e6,outputTokens:1e6});return{model:P.model,provider:P.provider,resolved:G,per1M:J}}function l(){let{json:P,rest:G}=M(process.argv.slice(2)),J=m(G);if(!J)console.error("usage: tlc harness prices lookup <model-id> [provider]"),console.error("   or: node --experimental-strip-types tools/price-lookup.ts <model-id> [provider]  (dev)"),process.exit(1);let Q=u(J);if(P)_(Q);else console.log(JSON.stringify(Q,null,2));if(!Q.resolved)process.exit(2)}if(k.main==k.module)l();export{m as parsePriceLookupArgs,u as lookupPrice};