@librechat/agents 2.3.1 → 2.3.3

package/src/messages/formatAgentMessages.test.ts

@@ -362,14 +362,14 @@ describe('formatAgentMessages', () => {
       { role: 'user', content: 'Hello' },
       { role: 'assistant', content: 'Hi there!' },
     ];
-    
+
     const indexTokenCountMap = {
       0: 5,  // 5 tokens for "Hello"
       1: 10, // 10 tokens for "Hi there!"
     };
-    
+
     const result = formatAgentMessages(payload, indexTokenCountMap);
-    
+
     expect(result.messages).toHaveLength(2);
     expect(result.indexTokenCountMap).toBeDefined();
     expect(result.indexTokenCountMap?.[0]).toBe(5);
@@ -399,23 +399,23 @@ describe('formatAgentMessages', () => {
         ],
       },
     ];
-    
+
     const indexTokenCountMap = {
       0: 10,  // 10 tokens for "What's the weather?"
       1: 50,  // 50 tokens for the assistant message with tool call
     };
-    
+
     const result = formatAgentMessages(payload, indexTokenCountMap);
-    
+
     // The original message at index 1 should be split into two messages
     expect(result.messages).toHaveLength(3);
     expect(result.indexTokenCountMap).toBeDefined();
     expect(result.indexTokenCountMap?.[0]).toBe(10);  // User message stays the same
-    
+
     // The assistant message tokens should be distributed across the resulting messages
     const totalAssistantTokens = Object.values(result.indexTokenCountMap || {})
       .reduce((sum, count) => sum + count, 0) - 10; // Subtract user message tokens
-    
+
     expect(totalAssistantTokens).toBe(50); // Should match the original token count
   });
 
@@ -460,23 +460,23 @@ describe('formatAgentMessages', () => {
         ],
       },
     ];
-    
+
     const indexTokenCountMap = {
       0: 100,  // 100 tokens for the complex assistant message
     };
-    
+
     const result = formatAgentMessages(payload, indexTokenCountMap);
-    
+
     // One message expands to 5 messages (2 tool calls + text before, between, and after)
     expect(result.messages).toHaveLength(5);
     expect(result.indexTokenCountMap).toBeDefined();
-    
+
     // The sum of all token counts should equal the original
     const totalTokens = Object.values(result.indexTokenCountMap || {})
       .reduce((sum, count) => sum + count, 0);
-    
+
     expect(totalTokens).toBe(100);
-    
+
     // Check that each resulting message has a token count
     for (let i = 0; i < result.messages.length; i++) {
       expect(result.indexTokenCountMap?.[i]).toBeDefined();
@@ -495,17 +495,17 @@ describe('formatAgentMessages', () => {
         ],
       },
     ];
-    
+
     const indexTokenCountMap = {
       0: 60,  // 60 tokens for the message with filtered content
     };
-    
+
     const result = formatAgentMessages(payload, indexTokenCountMap);
-    
+
     // Only one message should remain after filtering
     expect(result.messages).toHaveLength(1);
     expect(result.indexTokenCountMap).toBeDefined();
-    
+
     // All tokens should be assigned to the remaining message
     expect(result.indexTokenCountMap?.[0]).toBe(60);
   });
@@ -522,17 +522,17 @@ describe('formatAgentMessages', () => {
         ],
       },
     ];
-    
+
     const indexTokenCountMap = {
       0: 40,  // 40 tokens for the message with filtered content
     };
-    
+
     const result = formatAgentMessages(payload, indexTokenCountMap);
-    
+
     // No messages should remain after filtering
     expect(result.messages).toHaveLength(0);
     expect(result.indexTokenCountMap).toBeDefined();
-    
+
     // The token count map should be empty since there are no messages
     expect(Object.keys(result.indexTokenCountMap || {})).toHaveLength(0);
   });
@@ -561,29 +561,29 @@ describe('formatAgentMessages', () => {
         ],
       },
     ];
-    
+
     const indexTokenCountMap = {
       0: 15,  // 15 tokens for the user message
       1: 45,  // 45 tokens for the assistant message with tool call
     };
