Refactor graph storage limits and update @robosystems/client to 0.2.45

package.json

@@ -44,7 +44,7 @@
   "dependencies": {
     "@aws-sdk/client-sns": "^3.1000.0",
     "@monaco-editor/react": "^4.7.0",
-    "@robosystems/client": "0.2.44",
+    "@robosystems/client": "0.2.45",
     "date-fns": "^4.1.0",
     "flowbite": "^3.1",
     "flowbite-react": "^0.12.5",

src/app/(app)/usage/content.tsx

@@ -6,7 +6,6 @@ import {
   getCreditSummary,
   getGraphLimits,
   getGraphs,
-  getStorageUsage,
   listCreditTransactions,
 } from '@robosystems/client'
 import { Alert, Badge, Button, Card, Progress, Spinner } from 'flowbite-react'
@@ -32,12 +31,6 @@ interface CreditSummary {
   graph_tier: string
 }
 
-interface StorageUsage {
-  current_usage_gb: number
-  max_storage_gb: number
-  storage_cost_per_day?: number
-}
-
 interface GraphLimits {
   subscription_tier: string
   graph_tier: string
@@ -77,7 +70,6 @@ interface CreditTransaction {
 interface UsageData {
   graphInfo: GraphInfo
   creditSummary?: CreditSummary
-  storageUsage?: StorageUsage
   graphLimits?: GraphLimits
   recentTransactions?: CreditTransaction[]
 }
@@ -119,35 +111,17 @@ export function UsageContent() {
 
       const usageData: UsageData = { graphInfo }
 
-      // Determine if this is a repository
-      const isRepo = graphInfo.isRepository
-
-      // Fetch all usage data in parallel (skip storage for repositories)
-      const apiCalls = [
+      // Fetch all usage data in parallel
+      const [limitsRes, creditRes, transactionsRes] = await Promise.allSettled([
         getGraphLimits({ path: { graph_id: graphId } }),
         getCreditSummary({ path: { graph_id: graphId } }),
         listCreditTransactions({
           path: { graph_id: graphId },
           query: { limit: 10 },
         }),
-      ]
-
-      // Only fetch storage for user graphs, not repositories
-      if (!isRepo) {
-        apiCalls.splice(
-          2,
-          0,
-          getStorageUsage({ path: { graph_id: graphId } }) as any
-        )
-      }
+      ])
 
-      const results = await Promise.allSettled(apiCalls)
-
-      const [limitsRes, creditRes, storageRes, transactionsRes] = isRepo
-        ? [results[0], results[1], { status: 'rejected' as const }, results[2]]
-        : [results[0], results[1], results[2], results[3]]
-
-      // Process limits
+      // Process limits (includes instance storage usage)
       if (limitsRes.status === 'fulfilled' && limitsRes.value.data) {
         usageData.graphLimits = limitsRes.value.data as unknown as GraphLimits
       }
@@ -158,12 +132,6 @@ export function UsageContent() {
           .data as unknown as CreditSummary
       }
 
-      // Process storage
-      if (storageRes.status === 'fulfilled' && storageRes.value.data) {
-        usageData.storageUsage = storageRes.value
-          .data as unknown as StorageUsage
-      }
-
       // Process transactions
       if (
         transactionsRes.status === 'fulfilled' &&

src/lib/core/contexts/service-offerings-context.tsx

@@ -17,10 +17,7 @@ export interface ServiceOfferings {
       monthlyPrice: number
       monthlyCredits: number
       features: string[]
-      limits?: {
-        maxNodes?: number
-        maxRelationships?: number
-      }
+      instanceStorageLimitGb?: number
       creditMultiplier: number
     }
   }
@@ -115,10 +112,7 @@ export function ServiceOfferingsProvider({
               features: tier.features || [],
               creditMultiplier:
                 apiData.graph_subscriptions?.tier_multipliers?.[tier.name] || 1,
-              limits: {
-                maxNodes: tier.max_nodes,
-                maxRelationships: tier.max_relationships,
-              },
+              instanceStorageLimitGb: tier.instance_storage_limit_gb,
             }
           })
         }

src/lib/core/lib/__tests__/graph-tiers.test.ts

@@ -30,30 +30,18 @@ describe('graph-tiers', () => {
         display_name: 'Free',
         monthly_price: 0,
         max_subgraphs: 0,
-        limits: {
-          max_nodes: 1000,
-          max_relationships: 1000,
-        },
       },
       {
         tier: 'basic',
         display_name: 'Basic',
         monthly_price: 29,
         max_subgraphs: 1,
-        limits: {
-          max_nodes: 10000,
-          max_relationships: 10000,
-        },
       },
       {
         tier: 'premium',
         display_name: 'Premium',
         monthly_price: 99,
         max_subgraphs: 5,
-        limits: {
-          max_nodes: 100000,
-          max_relationships: 100000,
-        },
       },
     ] as GraphTier[],
   }