@botonic/plugin-ai-agents 0.37.0 → 0.37.2-alpha.0

package/src/runner.ts

@@ -1,10 +1,15 @@
-import { Runner } from '@openai/agents'
+import {
+  InputGuardrailTripwireTriggered,
+  Runner,
+  RunToolCallItem,
+} from '@openai/agents'
 
 import {
   AgenticInputMessage,
   AgenticOutputMessage,
   AIAgent,
   Context,
+  RunResult,
 } from './types'
 
 export class AIAgentRunner {
@@ -18,7 +23,7 @@ export class AIAgentRunner {
     messages: AgenticInputMessage[],
     context: Context,
     maxRetries: number = 1
-  ): Promise<AgenticOutputMessage[]> {
+  ): Promise<RunResult> {
     return this.runWithRetry(messages, context, maxRetries, 1)
   }
 
@@ -27,18 +32,58 @@ export class AIAgentRunner {
     context: Context,
     maxRetries: number,
     attempt: number
-  ): Promise<AgenticOutputMessage[]> {
+  ): Promise<RunResult> {
     try {
       const runner = new Runner({
         modelSettings: { temperature: 0 },
       })
       const result = await runner.run(this.agent, messages, { context })
-      return result.finalOutput?.messages || []
+
+      const outputMessages = result.finalOutput?.messages || []
+      const hasExit =
+        outputMessages.length === 0 ||
+        outputMessages.some(message => message.type === 'exit')
+      const toolsExecuted = this.getExecutedNameTools(result)
+
+      return {
+        messages: hasExit
+          ? []
+          : (outputMessages.filter(
+              message => message.type !== 'exit'
+            ) as AgenticOutputMessage[]),
+        toolsExecuted,
+        exit: hasExit,
+        inputGuardrailTriggered: [],
+        outputGuardrailTriggered: [],
+      }
     } catch (error) {
+      if (error instanceof InputGuardrailTripwireTriggered) {
+        return {
+          messages: [],
+          toolsExecuted: [],
+          exit: true,
+          inputGuardrailTriggered: error.result.output.outputInfo,
+          outputGuardrailTriggered: [],
+        }
+      }
       if (attempt > maxRetries) {
         throw error
       }
       return this.runWithRetry(messages, context, maxRetries, attempt + 1)
     }
   }
+
+  private getExecutedNameTools(result) {
+    return (
+      result.newItems
+        ?.filter(item => item instanceof RunToolCallItem)
+        .map((item: RunToolCallItem) => {
+          if (item.rawItem.type === 'function_call') {
+            return item.rawItem.name
+          }
+          return ''
+        })
+        .filter((toolName: string) => toolName !== '') || []
+    )
+  }
 }

package/src/structured-output/carousel.ts

@@ -1,20 +1,7 @@
+import { CarouselMessage } from '@botonic/core'
 import { z } from 'zod'
 
