@polka-codes/core 0.9.100 → 0.9.102

package/dist/_tsup-dts-rollup.d.ts

@@ -330,6 +330,7 @@ declare const configSchema: z.ZodOptional<z.ZodNullable<z.ZodObject<{
         }>>>;
         path: z.ZodDefault<z.ZodOptional<z.ZodString>>;
     }, z.core.$strict>>;
+    loadRules: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodBoolean>>;
 }, z.core.$strict>>>;
 export { configSchema }
 export { configSchema as configSchema_alias_1 }
@@ -363,6 +364,12 @@ export { convertJsonSchemaToZod }
 export { convertJsonSchemaToZod as convertJsonSchemaToZod_alias_1 }
 export { convertJsonSchemaToZod as convertJsonSchemaToZod_alias_2 }
 
+/**
+ * Convert Portkey pricing (cents per token) to ModelInfo (USD per 1M tokens)
+ * Formula: cents_per_token * 10 = usd_per_1M_tokens
+ */
+export declare function convertPortkeyToModelInfo(portkey: PortkeyPricingResponse): ModelInfo;
+
 declare function createContext<TTools extends ToolRegistry>(tools: WorkflowTools<TTools>, stepFn?: StepFn, logger?: Logger): WorkflowContext<TTools>;
 export { createContext }
 export { createContext as createContext_alias_1 }
@@ -889,6 +896,8 @@ export { ExitReason }
 export { ExitReason as ExitReason_alias_1 }
 export { ExitReason as ExitReason_alias_2 }
 
+export declare function fetchPricing(provider: string, model: string): Promise<PortkeyPricingResponse | null>;
+
 /**
  * File statistics
  */
@@ -1654,6 +1663,8 @@ declare type ModelInfo = {
 };
 export { ModelInfo }
 export { ModelInfo as ModelInfo_alias_1 }
+export { ModelInfo as ModelInfo_alias_2 }
+export { ModelInfo as ModelInfo_alias_3 }
 
 /**
  * Node.js file system provider implementation
@@ -1695,12 +1706,51 @@ export { ParseOutputResult }
 export { ParseOutputResult as ParseOutputResult_alias_1 }
 export { ParseOutputResult as ParseOutputResult_alias_2 }
 
+declare interface PortkeyPricingConfig {
+    provider: string;
+    model: string;
+    pricing: PortkeyPricingResponse;
+}
+export { PortkeyPricingConfig }
+export { PortkeyPricingConfig as PortkeyPricingConfig_alias_1 }
+export { PortkeyPricingConfig as PortkeyPricingConfig_alias_2 }
+
+declare interface PortkeyPricingResponse {
+    request_token?: {
+        price: number;
+    };
+    response_token?: {
+        price: number;
+    };
+    cache_write_input_token?: {
+        price: number;
+    };
+    cache_read_input_token?: {
+        price: number;
+    };
+    additional_units?: Record<string, {
+        price: number;
+    }>;
+}
+export { PortkeyPricingResponse }
+export { PortkeyPricingResponse as PortkeyPricingResponse_alias_1 }
+export { PortkeyPricingResponse as PortkeyPricingResponse_alias_2 }
+
 /**
  * Simplify boolean string preprocessing
  * Converts 'true'/'false' strings to actual booleans
  */
 export declare function preprocessBoolean(val: unknown): unknown;
 
+declare class PricingService {
+    #private;
+    constructor(fallbackPrices?: Record<string, Record<string, Partial<ModelInfo>>>);
+    getPricing(provider: string, model: string): Promise<ModelInfo>;
+}
+export { PricingService }
+export { PricingService as PricingService_alias_1 }
+export { PricingService as PricingService_alias_2 }
+
 declare type ProviderConfig = z.infer<typeof providerConfigSchema>;
 export { ProviderConfig }
 export { ProviderConfig as ProviderConfig_alias_1 }
