@prismiq/react 0.2.0 → 0.2.1

package/dist/{chunk-PG7QBH3G.cjs → chunk-NXXKG4GN.cjs}

@@ -2861,9 +2861,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",
@@ -2876,14 +2877,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) {
@@ -4022,12 +4027,13 @@ function useLLMChat() {
   const [streamingContent, setStreamingContent] = react.useState("");
   const [suggestedSql, setSuggestedSql] = react.useState(null);
   const [error, setError] = react.useState(null);
+  const [statusMessage, setStatusMessage] = react.useState(null);
   const abortRef = react.useRef(null);
   const isStreamingRef = react.useRef(false);
   const messagesRef = react.useRef([]);
   messagesRef.current = messages;
   const sendMessage = react.useCallback(
-    async (message, currentSql) => {
+    async (message, currentSql, widgetContext) => {
       if (!client || isStreamingRef.current) return;
       abortRef.current?.abort();
       const controller = new AbortController();
@@ -4039,6 +4045,7 @@ function useLLMChat() {
       setStreamingContent("");
       setSuggestedSql(null);
       setError(null);
+      setStatusMessage(null);
       let accumulatedText = "";
       let lastSql = null;
       try {
@@ -4050,22 +4057,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;
           }
         }
@@ -4076,6 +4089,7 @@ function useLLMChat() {
       } finally {
         isStreamingRef.current = false;
         setIsStreaming(false);
+        setStatusMessage(null);
         if (accumulatedText) {
           const assistantMsg = {
             role: "assistant",
@@ -4100,6 +4114,7 @@ function useLLMChat() {
     setSuggestedSql(null);
     setError(null);
     setIsStreaming(false);
+    setStatusMessage(null);
   }, []);
   return {
     messages,
@@ -4108,7 +4123,8 @@ function useLLMChat() {
     suggestedSql,
     sendMessage,
     clearHistory,
-    error
+    error,
+    statusMessage
   };
 }
 var nodeStyles = {
@@ -9459,7 +9475,7 @@ function ChatBubble({ message, onApplySql }) {
     return /* @__PURE__ */ jsxRuntime.jsx("span", { children: part.value }, i);
   }) });
 }
-function ChatPanel({ currentSql, onApplySql }) {
+function ChatPanel({ currentSql, onApplySql, widgetContext }) {
   const { theme } = chunkLMTG3LRC_cjs.useTheme();
   const {
     messages,
@@ -9468,7 +9484,8 @@ function ChatPanel({ currentSql, onApplySql }) {
     suggestedSql,
     sendMessage,
     clearHistory,
-    error
+    error,
+    statusMessage
   } = useLLMChat();
   const [input, setInput] = react.useState("");
   const messagesEndRef = react.useRef(null);
@@ -9479,8 +9496,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 = react.useCallback(
     (e) => {
       if (e.key === "Enter" && !e.shiftKey) {
@@ -9601,6 +9618,24 @@ function ChatPanel({ currentSql, onApplySql }) {
         "\u258D"
       ] }),
       isStreaming && !streamingContent && /* @__PURE__ */ jsxRuntime.jsx("div", { style: streamingStyle, "data-testid": "chat-streaming", children: "Thinking..." }),
+      isStreaming && statusMessage && /* @__PURE__ */ jsxRuntime.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__ */ jsxRuntime.jsx(Icon, { name: "sync", size: 12 }),
+            /* @__PURE__ */ jsxRuntime.jsx("span", { children: statusMessage })
+          ]
+        }
+      ),
       /* @__PURE__ */ jsxRuntime.jsx("div", { ref: messagesEndRef })
     ] }),
     error && /* @__PURE__ */ jsxRuntime.jsx("div", { style: errorStyle, "data-testid": "chat-error", children: error }),
@@ -9712,5 +9747,5 @@ exports.usePinnedDashboards = usePinnedDashboards;
 exports.useQuery = useQuery;
 exports.useSavedQueries = useSavedQueries;
 exports.useSchema = useSchema;
-//# sourceMappingURL=chunk-PG7QBH3G.cjs.map
-//# sourceMappingURL=chunk-PG7QBH3G.cjs.map
+//# sourceMappingURL=chunk-NXXKG4GN.cjs.map
+//# sourceMappingURL=chunk-NXXKG4GN.cjs.map