@prismiq/react 0.2.0 → 0.2.1

package/dist/{chunk-GELI7MDZ.js → chunk-VEFYFB5H.js}

@@ -2859,9 +2859,10 @@ var PrismiqClient = class {
    * @param history - Previous conversation messages.
    * @param currentSql - Current SQL in the editor (for context).
    * @param signal - Optional AbortSignal for cancellation.
+   * @param widgetContext - Optional widget context for targeted SQL generation.
    * @yields StreamChunk objects as the response is generated.
    */
-  async *streamChat(message, history, currentSql, signal) {
+  async *streamChat(message, history, currentSql, signal, widgetContext) {
     const url = `${this.endpoint}/llm/chat`;
     const headers = {
       "Content-Type": "application/json",
@@ -2874,14 +2875,18 @@ var PrismiqClient = class {
       const token = await this.getToken();
       headers["Authorization"] = `Bearer ${token}`;
     }
+    const body = {
+      message,
+      history,
+      current_sql: currentSql
+    };
+    if (widgetContext) {
+      body.widget_context = widgetContext;
+    }
     const response = await fetch(url, {
       method: "POST",
       headers,
-      body: JSON.stringify({
-        message,
-        history,
-        current_sql: currentSql
-      }),
+      body: JSON.stringify(body),
       signal
     });
     if (!response.ok) {
@@ -4020,12 +4025,13 @@ function useLLMChat() {
   const [streamingContent, setStreamingContent] = useState("");
   const [suggestedSql, setSuggestedSql] = useState(null);
   const [error, setError] = useState(null);
+  const [statusMessage, setStatusMessage] = useState(null);
   const abortRef = useRef(null);
   const isStreamingRef = useRef(false);
   const messagesRef = useRef([]);
   messagesRef.current = messages;
   const sendMessage = useCallback(
-    async (message, currentSql) => {
+    async (message, currentSql, widgetContext) => {
       if (!client || isStreamingRef.current) return;
       abortRef.current?.abort();
       const controller = new AbortController();
@@ -4037,6 +4043,7 @@ function useLLMChat() {
       setStreamingContent("");
       setSuggestedSql(null);
       setError(null);
+      setStatusMessage(null);
       let accumulatedText = "";
       let lastSql = null;
       try {
@@ -4048,22 +4055,28 @@ function useLLMChat() {
           message,
           history,
           currentSql,
-          controller.signal
+          controller.signal,
+          widgetContext
         )) {
           if (controller.signal.aborted) break;
           switch (chunk.type) {
             case "text":
               accumulatedText += chunk.content ?? "";
               setStreamingContent(accumulatedText);
+              setStatusMessage(null);
               break;
             case "sql":
               lastSql = chunk.content ?? null;
               setSuggestedSql(lastSql);
               break;
+            case "status":
+              setStatusMessage(chunk.content ?? null);
+              break;
             case "error":
               setError(chunk.content ?? "Unknown error");
               break;
             case "done":
+              setStatusMessage(null);
               break;
           }
         }
@@ -4074,6 +4087,7 @@ function useLLMChat() {
       } finally {
         isStreamingRef.current = false;
         setIsStreaming(false);
+        setStatusMessage(null);
         if (accumulatedText) {
           const assistantMsg = {
             role: "assistant",
@@ -4098,6 +4112,7 @@ function useLLMChat() {
     setSuggestedSql(null);
     setError(null);
     setIsStreaming(false);
+    setStatusMessage(null);
   }, []);
   return {
     messages,
@@ -4106,7 +4121,8 @@ function useLLMChat() {
     suggestedSql,
     sendMessage,
     clearHistory,
-    error
+    error,
+    statusMessage
   };
 }
 var nodeStyles = {
@@ -9457,7 +9473,7 @@ function ChatBubble({ message, onApplySql }) {
     return /* @__PURE__ */ jsx("span", { children: part.value }, i);
   }) });
 }
-function ChatPanel({ currentSql, onApplySql }) {
+function ChatPanel({ currentSql, onApplySql, widgetContext }) {
   const { theme } = useTheme();
   const {
     messages,
@@ -9466,7 +9482,8 @@ function ChatPanel({ currentSql, onApplySql }) {
     suggestedSql,
     sendMessage,
     clearHistory,
-    error
+    error,
+    statusMessage
   } = useLLMChat();
   const [input, setInput] = useState("");
   const messagesEndRef = useRef(null);
@@ -9477,8 +9494,8 @@ function ChatPanel({ currentSql, onApplySql }) {
     const trimmed = input.trim();
     if (!trimmed) return;
     setInput("");
-    void sendMessage(trimmed, currentSql);
-  }, [input, currentSql, sendMessage]);
+    void sendMessage(trimmed, currentSql, widgetContext);
+  }, [input, currentSql, widgetContext, sendMessage]);
   const handleKeyDown = useCallback(
     (e) => {
       if (e.key === "Enter" && !e.shiftKey) {
@@ -9599,6 +9616,24 @@ function ChatPanel({ currentSql, onApplySql }) {
         "\u258D"
       ] }),
       isStreaming && !streamingContent && /* @__PURE__ */ jsx("div", { style: streamingStyle, "data-testid": "chat-streaming", children: "Thinking..." }),
+      isStreaming && statusMessage && /* @__PURE__ */ jsxs(
+        "div",
+        {
+          style: {
+            display: "flex",
+            alignItems: "center",
+            gap: theme.spacing.xs,
+            fontSize: theme.fontSizes.xs,
+            color: theme.colors.textMuted,
+            padding: `${theme.spacing.xs} 0`
+          },
+          "data-testid": "chat-status",
+          children: [
+            /* @__PURE__ */ jsx(Icon, { name: "sync", size: 12 }),
+            /* @__PURE__ */ jsx("span", { children: statusMessage })
+          ]
+        }
+      ),
       /* @__PURE__ */ jsx("div", { ref: messagesEndRef })
     ] }),
     error && /* @__PURE__ */ jsx("div", { style: errorStyle, "data-testid": "chat-error", children: error }),
@@ -9633,5 +9668,5 @@ function ChatPanel({ currentSql, onApplySql }) {
 }
 
 export { AggregationPicker, AnalyticsProvider, AutoSaveIndicator, Badge, Button, CalculatedFieldBuilder, ChatBubble, ChatPanel, Checkbox, CollapsibleSection, ColorPaletteSelector, ColumnNode, ColumnSelector, CrossFilterProvider, CustomSQLEditor, Dialog, DialogFooter, DialogHeader, Dropdown, DropdownItem, DropdownSeparator, EmptyDashboard, EmptyState, ErrorBoundary, ErrorFallback, ExpressionEditor, FilterBuilder, FilterRow, FilterValueInput, Icon, Input, JoinBuilder, JoinRow, NoData, NoResults, Pagination, PrismiqClient, PrismiqError, QueryBuilder, QueryBuilderToolbar, QueryPreview, ResultsTable, SavedQueryPicker, SchemaExplorer, Select, SelectedColumn, Skeleton, SkeletonChart, SkeletonMetricCard, SkeletonTable, SkeletonText, SortBuilder, SortRow, TableCell, TableHeader, TableNode, TableRow, TableSelector, TimeSeriesConfig, Tooltip, WidgetErrorBoundary, useAnalytics, useAnalyticsCallbacks, useChartData, useCrossFilterOptional, useCustomSQL, useDashboard, useDashboardMutations, useDashboardPinStatus, useDashboards, useDebouncedLayoutSave, useLLMChat, useLLMStatus, usePinMutations, usePinnedDashboards, useQuery, useSavedQueries, useSchema };
-//# sourceMappingURL=chunk-GELI7MDZ.js.map
-//# sourceMappingURL=chunk-GELI7MDZ.js.map
+//# sourceMappingURL=chunk-VEFYFB5H.js.map
+//# sourceMappingURL=chunk-VEFYFB5H.js.map