Fix use infinite tasks hook (#14)

Removes the useTasks hook (was not in use) and fixes the useCreateTask
hook to respect the InfiniteData type that is returned from the
useInfiniteQuery hook

This caused a bug on the frontend where trying to create a task after
fetching all tasks would cause an error

Author: MichaelSun48

agentex-ui/hooks/use-create-task.ts

@@ -1,14 +1,18 @@
 'use client';
 
-import { useMutation, useQueryClient } from '@tanstack/react-query';
+import {
+  InfiniteData,
+  useMutation,
+  useQueryClient,
+} from '@tanstack/react-query';
 import { agentRPCNonStreaming } from 'agentex/lib';
 
 import { toast } from '@/components/agentex/toast';
 
 import { tasksKeys } from './use-tasks';
 
 import type AgentexSDK from 'agentex';
-import type { Task } from 'agentex/resources';
+import type { Task, TaskListResponse } from 'agentex/resources';
 
 type CreateTaskParams = {
   agentName: string;
@@ -51,16 +55,37 @@ export function useCreateTask({
 
       return response.result;
     },
-    onSuccess: newTask => {
-      // Add new task to the unfiltered cache
-      queryClient.setQueryData<Task[]>(tasksKeys.all, old => {
-        if (!old) return [newTask];
-        // Avoid duplicates
-        if (old.some(t => t.id === newTask.id)) {
-          return old;
-        }
-        return [...old, newTask]; // Add to end (will be reversed in UI to show first)
-      });
+    onSuccess: (newTask, variables) => {
+      // Helper function to update infinite query cache
+      const updateInfiniteCache = (queryKey: readonly unknown[]) => {
+        queryClient.setQueryData<InfiniteData<TaskListResponse>>(
+          queryKey,
+          old => {
+            if (!old) {
+              // If no cache exists, create initial structure
+              return {
+                pages: [[newTask]],
+                pageParams: [1],
+              };
+            }
+
+            // Add new task to the first page (prepend to show at top)
+            const firstPage = old.pages[0] ?? [];
+            if (firstPage.some(t => t.id === newTask.id)) {
+              return old; // Avoid duplicates
+            }
+
+            return {
+              ...old,
+              pages: [[newTask, ...firstPage], ...old.pages.slice(1)],
+            };
+          }
+        );
+      };
+
+      // Update both the agent-specific cache and the generic "all tasks" cache
+      updateInfiniteCache(tasksKeys.byAgentName(variables.agentName));
+      updateInfiniteCache(tasksKeys.all);
 
       // Invalidate all task queries to ensure consistency
       queryClient.invalidateQueries({ queryKey: tasksKeys.all });

agentex-ui/hooks/use-tasks.ts

@@ -19,29 +19,6 @@ export const tasksKeys = {
   byId: (taskId: string) => [...tasksKeys.all, taskId] as const,
 };
 
-/**
- * Fetches the list of tasks, optionally filtered by agent name
- */
-export function useTasks(
-  agentexClient: AgentexSDK,
-  options?: { agentName?: string }
-) {
-  const { agentName } = options || {};
-
-  return useQuery({
-    queryKey: tasksKeys.byAgentName(agentName),
-    queryFn: async (): Promise<TaskListResponse> => {
-      const params: TaskListParams = {
-        relationships: ['agents'],
-        ...(agentName ? { agent_name: agentName } : {}),
-      };
-      return agentexClient.tasks.list(params);
-    },
-    staleTime: 30 * 1000, // 30 seconds
-    refetchOnWindowFocus: true,
-  });
-}
-
 /**
  * Fetches a single task by ID
  */