@wopr-network/platform-core 1.12.2 → 1.13.1

package/src/monetization/adapters/rate-table.test.ts

@@ -11,6 +11,18 @@ describe("RATE_TABLE", () => {
     expect(premiumTTS).toEqual(expect.objectContaining({ capability: "tts", tier: "premium" }));
   });
 
+  it("contains both standard and premium tiers for embeddings", () => {
+    const standard = RATE_TABLE.find((e) => e.capability === "embeddings" && e.tier === "standard");
+    const premium = RATE_TABLE.find((e) => e.capability === "embeddings" && e.tier === "premium");
+
+    expect(standard).toEqual(
+      expect.objectContaining({ capability: "embeddings", tier: "standard", provider: "ollama-embeddings" }),
+    );
+    expect(premium).toEqual(
+      expect.objectContaining({ capability: "embeddings", tier: "premium", provider: "openrouter" }),
+    );
+  });
+
   it("contains both standard and premium tiers for text-generation", () => {
     const standard = RATE_TABLE.find((e) => e.capability === "text-generation" && e.tier === "standard");
     const premium = RATE_TABLE.find((e) => e.capability === "text-generation" && e.tier === "premium");
@@ -52,7 +64,10 @@ describe("RATE_TABLE", () => {
 
     for (const entry of standardEntries) {
       // Self-hosted providers include "self-hosted-" prefix or are known self-hosted names
-      const isSelfHosted = entry.provider.startsWith("self-hosted-") || entry.provider === "chatterbox-tts";
+      const isSelfHosted =
+        entry.provider.startsWith("self-hosted-") ||
+        entry.provider === "chatterbox-tts" ||
+        entry.provider === "ollama-embeddings";
       expect(isSelfHosted).toBe(true);
     }
   });
@@ -140,6 +155,13 @@ describe("getRatesForCapability", () => {
     expect(rates.map((r) => r.tier)).toContain("premium");
   });
 
+  it("returns both standard and premium for embeddings", () => {
+    const rates = getRatesForCapability("embeddings");
+    expect(rates).toHaveLength(2);
+    expect(rates.map((r) => r.tier)).toContain("standard");
+    expect(rates.map((r) => r.tier)).toContain("premium");
+  });
+
   it("returns premium-only for transcription (no standard tier yet)", () => {
     const rates = getRatesForCapability("transcription");
     expect(rates).toHaveLength(1);
@@ -186,6 +208,15 @@ describe("calculateSavings", () => {
     expect(savings).toBeCloseTo(2.01, 2);
   });
 
+  it("calculates savings for embeddings at 1M tokens", () => {
+    const savings = calculateSavings("embeddings", 1_000_000);
+
+    // Standard (ollama-embeddings): $0.006 per 1M tokens
+    // Premium (openrouter): $0.026 per 1M tokens
+    // Savings: $0.020 per 1M tokens
+    expect(savings).toBeCloseTo(0.02, 3);
+  });
+
   it("returns zero when capability has no standard tier", () => {
     // Transcription only has premium (deepgram) — no self-hosted whisper yet
     const savings = calculateSavings("transcription", 1000);

package/src/monetization/adapters/rate-table.ts

@@ -119,8 +119,15 @@ export const RATE_TABLE: RateEntry[] = [
   },
 
   // Embeddings
-  // NOTE: No self-hosted embeddings adapter yet — only premium (openrouter) available.
-  // When self-hosted-embeddings lands, add a standard tier entry here.
+  {
+    capability: "embeddings",
+    tier: "standard",
+    provider: "ollama-embeddings",
+    costPerUnit: 0.000000005, // Amortized GPU cost per token ($0.005 per 1M tokens)
+    billingUnit: "per-token",
+    margin: 1.2, // 20% — dashboard default; runtime uses getMargin()
+    effectivePrice: 0.000000006, // = costPerUnit * margin ($0.006 per 1M tokens)
+  },
   {
     capability: "embeddings",
     tier: "premium",
@@ -157,7 +164,6 @@ export const RATE_TABLE: RateEntry[] = [
 
   // Future self-hosted adapters will add more entries here:
   // - transcription: self-hosted-whisper (standard) — when GPU adapter exists
-  // - embeddings: self-hosted-embeddings (standard) — when GPU adapter exists
   // - image-generation: self-hosted-sdxl (standard) — when GPU adapter exists
 ];