feat: when agent responds with a exit_tool plugin follow with knowledge base or fallback

Author: Iru89

packages/botonic-plugin-ai-agents/src/types.ts

@@ -19,7 +19,7 @@ export interface AiAgentRequestDataTest {
   instructions: string
 }
 
-export type MessageRole = 'user' | 'assistant'
+export type MessageRole = 'user' | 'assistant' | 'tool'
 
 export interface AiAgentRequestData {
   message: string
@@ -29,5 +29,23 @@ export interface AiAgentRequestData {
 }
 
 export interface AiAgentResponse {
-  message: { role: string; content: string }
+  message: AgentMessage
 }
+
+export interface AssistantMessage {
+  role: 'assistant'
+  content: string
+}
+
+export interface UserMessage {
+  role: 'user'
+  content: string
+}
+
+export interface ToolMessage {
+  role: 'tool'
+  tool_name: string
+  tool_params?: Record<string, unknown>
+}
+
+export type AgentMessage = AssistantMessage | ToolMessage

packages/botonic-plugin-flow-builder/src/action/ai-agent.ts

@@ -42,8 +42,18 @@ function updateContentsWithAiAgentResponse(
   contents: FlowContent[],
   aiAgentResponse: AiAgentResponse
 ): FlowContent[] {
+  if (
+    aiAgentResponse.message.role === 'tool' &&
+    aiAgentResponse.message.tool_name === 'exit_agent'
+  ) {
+    return []
+  }
+
   return contents.map(content => {
-    if (content instanceof FlowAiAgent) {
+    if (
+      content instanceof FlowAiAgent &&
+      aiAgentResponse.message.role === 'assistant'
+    ) {
       content.text = aiAgentResponse.message.content
     }
 

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

@@ -89,9 +89,22 @@ export interface KnowledgeBaseResponse {
 }
 
 export interface AiAgentResponse {
-  message: { role: string; content: string }
+  message: AgentMessage
 }
 
+interface AssistantMessage {
+  role: 'assistant'
+  content: string
+}
+
+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