apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-01 03:46:25 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

refactor: integrate TodoListMiddleware and update related components

Browse source 
- Removed the write_todos tool as it is now included by default through TodoListMiddleware in the deep agent.
- Updated the system prompt and documentation to reflect the integration of TodoListMiddleware, clarifying its capabilities for managing planning and todo lists.
- Enhanced the chat handling logic to extract todos directly from the deep agent's command output, ensuring seamless user experience.
- Refactored UI components to align with the new data structure and improve rendering of todo items, including updates to the Plan and TodoItem components.
- Cleaned up code for better maintainability and readability, following recent refactoring efforts.
This commit is contained in:
Anish Sarkar 2025-12-27 15:18:34 +05:30
parent 8a3ab3dfac
commit c28a90fc29
10 changed files with 172 additions and 419 deletions
Download patch file Download diff file Expand all files Collapse all files

21
surfsense_web/atoms/chat/plan-state.atom.ts
View file

@ -11,15 +11,13 @@ import { atom } from "jotai";
export interface PlanTodo {
id: string;
label: string;
content: string;
status: "pending" | "in_progress" | "completed" | "cancelled";
description?: string;
}
export interface PlanState {
id: string;
title: string;
description?: string;
todos: PlanTodo[];
lastUpdated: number;
/** The toolCallId of the first component that rendered this plan */
@ -96,7 +94,6 @@ export const planStatesAtom = atom<Map<string, PlanState>>(new Map());
export interface UpdatePlanInput {
id: string;
title: string;
description?: string;
todos: PlanTodo[];
toolCallId: string;
}
@ -119,7 +116,6 @@ export const updatePlanStateAtom = atom(null, (get, set, plan: UpdatePlanInput)
states.set(canonicalTitle, {
id: plan.id,
title: canonicalTitle,
description: plan.description,
todos: plan.todos,
lastUpdated: Date.now(),
ownerToolCallId,
@ -152,12 +148,10 @@ export interface HydratePlanInput {
result: {
id?: string;
title?: string;
description?: string;
todos?: Array<{
id: string;
label: string;
id?: string;
content: string;
status: "pending" | "in_progress" | "completed" | "cancelled";
description?: string;
}>;
};
}
@ -166,7 +160,7 @@ export const hydratePlanStateAtom = atom(null, (get, set, plan: HydratePlanInput
if (!plan.result?.todos || plan.result.todos.length === 0) return;
const states = new Map(get(planStatesAtom));
const title = plan.result.title || "Planning Approach";
const title = plan.result.title || "Plan";
// Register this as the owner if no plan exists yet
registerPlanOwner(title, plan.toolCallId);
@ -181,8 +175,11 @@ export const hydratePlanStateAtom = atom(null, (get, set, plan: HydratePlanInput
states.set(canonicalTitle, {
id: plan.result.id || `plan-${Date.now()}`,
title: canonicalTitle,
description: plan.result.description,
todos: plan.result.todos,
todos: plan.result.todos.map((t, i) => ({
id: t.id || `todo-${i}`,
content: t.content,
status: t.status,
})),
lastUpdated: Date.now(),
ownerToolCallId,
});