@smallwebco/tinypivot-react 1.0.61 → 1.0.63

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();
@@ -234,7 +264,7 @@ What would you like to know about this data?`
       await fetchSampleData(dataSource);
     }
   }, [effectiveDataSources, fetchSchema, fetchSampleData, onConversationUpdate, onDataLoaded]);
-  const callAIEndpoint = useCallback(async (userInput, currentConversation, currentSchemas, currentDataSources) => {
+  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.");
     }
@@ -242,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 = [
@@ -501,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);