Remove submitted and unknown statuses

Author: LucaButBoring

src/examples/shared/inMemoryTaskStore.test.ts

@@ -13,7 +13,7 @@ describe('InMemoryTaskStore', () => {
     });
 
     describe('createTask', () => {
-        it('should create a new task with submitted status', async () => {
+        it('should create a new task with working status', async () => {
             const metadata: TaskMetadata = {
                 taskId: 'task-1',
                 keepAlive: 60000
@@ -28,7 +28,7 @@ describe('InMemoryTaskStore', () => {
             const task = await store.getTask('task-1');
             expect(task).toBeDefined();
             expect(task?.taskId).toBe('task-1');
-            expect(task?.status).toBe('submitted');
+            expect(task?.status).toBe('working');
             expect(task?.keepAlive).toBe(60000);
             expect(task?.pollInterval).toBe(500);
         });
@@ -99,9 +99,7 @@ describe('InMemoryTaskStore', () => {
             });
         });
 
-        it('should update task status from submitted to working', async () => {
-            await store.updateTaskStatus('status-test', 'working');
-
+        it('should keep task status as working', async () => {
             const task = await store.getTask('status-test');
             expect(task?.status).toBe('working');
         });

src/examples/shared/inMemoryTaskStore.ts

@@ -30,7 +30,7 @@ export class InMemoryTaskStore implements TaskStore {
 
         const task: Task = {
             taskId,
-            status: 'submitted',
+            status: 'working',
             keepAlive: metadata.keepAlive ?? null,
             pollInterval: metadata.pollInterval ?? 500
         };

src/shared/protocol.test.ts

@@ -40,11 +40,11 @@ function createMockTaskStore(options?: {
         createTask: jest.fn((taskMetadata: TaskMetadata, _1: RequestId, _2: Request) => {
             const task = (tasks[taskMetadata.taskId] = {
                 taskId: taskMetadata.taskId,
-                status: (taskMetadata.status as Task['status'] | undefined) ?? 'submitted',
+                status: (taskMetadata.status as Task['status'] | undefined) ?? 'working',
                 keepAlive: taskMetadata.keepAlive ?? null,
                 pollInterval: (taskMetadata.pollInterval as Task['pollInterval'] | undefined) ?? 1000
             });
-            options?.onStatus?.('submitted');
+            options?.onStatus?.('working');
             return Promise.resolve(task);
         }),
         getTask: jest.fn((taskId: string) => {
@@ -1362,7 +1362,6 @@ describe('Task-based execution', () => {
             await mockTaskStore.createTask(
                 {
                     taskId: 'task-3',
-                    status: 'submitted',
                     pollInterval: 500
                 },
                 1,
@@ -1398,7 +1397,7 @@ describe('Task-based execution', () => {
             const sentMessage = sendSpy.mock.calls[0][0];
             expect(sentMessage.jsonrpc).toBe('2.0');
             expect(sentMessage.id).toBe(2);
-            expect(sentMessage.result.tasks).toEqual([{ taskId: 'task-3', status: 'submitted', keepAlive: null, pollInterval: 500 }]);
+            expect(sentMessage.result.tasks).toEqual([{ taskId: 'task-3', status: 'working', keepAlive: null, pollInterval: 500 }]);
             expect(sentMessage.result.nextCursor).toBeUndefined();
             expect(sentMessage.result._meta).toEqual({});
         });

src/shared/task.ts

@@ -79,8 +79,8 @@ export interface TaskStore {
  * Terminal states are those where the task has finished and will not change.
  *
  * @param status - The task status to check
- * @returns True if the status is terminal (completed, failed, cancelled, or unknown)
+ * @returns True if the status is terminal (completed, failed, or cancelled)
  */
 export function isTerminal(status: Task['status']): boolean {
-    return status === 'completed' || status === 'failed' || status === 'cancelled' || status === 'unknown';
+    return status === 'completed' || status === 'failed' || status === 'cancelled';
 }

src/types.ts

@@ -604,7 +604,7 @@ export const PaginatedResultSchema = ResultSchema.extend({
  */
 export const TaskSchema = z.object({
     taskId: z.string(),
-    status: z.enum(['submitted', 'working', 'input_required', 'completed', 'failed', 'cancelled', 'unknown']),
+    status: z.enum(['working', 'input_required', 'completed', 'failed', 'cancelled']),
     keepAlive: z.union([z.number(), z.null()]),
     pollInterval: z.optional(z.number()),
     error: z.optional(z.string())