test: add a test that check behavior for exit_agent tool

Author: Iru89

packages/botonic-plugin-flow-builder/src/types.ts

@@ -1,9 +1,4 @@
-import {
-  BotContext,
-  PluginPreRequest,
-  ResolvedPlugins,
-  Session,
-} from '@botonic/core'
+import { BotContext, PluginPreRequest, ResolvedPlugins } from '@botonic/core'
 
 import { HtFlowBuilderData } from './content-fields/hubtype-fields'
 
@@ -92,19 +87,18 @@ export interface AiAgentResponse {
   message: AgentMessage
 }
 
-interface AssistantMessage {
+export type AgentMessage = AssistantMessage | ToolMessage
+export interface AssistantMessage {
   role: 'assistant'
   content: string
 }
 
-interface ToolMessage {
+export interface ToolMessage {
   role: 'tool'
   tool_name: string
   tool_params?: Record<string, unknown>
 }
 
-export type AgentMessage = AssistantMessage | ToolMessage
-
 export interface SmartIntentResponse {
   data: {
     smart_intent_title: string

packages/botonic-plugin-flow-builder/tests/__mocks__/ai-agent.ts

@@ -1,8 +1,6 @@
-interface MockAiAgentOptions {
-  message: { role: string; content: string }
-}
+import { AiAgentResponse } from '../../src/types'
 
-export function mockAiAgentResponse({ message }: MockAiAgentOptions) {
+export function mockAiAgentResponse({ message }: AiAgentResponse) {
   return jest.fn(() => {
     return Promise.resolve({ message })
   })

packages/botonic-plugin-flow-builder/tests/ai-agent.test.ts

@@ -59,4 +59,26 @@ describe('Check the contents returned by the plugin when it use an ai agent', ()
       'I can provide you with information about current temperatures, forecasts, and the probability of rain for your location. Just let me know where you are or where you’re interested in, and I’ll give you the details!'
     )
   })
+
+  test('When agent response is of role tool and name exit_agent, the plugin should respond with the tool response of fallback flow', async () => {
+    const { contents } = await createFlowBuilderPluginAndGetContents({
+      flowBuilderOptions: {
+        flow: aiAgentTestFlow,
+        getAiAgentResponse: mockAiAgentResponse({
+          message: {
+            role: 'tool',
+            tool_name: 'exit_agent',
+          },
+        }),
+      },
+      requestArgs: {
+        input: {
+          data: 'Who has won the league this year?',
+          type: INPUT.TEXT,
+        },
+      },
+    })
+
+    expect((contents[0] as FlowText).text).toEqual('Fallback')
+  })
 })