@@ -2637,6 +2687,7 @@ declare class UsageMeter {
     constructor(modelInfos?: Record<string, Record<string, Partial<ModelInfo>>>, opts?: {
         maxMessages?: number;
         maxCost?: number;
+        pricingService?: PricingService;
     });
     addUsage(llm: LanguageModelV2, resp: {
         usage: LanguageModelV2Usage;
@@ -2646,7 +2697,7 @@ declare class UsageMeter {
         providerMetadata?: any;
     }, options?: {
         modelInfo?: ModelInfo;
-    }): void;
+    }): Promise<void>;
     /** Override the running totals (e.g., restore from saved state). */
     setUsage(newUsage: Partial<Totals>, options?: {
         clearMetadata?: boolean;
@@ -2687,11 +2738,13 @@ declare class UsageMeter {
     clearProviderMetadata(): void;
     /** Merge another UsageMeter's totals into this one. */
     merge(other: UsageMeter): void;
+    /** Wait for all pending usage updates to complete. */
+    waitForPending(): Promise<void>;
     getUsageText(): string;
     onFinishHandler(llm: LanguageModelV2): (evt: {
         totalUsage: LanguageModelV2Usage;
         providerMetadata: any;
-    }) => void;
+    }) => Promise<void>;
 }
 export { UsageMeter }
 export { UsageMeter as UsageMeter_alias_1 }

package/dist/index.d.ts

@@ -40,6 +40,10 @@ export { MemoryOperation } from './_tsup-dts-rollup.js';
 export { DatabaseStats } from './_tsup-dts-rollup.js';
 export { IMemoryStore } from './_tsup-dts-rollup.js';
 export { MemoryStoreConfig } from './_tsup-dts-rollup.js';
+export { PricingService } from './_tsup-dts-rollup.js';
+export { ModelInfo_alias_1 as ModelInfo } from './_tsup-dts-rollup.js';
+export { PortkeyPricingConfig } from './_tsup-dts-rollup.js';
+export { PortkeyPricingResponse } from './_tsup-dts-rollup.js';
 export { SKILL_LIMITS } from './_tsup-dts-rollup.js';
 export { IGNORED_DIRECTORIES } from './_tsup-dts-rollup.js';
 export { SUSPICIOUS_PATTERNS } from './_tsup-dts-rollup.js';
@@ -117,7 +121,6 @@ export { UpdateTodoItemInputSchema } from './_tsup-dts-rollup.js';
 export { UpdateTodoItemInput } from './_tsup-dts-rollup.js';
 export { UpdateTodoItemOutputSchema } from './_tsup-dts-rollup.js';
 export { UpdateTodoItemOutput } from './_tsup-dts-rollup.js';
-export { ModelInfo_alias_1 as ModelInfo } from './_tsup-dts-rollup.js';
 export { UsageMeter_alias_1 as UsageMeter } from './_tsup-dts-rollup.js';
 export { deepMerge } from './_tsup-dts-rollup.js';
 export { AgentWorkflowInput } from './_tsup-dts-rollup.js';

package/dist/index.js

@@ -329,7 +329,8 @@ var configSchema = z3.object({
   rules: z3.array(ruleSchema).optional().or(z3.string()).optional(),
   excludeFiles: z3.array(z3.string()).optional(),
   agent: agentSchema,
-  memory: memoryConfigSchema
+  memory: memoryConfigSchema,
+  loadRules: z3.record(z3.string(), z3.boolean()).optional()
 }).strict().nullish();
 
 // src/errors/base.ts
@@ -402,6 +403,179 @@ var NodeFileSystemProvider = class {
   }
 };
 
