@smallwebco/tinypivot-react 1.0.59 → 1.0.61

package/dist/index.js

@@ -143,6 +143,40 @@ function useAIAnalyst(options) {
       console.warn("Failed to fetch schema:", err);
     }
   }, []);
+  const fetchSampleData = useCallback(async (dataSource) => {
+    if (!configRef.current.endpoint)
+      return;
+    try {
+      const sql = `SELECT * FROM ${dataSource.table} LIMIT 100`;
+      const response = await fetch(configRef.current.endpoint, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({
+          action: "query",
+          sql,
+          table: dataSource.table
+        })
+      });
+      if (!response.ok) {
+        throw new Error(`Failed to fetch sample data: ${response.statusText}`);
+      }
+      const result = await response.json();
+      if (result.error) {
+        throw new Error(result.error);
+      }
+      if (result.data && result.data.length > 0) {
+        setLastLoadedData(result.data);
+        onDataLoaded?.({
+          data: result.data,
+          query: sql,
+          dataSourceId: dataSource.id,
+          rowCount: result.data.length
+        });
+      }
+    } catch (err) {
+      console.warn("Failed to fetch sample data:", err);
+    }
+  }, [onDataLoaded]);
   const selectDataSource = useCallback(async (dataSourceId) => {
     const dataSource = effectiveDataSources.find((ds) => ds.id === dataSourceId);
     if (!dataSource) {
@@ -197,8 +231,9 @@ What would you like to know about this data?`
       }
     } else if (configRef.current.endpoint) {
       await fetchSchema(dataSource);
+      await fetchSampleData(dataSource);
     }
-  }, [effectiveDataSources, fetchSchema, onConversationUpdate, onDataLoaded]);
+  }, [effectiveDataSources, fetchSchema, fetchSampleData, onConversationUpdate, onDataLoaded]);
   const callAIEndpoint = useCallback(async (userInput, currentConversation, currentSchemas, currentDataSources) => {
     if (!configRef.current.endpoint) {
       throw new Error("No endpoint configured. Set `endpoint` in AI analyst config.");