filters

Author: alrocar

dashboard/ai-analytics/src/services/tinybird.ts

@@ -12,12 +12,23 @@ export interface TinybirdParams {
 
 export async function fetchLLMUsage(params: TinybirdParams = {}) {
   const searchParams = new URLSearchParams();
+  
+  // Handle column_name separately as it's used for grouping
+  if (params.column_name) {
+    searchParams.set('column_name', params.column_name);
+  }
+
+  // Handle all other filter parameters
+  const filterParams = ['model', 'provider', 'organization', 'project', 'environment', 'user'];
+  filterParams.forEach(param => {
+    if (params[param as keyof TinybirdParams]) {
+      searchParams.set(param, params[param as keyof TinybirdParams]!);
+    }
+  });
+
+  // Handle date range
   if (params.start_date) searchParams.set('start_date', params.start_date);
   if (params.end_date) searchParams.set('end_date', params.end_date);
-  if (params.organization) searchParams.set('organization', params.organization);
-  if (params.project) searchParams.set('project', params.project);
-  // Default to 'model' if column_name is not provided
-  searchParams.set('column_name', params.column_name || 'model');
 
   const response = await fetch(
     `${TINYBIRD_API_URL}/v0/pipes/llm_usage.json?${searchParams.toString()}`,
@@ -28,6 +39,10 @@ export async function fetchLLMUsage(params: TinybirdParams = {}) {
     }
   );
 
+  if (!response.ok) {
+    throw new Error('Network response was not ok');
+  }
+
   return response.json();
 }