-    
+
     const result = formatAgentMessages(payload, indexTokenCountMap);
-    
+
     // 2 input messages become 3 output messages (user + assistant + tool)
     expect(payload).toHaveLength(2);
     expect(result.messages).toHaveLength(3);
     expect(result.indexTokenCountMap).toBeDefined();
     expect(Object.keys(result.indexTokenCountMap ?? {}).length).toBe(3);
-    
+
     // Check message types
     expect(result.messages[0]).toBeInstanceOf(HumanMessage);
     expect(result.messages[1]).toBeInstanceOf(AIMessage);
     expect(result.messages[2]).toBeInstanceOf(ToolMessage);
-    
+
     // The sum of all token counts should equal the original total
     const totalTokens = Object.values(result.indexTokenCountMap || {})
       .reduce((sum, count) => sum + count, 0);
-    
+
     expect(totalTokens).toBe(60); // 15 + 45
   });
 
@@ -600,30 +600,30 @@ describe('formatAgentMessages', () => {
         ],
       },
     ];
-    
+
     const indexTokenCountMap = {
       0: 10,  // 10 tokens for the user message
       1: 30,  // 30 tokens for the assistant message that will be filtered out
     };
-    
+
     const result = formatAgentMessages(payload, indexTokenCountMap);
-    
+
     // 2 input messages become 1 output message (only the user message remains)
     expect(payload).toHaveLength(2);
     expect(result.messages).toHaveLength(1);
     expect(result.indexTokenCountMap).toBeDefined();
     expect(Object.keys(result.indexTokenCountMap ?? {}).length).toBe(1);
-    
+
     // Check message type
     expect(result.messages[0]).toBeInstanceOf(HumanMessage);
-    
+
     // Only the user message tokens should remain
     expect(result.indexTokenCountMap?.[0]).toBe(10);
-    
+
     // The total tokens should be just the user message tokens
     const totalTokens = Object.values(result.indexTokenCountMap || {})
       .reduce((sum, count) => sum + count, 0);
-    
+
     expect(totalTokens).toBe(10);
   });
 });

package/src/messages/formatAgentMessages.tools.test.ts

@@ -1,4 +1,4 @@
-import { HumanMessage, AIMessage, SystemMessage, ToolMessage } from '@langchain/core/messages';
+import { HumanMessage, AIMessage, ToolMessage } from '@langchain/core/messages';
 import type { TPayload } from '@/types';
 import { formatAgentMessages } from './format';
 import { ContentTypes } from '@/common';
@@ -27,9 +27,9 @@ describe('formatAgentMessages with tools parameter', () => {
         ],
       },
     ];
-    
+
     const result = formatAgentMessages(payload);
-    
+
     expect(result.messages).toHaveLength(3);
     expect(result.messages[0]).toBeInstanceOf(HumanMessage);
     expect(result.messages[1]).toBeInstanceOf(AIMessage);
@@ -61,17 +61,17 @@ describe('formatAgentMessages with tools parameter', () => {
         ],
       },
     ];
-    
+
     // Provide an empty set of allowed tools
     const allowedTools = new Set<string>();
-    
+
     const result = formatAgentMessages(payload, undefined, allowedTools);
-    
+
     // Should convert to a single AIMessage with string content
     expect(result.messages).toHaveLength(2);
     expect(result.messages[0]).toBeInstanceOf(HumanMessage);
     expect(result.messages[1]).toBeInstanceOf(AIMessage);
-    
+
     // The content should be a string representation of both messages
     expect(typeof result.messages[1].content).toBe('string');
     expect(result.messages[1].content).toEqual('AI: Let me check the weather for you.\nTool: check_weather, Sunny, 75°F');
@@ -100,17 +100,17 @@ describe('formatAgentMessages with tools parameter', () => {
         ],
       },
     ];
-    
+
     // Provide a set of allowed tools that doesn't include 'check_weather'
     const allowedTools = new Set(['search', 'calculator']);
-    
+
     const result = formatAgentMessages(payload, undefined, allowedTools);
