@smallwebco/tinypivot-react 1.0.60 → 1.0.62

package/dist/index.js

@@ -57,6 +57,7 @@ function useAIAnalyst(options) {
   }, [storageKey]);
   const [conversation, setConversation] = useState(() => loadFromStorage());
   const [schemas, setSchemas] = useState(/* @__PURE__ */ new Map());
+  const [allSchemas, setAllSchemas] = useState([]);
   const [isLoading, setIsLoading] = useState(false);
   const [error, setError] = useState(null);
   const [lastLoadedData, setLastLoadedData] = useState(null);
@@ -78,6 +79,34 @@ function useAIAnalyst(options) {
   );
   const messages = conversation.messages;
   const hasMessages = conversation.messages.length > 0;
+  const fetchAllSchemas = useCallback(async () => {
+    if (!configRef.current.endpoint)
+      return;
+    try {
+      const response = await fetch(configRef.current.endpoint, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ action: "get-all-schemas" })
+      });
+      if (!response.ok) {
+        throw new Error(`Failed to fetch all schemas: ${response.statusText}`);
+      }
+      const data = await response.json();
+      if (data.error) {
+        throw new Error(data.error);
+      }
+      setAllSchemas(data.schemas);
+      setSchemas((prev) => {
+        const newMap = new Map(prev);
+        for (const schema of data.schemas) {
+          newMap.set(schema.table, schema);
+        }
+        return newMap;
+      });
+    } catch (err) {
+      console.warn("[TinyPivot] Failed to fetch all schemas:", err);
+    }
+  }, []);
   const fetchTables = useCallback(async () => {
     if (!configRef.current.endpoint)
       return;
@@ -102,6 +131,7 @@ function useAIAnalyst(options) {
         // Capitalize
         description: t.description
       })));
+      await fetchAllSchemas();
     } catch (err) {
       console.warn("[TinyPivot] Failed to fetch tables:", err);
       onError?.({
@@ -111,7 +141,7 @@ function useAIAnalyst(options) {
     } finally {
       setIsLoadingTables(false);
     }
-  }, [onError]);
+  }, [onError, fetchAllSchemas]);
   useEffect(() => {
     if (configRef.current.endpoint && (!config.dataSources || config.dataSources.length === 0)) {
       fetchTables();
@@ -143,6 +173,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,9 +261,10 @@ What would you like to know about this data?`
       }
     } else if (configRef.current.endpoint) {
       await fetchSchema(dataSource);
+      await fetchSampleData(dataSource);
     }
-  }, [effectiveDataSources, fetchSchema, onConversationUpdate, onDataLoaded]);
-  const callAIEndpoint = useCallback(async (userInput, currentConversation, currentSchemas, currentDataSources) => {
+  }, [effectiveDataSources, fetchSchema, fetchSampleData, onConversationUpdate, onDataLoaded]);
+  const callAIEndpoint = useCallback(async (userInput, currentConversation, currentSchemas, currentDataSources, currentAllSchemas) => {
     if (!configRef.current.endpoint) {
       throw new Error("No endpoint configured. Set `endpoint` in AI analyst config.");
     }
@@ -207,7 +272,8 @@ What would you like to know about this data?`
     const systemPrompt = buildSystemPrompt(
       currentDataSources,
       currentSchemas,
-      dataSourceId
+      dataSourceId,
+      currentAllSchemas.length > 0 ? currentAllSchemas : void 0
     );
     const apiMessages = getMessagesForAPI(currentConversation);
     const messages2 = [
@@ -466,7 +532,7 @@ What would you like to know about this data?`
         });
         return;
       }
-      const aiResponse = await callAIEndpoint(content, currentConv, schemas, effectiveDataSources);
+      const aiResponse = await callAIEndpoint(content, currentConv, schemas, effectiveDataSources, allSchemas);
       const sqlQuery = extractSQLFromResponse(aiResponse);
       if (sqlQuery) {
         const validation = validateSQLSafety(sqlQuery);