update workflow code

Author: leehuwuj

templates/components/agents/typescript/financial_report/app/api/chat/workflow/finReport.ts

@@ -10,7 +10,6 @@ import {
   ChatMemoryBuffer,
   ChatMessage,
   ChatResponseChunk,
-  MessageContent,
   Settings,
   ToolCall,
   ToolCallLLM,
@@ -41,7 +40,7 @@ For the query engine tool, you should break down the user request into a list of
 
 export class FinancialReportWorkflow extends Workflow<
   null,
-  string | MessageContent,
+  AgentInput,
   ChatResponseChunk
 > {
   llm: ToolCallLLM;
@@ -132,7 +131,7 @@ export class FinancialReportWorkflow extends Workflow<
     ctx: HandlerContext<null>,
     ev: StartEvent<AgentInput>,
   ) {
-    const message = ev.data;
+    const { message } = ev.data;
 
     if (this.systemPrompt) {
       this.memory.put({ role: "system", content: this.systemPrompt });

templates/components/agents/typescript/form_filling/app/api/chat/workflow/factory.ts

@@ -44,7 +44,6 @@ const prepareChatHistory = (chatHistory: Message[]): ChatMessage[] => {
 export async function createWorkflow(options: {
   chatHistory: Message[];
   llm?: ToolCallLLM;
-  writeEvents?: boolean;
 }) {
   const enhancedChatHistory = prepareChatHistory(options.chatHistory);
 
@@ -71,7 +70,6 @@ export async function createWorkflow(options: {
     queryEngineTool,
     fillMissingCellsTool,
     llm: options.llm,
-    writeEvents: options.writeEvents,
     timeout: TIMEOUT,
   });
 

templates/components/agents/typescript/form_filling/app/api/chat/workflow/formFilling.ts

@@ -1,5 +1,4 @@
 import {
-  Context,
   HandlerContext,
   StartEvent,
   StopEvent,
@@ -11,7 +10,6 @@ import {
   ChatMemoryBuffer,
   ChatMessage,
   ChatResponseChunk,
-  MessageContent,
   Settings,
   ToolCall,
   ToolCallLLM,
@@ -35,15 +33,16 @@ class FillMissingCellsEvent extends WorkflowEvent<{
 }> {}
 
 const DEFAULT_SYSTEM_PROMPT = `
-You are a helpful assistant who helps fill missing cells in a CSV file. Only use the local file path for the tools.
-Only use provided data - never make up any information yourself. Fill N/A if an answer is not found.
+You are a helpful assistant who helps fill missing cells in a CSV file.
+Only use the information from the query engine tool - don't make up any information yourself. Fill N/A if an answer is not found.
 If there is no query engine tool or the gathered information has many N/A values indicating the questions don't match the data, respond with a warning and ask the user to upload a different file or connect to a knowledge base.
 You can make multiple tool calls at once but only call with the same tool.
+Only use the local file path for the tools.
 `;
 
 export class FormFillingWorkflow extends Workflow<
   null,
-  string | MessageContent,
+  AgentInput,
   ChatResponseChunk
 > {
   llm: ToolCallLLM;
@@ -52,7 +51,6 @@ export class FormFillingWorkflow extends Workflow<
   queryEngineTool?: BaseToolWithCall;
   fillMissingCellsTool: BaseToolWithCall;
   systemPrompt?: string;
-  writeEvents?: boolean;
 
   constructor(options: {
     llm?: ToolCallLLM;
@@ -72,7 +70,6 @@ export class FormFillingWorkflow extends Workflow<
 
     this.llm = options.llm ?? (Settings.llm as ToolCallLLM);
     this.systemPrompt = options.systemPrompt ?? DEFAULT_SYSTEM_PROMPT;
-    this.writeEvents = options.writeEvents;
     this.extractorTool = options.extractorTool;
     this.queryEngineTool = options.queryEngineTool;
     this.fillMissingCellsTool = options.fillMissingCellsTool;
@@ -133,7 +130,7 @@ export class FormFillingWorkflow extends Workflow<
     ctx: HandlerContext<null>,
     ev: StartEvent<AgentInput>,
   ) {
-    const message = ev.data;
+    const { message } = ev.data;
 
     if (this.systemPrompt) {
       this.memory.put({ role: "system", content: this.systemPrompt });
@@ -190,7 +187,7 @@ export class FormFillingWorkflow extends Workflow<
   }
 
   private async handleExtractMissingCells(
-    ctx: Context,
+    ctx: HandlerContext<null>,
     ev: ExtractMissingCellsEvent,
   ) {
     ctx.sendEvent(
@@ -213,7 +210,10 @@ export class FormFillingWorkflow extends Workflow<
     return new InputEvent({ input: this.memory.getMessages() });
   }
 
-  private async handleFindAnswers(ctx: Context, ev: FindAnswersEvent) {
+  private async handleFindAnswers(
+    ctx: HandlerContext<null>,
+    ev: FindAnswersEvent,
+  ) {
     const { toolCalls } = ev.data;
     if (!this.queryEngineTool) {
       throw new Error("Query engine tool is not available");

templates/components/llamaindex/typescript/documents/upload.ts

@@ -1,7 +1,5 @@
 import { Document, LLamaCloudFileService, VectorStoreIndex } from "llamaindex";
 import { LlamaCloudIndex } from "llamaindex/cloud/LlamaCloudIndex";
-import fs from "node:fs/promises";
-import path from "node:path";
 import { DocumentFile } from "../streaming/annotations";
 import { parseFile, storeFile } from "./helper";
 import { runPipeline } from "./pipeline";
@@ -61,14 +59,3 @@ export async function uploadDocument(
   fileMetadata.refs = documentIds;
   return fileMetadata;
 }
-
-const hasCodeExecutorTool = async () => {
-  const codeExecutorTools = ["interpreter", "artifact"];
-
-  const configFile = path.join("config", "tools.json");
-  const toolConfig = JSON.parse(await fs.readFile(configFile, "utf8"));
-
-  const localTools = toolConfig.local || {};
-  // Check if local tools contains codeExecutorTools
-  return codeExecutorTools.some((tool) => localTools[tool] !== undefined);
-};

templates/components/multiagent/typescript/nextjs/route.ts

@@ -1,6 +1,5 @@
 import { initObservability } from "@/app/observability";
 import { StreamingTextResponse, type Message } from "ai";
-import { MessageContent } from "llamaindex";
 import { NextRequest, NextResponse } from "next/server";
 import { initSettings } from "./engine/settings";
 import {
@@ -33,10 +32,12 @@ export async function POST(request: NextRequest) {
     const userMessageContent = retrieveMessageContent(messages);
     const workflow = await createWorkflow({
       chatHistory: messages,
-      writeEvents: true,
     });
 
-    const context = workflow.run(userMessageContent as MessageContent);
+    const context = workflow.run({
+      message: userMessageContent,
+      streaming: true,
+    });
     const { stream, streamData } =
       await createStreamFromWorkflowContext(context);
 

templates/components/multiagent/typescript/workflow/type.ts

@@ -2,7 +2,7 @@ import { WorkflowEvent } from "@llamaindex/workflow";
 import { MessageContent } from "llamaindex";
 
 export type AgentInput = {
-  message: string | MessageContent;
+  message: MessageContent;
   streaming?: boolean;
 };