-    
+
     // Should convert to a single AIMessage with string content
     expect(result.messages).toHaveLength(2);
     expect(result.messages[0]).toBeInstanceOf(HumanMessage);
     expect(result.messages[1]).toBeInstanceOf(AIMessage);
-    
+
     // The content should be a string representation of both messages
     expect(typeof result.messages[1].content).toBe('string');
     expect(result.messages[1].content).toEqual('AI: Let me check the weather for you.\nTool: check_weather, Sunny, 75°F');
@@ -139,12 +139,12 @@ describe('formatAgentMessages with tools parameter', () => {
         ],
       },
     ];
-    
+
     // Provide a set of allowed tools that includes 'check_weather'
     const allowedTools = new Set(['check_weather', 'search']);
-    
+
     const result = formatAgentMessages(payload, undefined, allowedTools);
-    
+
     // Should keep the original structure
     expect(result.messages).toHaveLength(3);
     expect(result.messages[0]).toBeInstanceOf(HumanMessage);
@@ -189,17 +189,17 @@ describe('formatAgentMessages with tools parameter', () => {
         ],
       },
     ];
-    
+
     // Allow calculator but not check_weather
     const allowedTools = new Set(['calculator', 'search']);
-    
+
     const result = formatAgentMessages(payload, undefined, allowedTools);
-    
+
     // Should convert the entire sequence to a single AIMessage
     expect(result.messages).toHaveLength(2);
     expect(result.messages[0]).toBeInstanceOf(HumanMessage);
     expect(result.messages[1]).toBeInstanceOf(AIMessage);
-    
+
     // The content should include all parts
     expect(typeof result.messages[1].content).toBe('string');
     expect(result.messages[1].content).toContain('Let me check the weather first.');
@@ -231,24 +231,24 @@ describe('formatAgentMessages with tools parameter', () => {
         ],
       },
     ];
-    
+
     const indexTokenCountMap = {
       0: 10,  // 10 tokens for user message
       1: 40,  // 40 tokens for assistant message with tool call
     };
-    
+
     // Provide a set of allowed tools that doesn't include 'check_weather'
     const allowedTools = new Set(['search', 'calculator']);
-    
+
     const result = formatAgentMessages(payload, indexTokenCountMap, allowedTools);
-    
+
     // Should have 2 messages and 2 entries in the token count map
     expect(result.messages).toHaveLength(2);
     expect(Object.keys(result.indexTokenCountMap || {}).length).toBe(2);
-    
+
     // User message token count should be unchanged
     expect(result.indexTokenCountMap?.[0]).toBe(10);
-    
+
     // All assistant message tokens should be assigned to the single AIMessage
     expect(result.indexTokenCountMap?.[1]).toBe(40);
   });
@@ -315,27 +315,27 @@ describe('formatAgentMessages with tools parameter', () => {
       },
       { role: 'assistant', content: 'Here\'s your answer based on all that information.' },
     ];
-    
+
     // Allow search and calculator but not check_weather
     const allowedTools = new Set(['search', 'calculator']);
-    
+
     const result = formatAgentMessages(payload, undefined, allowedTools);
-    
+
     // Should have the user message, search tool sequence (2 messages),
     // a combined message for weather and calculator (since one has an invalid tool),
     // and final message
     expect(result.messages).toHaveLength(5);
-    
+
     // Check the types of messages
     expect(result.messages[0]).toBeInstanceOf(HumanMessage);
     expect(result.messages[1]).toBeInstanceOf(AIMessage);  // Search message
     expect(result.messages[2]).toBeInstanceOf(ToolMessage); // Search tool response
     expect(result.messages[3]).toBeInstanceOf(AIMessage); // Converted weather+calculator message
     expect(result.messages[4]).toBeInstanceOf(AIMessage); // Final message
-    
+
     // Check that the combined message was converted to a string
     expect(typeof result.messages[3].content).toBe('string');
-    
+
     // The format might vary based on the getBufferString implementation
     // but we should check that all the key information is present
     const content = result.messages[3].content as string;