diff --git a/templates/components/workflows/typescript/deep_research/workflow.ts b/templates/components/workflows/typescript/deep_research/workflow.ts index b387a5f12..007144775 100644 --- a/templates/components/workflows/typescript/deep_research/workflow.ts +++ b/templates/components/workflows/typescript/deep_research/workflow.ts @@ -1,8 +1,4 @@ -import { - DeepResearchEvent, - toSourceEvent, - toStreamGenerator, -} from "@llamaindex/server"; +import { toSourceEvent, toStreamGenerator } from "@llamaindex/server"; import { AgentInputData, AgentWorkflowContext, @@ -123,6 +119,19 @@ class ResearchEvent extends WorkflowEvent {} class ReportEvent extends WorkflowEvent<{}> {} class StopEvent extends StopEventBase> {} +type DeepResearchEventData = { + event: "retrieve" | "analyze" | "answer"; + state: "pending" | "inprogress" | "done" | "error"; + id?: string; + question?: string; + answer?: string; +}; + +class DeepResearchEvent extends WorkflowEvent<{ + type: "deep_research_event"; + data: DeepResearchEventData; +}> {} + // workflow definition class DeepResearchWorkflow extends Workflow< AgentWorkflowContext,