refactor: clean up chat-atoms by removing unused imports and optimizing cancelMessageAtom logic, fix: enhance time field detection in ChatArtifactEnhanced and ArtifactRenderer components, feat: implement time field selector with improved detection logic and user experience, fix: update query processor to handle table variable replacement, chore: update PanelEdit to ensure useParentTimeFilter is set for new panels and improve time range handling, feat: add Switch component for better UI control

Author: caioricciuti

bun.lock

@@ -1,6 +1,6 @@
 {
   "name": "gigapi-ui",
-  "version": "1.0.19",
+  "version": "1.0.20",
   "type": "module",
   "description": "UI interface for GigAPI: The Infinite Timeseries Lakehouse. GigAPI UI provides a slick web interface to query time-series using GigAPI Catalog Metadata + DuckDB",
   "scripts": {
@@ -27,6 +27,7 @@
     "@radix-ui/react-select": "^2.2.5",
     "@radix-ui/react-separator": "^1.1.7",
     "@radix-ui/react-slot": "^1.2.3",
+    "@radix-ui/react-switch": "^1.2.5",
     "@radix-ui/react-tabs": "^1.1.12",
     "@radix-ui/react-tooltip": "^1.2.7",
     "@tailwindcss/vite": "^4.1.11",

package.json

@@ -14,7 +14,6 @@ import { SchemaContextBuilder } from "@/lib/schema-context-builder";
 import { ConversationAnalyzer } from "@/lib/conversation-analyzer";
 import { StorageUtils } from "@/lib/storage-utils";
 import { schemaCacheAtom, loadAndCacheTableSchemaAtom } from "./database-atoms";
-import { apiUrlAtom } from "./connection-atoms";
 import { getCurrentTabData } from "./tab-atoms";
 import type {
   ChatSession,
@@ -244,7 +243,6 @@ export const sessionListAtom = atom((get) => {
   );
 });
 
-
 // ============================================================================
 // Helper Functions
 // ============================================================================
@@ -332,54 +330,51 @@ export const createSessionAtom = atom(
 );
 
 // Cancel streaming message
-export const cancelMessageAtom = atom(
-  null,
-  (get, set, sessionId: string) => {
-    const abortControllers = get(activeAbortControllersAtom);
-    const controller = abortControllers.get(sessionId);
-    
-    if (controller) {
-      // Abort the fetch request
-      controller.abort();
-      
-      // Remove from active controllers
-      const newControllers = new Map(abortControllers);
-      newControllers.delete(sessionId);
-      set(activeAbortControllersAtom, newControllers);
-      
-      // Update the session to remove streaming flag from the last message
-      const sessions = get(chatSessionsAtom);
-      const session = sessions[sessionId];
-      if (session && session.messages.length > 0) {
-        const lastMessage = session.messages[session.messages.length - 1];
-        if (lastMessage.metadata?.isStreaming) {
-          const updatedMessages = [...session.messages];
-          updatedMessages[updatedMessages.length - 1] = {
-            ...lastMessage,
-            metadata: {
-              ...lastMessage.metadata,
-              isStreaming: false,
-              wasCancelled: true,
-            },
-          };
-          
-          set(chatSessionsAtom, {
-            ...sessions,
-            [sessionId]: {
-              ...session,
-              messages: updatedMessages,
-              updatedAt: new Date().toISOString(),
-            },
-          });
-        }
+export const cancelMessageAtom = atom(null, (get, set, sessionId: string) => {
+  const abortControllers = get(activeAbortControllersAtom);
+  const controller = abortControllers.get(sessionId);
+
+  if (controller) {
+    // Abort the fetch request
+    controller.abort();
+
+    // Remove from active controllers
+    const newControllers = new Map(abortControllers);
+    newControllers.delete(sessionId);
+    set(activeAbortControllersAtom, newControllers);
+
+    // Update the session to remove streaming flag from the last message
+    const sessions = get(chatSessionsAtom);
+    const session = sessions[sessionId];
+    if (session && session.messages.length > 0) {
+      const lastMessage = session.messages[session.messages.length - 1];
+      if (lastMessage.metadata?.isStreaming) {
+        const updatedMessages = [...session.messages];
+        updatedMessages[updatedMessages.length - 1] = {
+          ...lastMessage,
+          metadata: {
+            ...lastMessage.metadata,
+            isStreaming: false,
+            wasCancelled: true,
+          },
+        };
+
+        set(chatSessionsAtom, {
+          ...sessions,
+          [sessionId]: {
+            ...session,
+            messages: updatedMessages,
+            updatedAt: new Date().toISOString(),
+          },
+        });
       }
-      
-      return true;
     }
-    
-    return false;
+
+    return true;
   }
-);
+
+  return false;
+});
 
 // Send message with streaming support
 export const sendMessageAtom = atom(
@@ -515,7 +510,9 @@ export const sendMessageAtom = atom(
 
       // Get time context from current tab
       const currentTab = getCurrentTabData();
-      const timeRange = currentTab?.timeRange ? JSON.stringify(currentTab.timeRange) : null;
+      const timeRange = currentTab?.timeRange
+        ? JSON.stringify(currentTab.timeRange)
+        : null;
       const selectedDb = currentTab?.database || "";
       const selectedTable = currentTab?.table || "";
 
@@ -686,7 +683,7 @@ export const sendMessageAtom = atom(
         },
         abortController.signal
       );
-      
+
       // Clean up abort controller after successful completion
       const currentControllers = get(activeAbortControllersAtom);
       const updatedControllers = new Map(currentControllers);
@@ -700,9 +697,9 @@ export const sendMessageAtom = atom(
       set(activeAbortControllersAtom, updatedControllers);
 
       // Check if the error is due to abort
-      if (error instanceof Error && error.name === 'AbortError') {
+      if (error instanceof Error && error.name === "AbortError") {
         // Don't show error message for cancelled requests
-        console.log('Message streaming was cancelled');
+        console.log("Message streaming was cancelled");
         return;
       }
 
@@ -789,7 +786,11 @@ export const deleteSessionAtom = atom(null, (get, set, sessionId: string) => {
 // Edit message
 export const editMessageAtom = atom(
   null,
-  async (get, set, options: { sessionId: string; messageId: string; newContent: string }) => {
+  async (
+    get,
+    set,
+    options: { sessionId: string; messageId: string; newContent: string }
+  ) => {
     const { sessionId, messageId, newContent } = options;
     const sessions = get(chatSessionsAtom);
     const session = sessions[sessionId];
@@ -844,7 +845,11 @@ export const editMessageAtom = atom(
 // Regenerate from message
 export const regenerateFromMessageAtom = atom(
   null,
-  async (get, set, options: { sessionId: string; messageId: string; isAgentic?: boolean }) => {
+  async (
+    get,
+    set,
+    options: { sessionId: string; messageId: string; isAgentic?: boolean }
+  ) => {
     const { sessionId, messageId, isAgentic } = options;
     const sessions = get(chatSessionsAtom);
     const session = sessions[sessionId];
@@ -911,7 +916,9 @@ export const regenerateFromMessageAtom = atom(
 
       // Get time context from current tab
       const currentTab = getCurrentTabData();
-      const timeRange = currentTab?.timeRange ? JSON.stringify(currentTab.timeRange) : null;
+      const timeRange = currentTab?.timeRange
+        ? JSON.stringify(currentTab.timeRange)
+        : null;
       const selectedDb = currentTab?.database || "";
       const selectedTable = currentTab?.table || "";
 
@@ -963,7 +970,10 @@ export const regenerateFromMessageAtom = atom(
           function processChunkForThinking(text: string) {
             const thinkingStartMatch = text.match(/<think>/);
             if (thinkingStartMatch && thinkingStartMatch.index !== undefined) {
-              const beforeThinking = text.substring(0, thinkingStartMatch.index);
+              const beforeThinking = text.substring(
+                0,
+                thinkingStartMatch.index
+              );
               const afterThinking = text.substring(
                 thinkingStartMatch.index + thinkingStartMatch[0].length
               );
@@ -1049,7 +1059,7 @@ export const regenerateFromMessageAtom = atom(
         },
         abortController.signal
       );
-      
+
       // Clean up abort controller after successful completion
       const currentControllers = get(activeAbortControllersAtom);
       const updatedControllers = new Map(currentControllers);
@@ -1063,8 +1073,8 @@ export const regenerateFromMessageAtom = atom(
       set(activeAbortControllersAtom, updatedControllers);
 
       // Check if the error is due to abort
-      if (error instanceof Error && error.name === 'AbortError') {
-        console.log('Message streaming was cancelled');
+      if (error instanceof Error && error.name === "AbortError") {
+        console.log("Message streaming was cancelled");
         return;
       }
 
@@ -1122,7 +1132,6 @@ export const renameSessionAtom = atom(
   }
 );
 
-
 // Delete connection
 export const deleteConnectionAtom = atom(
   null,
@@ -1325,11 +1334,9 @@ Example: When user asks "show me sales by month", respond with:
         includeIndices: true,
         summaryOnly: false,
         includeRecentContext: true, // Always include recent context for better continuity
-        isAgentic, // Pass agentic mode flag
-        // Enhanced with sample data
+        isAgentic,
         includeSampleData: true,
         sampleDataLimit: 5,
-        apiUrl: queryContext.apiUrl || "http://localhost:7971/query",
       }
     );
 

src/atoms/chat-atoms.ts

@@ -1,5 +1,4 @@
 import { atom } from "jotai";
-import { atomWithStorage } from "jotai/utils";
 import type { TimeRange } from "@/types/tab.types";
 import { currentTabTimeFieldAtom, currentTabTimeRangeAtom, currentTabTimeZoneAtom } from "./tab-atoms";