-import { BaseMessage } from './shared'
-
-interface CarouselElement {
-  title: string
-  subtitle: string
-  image: string
-  button: { text: string; url: string }
-}
-
-export interface CarouselMessage extends BaseMessage {
-  type: 'carousel'
-  content: {
-    elements: CarouselElement[]
-  }
-}
+export type { CarouselMessage }
 
 export const CarouselSchema = z
   .object({

package/src/structured-output/exit.ts

@@ -1,10 +1,7 @@
+import { ExitMessage } from '@botonic/core'
 import z from 'zod'
 
-import { BaseMessage } from './shared'
-
-export interface ExitMessage extends BaseMessage {
-  type: 'exit'
-}
+export type { ExitMessage }
 
 export const ExitSchema = z
   .object({

package/src/structured-output/index.ts

@@ -1,18 +1,10 @@
+import { OutputMessage } from '@botonic/core'
 import { z } from 'zod'
 
-import { CarouselMessage, CarouselSchema } from './carousel'
-import { ExitMessage, ExitSchema } from './exit'
-import { TextMessage, TextSchema } from './text'
-import {
-  TextWithButtonsMessage,
-  TextWithButtonsSchema,
-} from './text-with-buttons'
-
-export type OutputMessage =
-  | TextMessage
-  | TextWithButtonsMessage
-  | CarouselMessage
-  | ExitMessage
+import { CarouselSchema } from './carousel'
+import { ExitSchema } from './exit'
+import { TextSchema } from './text'
+import { TextWithButtonsSchema } from './text-with-buttons'
 
 export interface Output {
   messages: OutputMessage[]

package/src/structured-output/text-with-buttons.ts

@@ -1,14 +1,7 @@
+import { TextWithButtonsMessage } from '@botonic/core'
 import z from 'zod'
 
-import { BaseMessage } from './shared'
-
-export interface TextWithButtonsMessage extends BaseMessage {
-  type: 'textWithButtons'
-  content: {
-    text: string
-    buttons: string[]
-  }
-}
+export type { TextWithButtonsMessage }
 
 export const TextWithButtonsSchema = z
   .object({

package/src/structured-output/text.ts

@@ -1,13 +1,7 @@
+import { TextMessage } from '@botonic/core'
 import { z } from 'zod'
 
-import { BaseMessage } from './shared'
-
-export interface TextMessage extends BaseMessage {
-  type: 'text'
-  content: {
-    text: string
-  }
-}
+export type { TextMessage }
 
 export const TextSchema = z
   .object({

package/src/types.ts

@@ -1,27 +1,40 @@
+import {
+  AgenticOutputMessage,
+  InferenceResponse,
+  RunResult,
+} from '@botonic/core'
 import {
   Agent,
   AgentInputItem,
-  RunContext,
+  RunContext as OpenAIRunContext,
   Tool as OpenAITool,
 } from '@openai/agents'
 import { ZodSchema } from 'zod'
 
-import { OutputMessage, OutputSchema } from './structured-output'
+import { OutputSchema } from './structured-output'
 
 export interface Context {
   authToken: string
 }
 
+export type RunContext = OpenAIRunContext<Context>
+
 export interface CustomTool {
   name: string
   description: string
   schema: ZodSchema
-  func: (input?: any, runContext?: RunContext<Context>) => Promise<any>
+  func: (input?: any, runContext?: RunContext) => Promise<any>
 }
 
+export interface GuardrailRule {
+  name: string
+  description: string
+}
+
+export type ContactInfo = Record<string, string>
+
 export type Tool = OpenAITool<Context>
 export type AIAgent = Agent<Context, typeof OutputSchema>
-
 export interface PluginAiAgentOptions {
   authToken?: string
   customTools?: CustomTool[]
@@ -31,7 +44,9 @@ export interface AiAgentArgs {
   name: string
   instructions: string
   activeTools?: { name: string }[]
+  inputGuardrailRules?: GuardrailRule[]
 }
 
 export type AgenticInputMessage = AgentInputItem
-export type AgenticOutputMessage = OutputMessage
+
+export type { AgenticOutputMessage, InferenceResponse, RunResult }

package/lib/cjs/structured-output/shared.d.ts

@@ -1,3 +0,0 @@
-export interface BaseMessage {
-    type: 'text' | 'textWithButtons' | 'carousel' | 'exit';
-}

package/lib/cjs/structured-output/shared.js

@@ -1,3 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-//# sourceMappingURL=shared.js.map

package/lib/cjs/structured-output/shared.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"shared.js","sourceRoot":"","sources":["../../../src/structured-output/shared.ts"],"names":[],"mappings":""}

package/lib/esm/structured-output/shared.d.ts

@@ -1,3 +0,0 @@
-export interface BaseMessage {
-    type: 'text' | 'textWithButtons' | 'carousel' | 'exit';
-}

package/lib/esm/structured-output/shared.js

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=shared.js.map

package/lib/esm/structured-output/shared.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"shared.js","sourceRoot":"","sources":["../../../src/structured-output/shared.ts"],"names":[],"mappings":""}

package/src/structured-output/shared.ts

@@ -1,3 +0,0 @@
-export interface BaseMessage {
-  type: 'text' | 'textWithButtons' | 'carousel' | 'exit'
-}