improve code

Author: leehuwuj

templates/components/agents/typescript/blog/workflow/agents.ts

@@ -36,7 +36,6 @@ If you use the tools but don't find any related information, please return "I di
 If the request doesn't need any new information because it was in the conversation history, please return "The task doesn't need any new information. Please reuse the existing content in the conversation history.
 `,
     chatHistory,
-    writeEvents: true,
   });
 };
 
@@ -57,7 +56,6 @@ Example:
     -> This is not your task: Create a PDF
     Please note that a localhost link is acceptable, but dummy links like "example.com" or "your-website.com" are not valid.`,
     chatHistory,
-    writeEvents: true,
   });
 };
 
@@ -76,7 +74,6 @@ Example:
     -> Your task: Review whether the main content of the post is about the history of the internet and if it is written in English.
     -> This is not your task: Create blog post, create PDF, write in English.`,
     chatHistory,
-    writeEvents: true,
   });
 };
 
@@ -94,6 +91,5 @@ Otherwise, simply return the content of the post.`;
     tools: tools,
     systemPrompt: systemPrompt,
     chatHistory,
-    writeEvents: true,
   });
 };

templates/components/agents/typescript/financial_report/workflow/factory.ts

@@ -1,5 +1,5 @@
 import { ChatMessage, ToolCallLLM } from "llamaindex";
-import { FinancialReportWorkflow } from "./finReport";
+import { FinancialReportWorkflow } from "./fin-report";
 import { getAvailableTools } from "./tools";
 
 const TIMEOUT = 360 * 1000;

templates/components/agents/typescript/financial_report/workflow/finReport.ts renamed to templates/components/agents/typescript/financial_report/workflow/fin-report.ts

@@ -1,5 +1,5 @@
 import { ChatMessage, ToolCallLLM } from "llamaindex";
-import { FormFillingWorkflow } from "./formFilling";
+import { FormFillingWorkflow } from "./form-filling";
 import { getAvailableTools } from "./tools";
 
 const TIMEOUT = 360 * 1000;

templates/components/agents/typescript/form_filling/workflow/factory.ts

@@ -64,7 +64,25 @@ export function retrieveMessageContent(messages: Message[]): MessageContent {
   ];
 }
 
-export function retrieveAgentHistoryMessage(
+export function convertToChatHistory(messages: Message[]): ChatMessage[] {
+  if (!messages || !Array.isArray(messages)) {
+    return [];
+  }
+  const agentHistory = retrieveAgentHistoryMessage(messages);
+  if (agentHistory) {
+    const previousMessages = messages.slice(0, -1);
+    return [...previousMessages, agentHistory].map((msg) => ({
+      role: msg.role as MessageType,
+      content: msg.content,
+    }));
+  }
+  return messages.map((msg) => ({
+    role: msg.role as MessageType,
+    content: msg.content,
+  }));
+}
+
+function retrieveAgentHistoryMessage(
   messages: Message[],
   maxAgentMessages = 10,
 ): ChatMessage | null {
@@ -85,24 +103,6 @@ export function retrieveAgentHistoryMessage(
   return null;
 }
 
-export function convertToChatHistory(messages: Message[]): ChatMessage[] {
-  if (!messages || !Array.isArray(messages)) {
-    return [];
-  }
-  const agentHistory = retrieveAgentHistoryMessage(messages);
-  if (agentHistory) {
-    const previousMessages = messages.slice(0, -1);
-    return [...previousMessages, agentHistory].map((msg) => ({
-      role: msg.role as MessageType,
-      content: msg.content,
-    }));
-  }
-  return messages.map((msg) => ({
-    role: msg.role as MessageType,
-    content: msg.content,
-  }));
-}
-
 function getFileContent(file: DocumentFile): string {
   let defaultContent = `=====File: ${file.name}=====\n`;
   // Include file URL if it's available

templates/components/agents/typescript/form_filling/workflow/formFilling.ts renamed to templates/components/agents/typescript/form_filling/workflow/form-filling.ts

@@ -1,4 +1,4 @@
-import { Message, StreamingTextResponse } from "ai";
+import { Message, streamToResponse } from "ai";
 import { Request, Response } from "express";
 import {
   convertToChatHistory,
@@ -28,7 +28,7 @@ export const chat = async (req: Request, res: Response) => {
     const { stream, dataStream } =
       await createStreamFromWorkflowContext(context);
 
-    return new StreamingTextResponse(stream, {}, dataStream);
+    return streamToResponse(stream, res, {}, dataStream);
   } catch (error) {
     console.error("[LlamaIndex]", error);
     return res.status(500).json({