@stigg/react-sdk 7.6.0 → 7.7.1-beta.0

package/src/stories/CreditBalance.stories.tsx

@@ -110,8 +110,10 @@ export const MockedImplementation: Story = {
         currentBalance: args.currentBalance || 1250,
         currencyId: 'cred-type-ai-tokens',
         displayName: 'AI Tokens',
-        symbol: null,
-        units: { singular: 'token', plural: 'tokens' },
+        symbol: undefined,
+        units: undefined,
+        totalConsumed: 0,
+        totalGranted: 1250,
       },
       isLoading: false,
       error: null,
@@ -130,8 +132,10 @@ export const CustomThemedBalance: Story = {
         currentBalance: 500.75,
         currencyId: 'usd',
         displayName: 'US Dollars',
-        symbol: null,
-        units: { singular: 'dollar', plural: 'dollars' },
+        symbol: undefined,
+        units: undefined,
+        totalConsumed: 0,
+        totalGranted: 500.75,
       },
       isLoading: false,
       error: null,
@@ -188,8 +192,10 @@ export const CustomBalance: Story = {
         currentBalance: 2500,
         currencyId: 'tokens',
         displayName: 'AI Tokens',
-        symbol: null,
-        units: { singular: 'token', plural: 'tokens' },
+        symbol: undefined,
+        units: undefined,
+        totalConsumed: 0,
+        totalGranted: 2500,
       },
       isLoading: false,
       error: null,
@@ -208,7 +214,9 @@ export const CustomTitleOverride: Story = {
         currencyId: 'cred-type-ai-tokens',
         displayName: 'AI Tokens',
         symbol: '✨',
-        units: { singular: 'token', plural: 'tokens' },
+        units: undefined,
+        totalConsumed: 0,
+        totalGranted: 1800,
       },
       isLoading: false,
       error: null,
@@ -229,7 +237,9 @@ export const ClippedValue: Story = {
         currencyId: 'cred-type-ai-tokens',
         displayName: 'AI Tokens',
         symbol: '✨',
-        units: { singular: 'token', plural: 'tokens' },
+        units: undefined,
+        totalConsumed: 0,
+        totalGranted: 1234.56789123456,
       },
       isLoading: false,
       error: null,

package/src/stories/CreditUsageChart.stories.tsx

@@ -92,10 +92,11 @@ export const MockedImplementation: Story = {
     const mockDataCache = new Map<CreditUsageTimeRange, any>();
 
     // Create a function that generates mock data based on the current timeRange
-    const generateMockData = (timeRange: CreditUsageTimeRange) => {
+    const generateMockData = (timeRange: CreditUsageTimeRange): ReturnType<typeof useCreditUsage> => {
       // Check if we already have cached data for this time range
-      if (mockDataCache.has(timeRange)) {
-        return mockDataCache.get(timeRange);
+      const cached = mockDataCache.get(timeRange) as ReturnType<typeof useCreditUsage> | undefined;
+      if (cached) {
+        return cached;
       }
 
       // Generate data points based on the selected time range