+// src/pricing/pricing-service.ts
+import { randomUUID } from "crypto";
+import { mkdir, readFile as readFile2, rename, writeFile } from "fs/promises";
+import { homedir } from "os";
+import { dirname, join as join2 } from "path";
+
+// src/pricing/converter.ts
+function convertPortkeyToModelInfo(portkey) {
+  return {
+    inputPrice: (portkey.request_token?.price ?? 0) * 10,
+    outputPrice: (portkey.response_token?.price ?? 0) * 10,
+    cacheWritesPrice: (portkey.cache_write_input_token?.price ?? 0) * 10,
+    cacheReadsPrice: (portkey.cache_read_input_token?.price ?? 0) * 10
+  };
+}
+
+// src/pricing/portkey-client.ts
+var PORTKEY_BASE_URL = "https://api.portkey.ai/model-configs/pricing";
+var TIMEOUT_MS = 5e3;
+var MAX_RETRIES = 2;
+async function fetchWithTimeout(url, timeoutMs) {
+  const controller = new AbortController();
+  const timeout = setTimeout(() => controller.abort(), timeoutMs);
+  try {
+    const response = await fetch(url, { signal: controller.signal });
+    return response;
+  } finally {
+    clearTimeout(timeout);
+  }
+}
+async function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function shouldRetry(response, error) {
+  if (error) return true;
+  if (response.status >= 500) return true;
+  return false;
+}
+async function fetchPricing(provider, model) {
+  const url = `${PORTKEY_BASE_URL}/${provider}/${model}`;
+  for (let attempt = 0; attempt <= MAX_RETRIES; attempt++) {
+    try {
+      const response = await fetchWithTimeout(url, TIMEOUT_MS);
+      if (!response.ok) {
+        if (!shouldRetry(response, null)) {
+          return null;
+        }
+        if (attempt < MAX_RETRIES) {
+          await sleep(2 ** attempt * 1e3);
+          continue;
+        }
+        return null;
+      }
+      const data = await response.json();
+      return data;
+    } catch (error) {
+      if (attempt < MAX_RETRIES && shouldRetry(new Response(null, { status: 500 }), error)) {
+        await sleep(2 ** attempt * 1e3);
+        continue;
+      }
+      return null;
+    }
+  }
+  return null;
+}
+
+// src/pricing/pricing-service.ts
+var CACHE_TTL_MS = 7 * 24 * 60 * 60 * 1e3;
+var PricingService = class {
+  #fallbackPrices;
+  #cacheFile;
+  #cache = null;
+  #loadPromise = null;
+  constructor(fallbackPrices = {}) {
+    const normalized = {};
+    for (const [provider, providerInfo] of Object.entries(fallbackPrices)) {
+      const normalizedProvider = provider.split("-")[0];
+      normalized[normalizedProvider] = {};
+      for (const [model, modelInfo] of Object.entries(providerInfo)) {
+        const normalizedModel = model.replace(/[.-]/g, "");
+        normalized[normalizedProvider][normalizedModel] = {
+          inputPrice: modelInfo.inputPrice ?? 0,
+          outputPrice: modelInfo.outputPrice ?? 0,
+          cacheWritesPrice: modelInfo.cacheWritesPrice ?? 0,
+          cacheReadsPrice: modelInfo.cacheReadsPrice ?? 0
+        };
+      }
+    }
+    this.#fallbackPrices = normalized;
+    this.#cacheFile = join2(homedir(), ".config", "polkacodes", "pricing-cache.json");
+  }
+  async #load() {
+    if (this.#loadPromise) {
+      return this.#loadPromise;
+    }
+    this.#loadPromise = (async () => {
+      this.#cache = /* @__PURE__ */ new Map();
+      try {
+        const content = await readFile2(this.#cacheFile, "utf-8");
+        const data = JSON.parse(content);
+        for (const [key, value] of Object.entries(data)) {
+          this.#cache.set(key, value);
+        }
+      } catch {
+      }
+    })();
+    return this.#loadPromise;
+  }
+  async #get(provider, model) {
+    await this.#load();
+    const key = `${provider}:${model}`;
+    const entry = this.#cache?.get(key);
+    if (!entry) {
+      return null;
+    }
+    if (Date.now() - entry.timestamp > CACHE_TTL_MS) {
+      this.#cache?.delete(key);
+      return null;
+    }
+    return entry.pricing;
+  }
+  async #set(provider, model, pricing) {
+    await this.#load();
+    const key = `${provider}:${model}`;
+    this.#cache?.set(key, {
+      pricing,
+      timestamp: Date.now()
+    });
+    await this.#save();
+  }
+  async #save() {
+    if (!this.#cache) {
+      return;
+    }
+    try {
+      const dir = dirname(this.#cacheFile);
+      await mkdir(dir, { recursive: true });
+      const data = {};
+      for (const [key, value] of this.#cache.entries()) {
+        data[key] = value;
+      }
+      const tempFile = `${this.#cacheFile}.${randomUUID()}.tmp`;
+      await writeFile(tempFile, JSON.stringify(data, null, 2), "utf-8");
+      await rename(tempFile, this.#cacheFile);
+    } catch {
+    }
+  }
+  async getPricing(provider, model) {
+    const normalizedProvider = provider.split("-")[0];
+    const normalizedModel = model.replace(/[.-]/g, "");
+    const cached = await this.#get(normalizedProvider, normalizedModel);
+    if (cached) {
+      return cached;
+    }
+    const fallbackPrice = this.#fallbackPrices[normalizedProvider]?.[normalizedModel];
+    if (fallbackPrice) {
+      return fallbackPrice;
+    }
+    const portkeyPricing = await fetchPricing(normalizedProvider, model);
+    if (portkeyPricing) {
+      const modelInfo = convertPortkeyToModelInfo(portkeyPricing);
+      await this.#set(normalizedProvider, normalizedModel, modelInfo);
+      return modelInfo;
+    }
+    return {
+      inputPrice: 0,
+      outputPrice: 0,
+      cacheWritesPrice: 0,
+      cacheReadsPrice: 0
+    };
+  }
+};
+
 // src/skills/constants.ts
 var SKILL_LIMITS = {
   MAX_FILE_SIZE: 1024 * 1024,
@@ -455,7 +629,7 @@ var SOURCE_ICONS = {
 };
 
 // src/skills/discovery.ts
-import { homedir } from "os";
+import { homedir as homedir2 } from "os";
 import { parse } from "yaml";
 import { ZodError } from "zod";
 
@@ -545,7 +719,7 @@ var SkillDiscoveryService = class {
   pluginSkillsDirs;
   constructor(options) {
     this.fs = options.fs ?? new NodeFileSystemProvider();
-    this.personalSkillsDir = options.personalSkillsDir ?? this.fs.join(homedir(), ".claude", "skills");
+    this.personalSkillsDir = options.personalSkillsDir ?? this.fs.join(homedir2(), ".claude", "skills");
     this.projectSkillsDir = this.fs.join(options.cwd, ".claude", "skills");
     this.pluginSkillsDirs = options.pluginSkillsDirs ?? [];
   }
@@ -787,19 +961,19 @@ var SkillDiscoveryService = class {
 };
 
 // src/skills/validation.ts
-import { join as join2, normalize as normalize2 } from "path";
+import { join as join3, normalize as normalize2 } from "path";
 function validateSkillSecurity(skill) {
   const { MAX_FILE_SIZE, MAX_SKILL_SIZE } = SKILL_LIMITS;
   let totalSize = 0;
   const contentSize = Buffer.byteLength(skill.content, "utf8");
   if (contentSize > MAX_FILE_SIZE) {
-    throw new SkillValidationError(`SKILL.md content exceeds size limit (${contentSize} > ${MAX_FILE_SIZE})`, join2(skill.path, "SKILL.md"));
+    throw new SkillValidationError(`SKILL.md content exceeds size limit (${contentSize} > ${MAX_FILE_SIZE})`, join3(skill.path, "SKILL.md"));
   }
   totalSize += contentSize;
   for (const [filename, content] of skill.files) {
     const fileSize = Buffer.byteLength(content, "utf8");
     if (fileSize > MAX_FILE_SIZE) {
-      throw new SkillValidationError(`File ${filename} exceeds size limit (${fileSize} > ${MAX_FILE_SIZE})`, join2(skill.path, filename));
+      throw new SkillValidationError(`File ${filename} exceeds size limit (${fileSize} > ${MAX_FILE_SIZE})`, join3(skill.path, filename));
     }
     totalSize += fileSize;
   }
@@ -808,7 +982,7 @@ function validateSkillSecurity(skill) {
   }
   validateContentSecurity(skill.content, skill.path);
   for (const [filename, content] of skill.files) {
-    validateContentSecurity(content, join2(skill.path, filename));
+    validateContentSecurity(content, join3(skill.path, filename));
   }
 }
 function validateContentSecurity(content, path) {
@@ -2270,9 +2444,11 @@ var writeToFile_default = {
 var UsageMeter = class {
   #totals = { input: 0, output: 0, cachedRead: 0, cost: 0, messageCount: 0 };
   #providerMetadataEntries = [];
+  #pendingUpdates = /* @__PURE__ */ new Set();
   #modelInfos;
   #maxMessages;
   #maxCost;
+  #pricingService;
   constructor(modelInfos = {}, opts = {}) {
     const infos = {};
     for (const [provider, providerInfo] of Object.entries(modelInfos)) {
@@ -2288,6 +2464,7 @@ var UsageMeter = class {
     this.#modelInfos = infos;
     this.#maxMessages = opts.maxMessages ?? 1e3;
     this.#maxCost = opts.maxCost ?? 100;
+    this.#pricingService = opts.pricingService;
   }
   #calculateUsage(usage, providerMetadata, modelInfo) {
     const providerMetadataKey = Object.keys(providerMetadata ?? {})[0];
@@ -2338,30 +2515,52 @@ var UsageMeter = class {
     }
   }
   addUsage(llm, resp, options = {}) {
-    const modelInfo = options.modelInfo ?? // make google.vertex.chat to google
-    // and anthropic.messages to anthropic
-    this.#modelInfos[`${llm.provider.split(".")[0]}:${llm.modelId.replace(/[.-]/g, "")}`] ?? {
-      inputPrice: 0,
-      outputPrice: 0,
-      cacheWritesPrice: 0,
-      cacheReadsPrice: 0
-    };
-    const usage = "totalUsage" in resp ? resp.totalUsage : resp.usage;
-    const result = this.#calculateUsage(usage, resp.providerMetadata, modelInfo);
-    this.#totals.input += result.input || 0;
-    this.#totals.output += result.output || 0;
-    this.#totals.cachedRead += result.cachedRead || 0;
-    this.#totals.cost += result.cost || 0;
-    this.#totals.messageCount += 1;
-    if (resp.providerMetadata && Object.keys(resp.providerMetadata).length > 0) {
-      const providerKey = Object.keys(resp.providerMetadata)[0];
-      this.#providerMetadataEntries.push({
-        provider: providerKey || llm.provider,
-        model: llm.modelId,
-        metadata: resp.providerMetadata[providerKey] || resp.providerMetadata,
-        timestamp: Date.now()
-      });
-    }
+    const provider = llm.provider.split(".")[0];
+    const normalizedModel = llm.modelId.replace(/[.-]/g, "");
+    const key = `${provider}:${normalizedModel}`;
+    let modelInfo = options.modelInfo ?? this.#modelInfos[key];
+    const updatePromise = (async () => {
+      try {
+        if (!modelInfo && this.#pricingService) {
+          modelInfo = await this.#pricingService.getPricing(provider, llm.modelId);
+          this.#modelInfos[key] = modelInfo;
+        }
+      } catch {
+        modelInfo = {
+          inputPrice: 0,
+          outputPrice: 0,
+          cacheWritesPrice: 0,
+          cacheReadsPrice: 0
+        };
+      }
+      modelInfo = modelInfo ?? {
+        inputPrice: 0,
+        outputPrice: 0,
+        cacheWritesPrice: 0,
+        cacheReadsPrice: 0
+      };
+      const usage = "totalUsage" in resp ? resp.totalUsage : resp.usage;
+      const result = this.#calculateUsage(usage, resp.providerMetadata, modelInfo);
+      this.#totals.input += result.input || 0;
+      this.#totals.output += result.output || 0;
+      this.#totals.cachedRead += result.cachedRead || 0;
+      this.#totals.cost += result.cost || 0;
+      this.#totals.messageCount += 1;
+      if (resp.providerMetadata && Object.keys(resp.providerMetadata).length > 0) {
+        const providerKey = Object.keys(resp.providerMetadata)[0];
+        this.#providerMetadataEntries.push({
+          provider: providerKey || llm.provider,
+          model: llm.modelId,
+          metadata: resp.providerMetadata[providerKey] || resp.providerMetadata,
+          timestamp: Date.now()
+        });
+      }
+    })();
+    this.#pendingUpdates.add(updatePromise);
+    updatePromise.finally(() => {
+      this.#pendingUpdates.delete(updatePromise);
+    });
+    return updatePromise;
   }
   /** Override the running totals (e.g., restore from saved state). */
   setUsage(newUsage, options = {}) {
@@ -2452,13 +2651,18 @@ var UsageMeter = class {
     this.#totals.messageCount += otherUsage.messageCount;
     this.#providerMetadataEntries.push(...other.providerMetadata);
   }
+  /** Wait for all pending usage updates to complete. */
+  async waitForPending() {
+    const pending = Array.from(this.#pendingUpdates);
+    await Promise.allSettled(pending);
+  }
   getUsageText() {
     const u = this.usage;
     return `Usage - messages: ${u.messageCount}, input: ${u.input}, cached: ${u.cachedRead}, output: ${u.output}, cost: $${u.cost.toFixed(4)}`;
   }
   onFinishHandler(llm) {
-    return (evt) => {
-      this.addUsage(llm, evt);
+    return async (evt) => {
+      await this.addUsage(llm, evt);
     };
   }
 };
@@ -3775,6 +3979,7 @@ export {
   LoadSkillOutputSchema,
   MockProvider,
   NodeFileSystemProvider,
+  PricingService,
   ReadSkillFileInputSchema,
   ReadSkillFileOutputSchema,
   SKILL_ERROR_MESSAGES,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.9.100",
+  "version": "0.9.102",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",