@botonic/plugin-ai-agents 0.46.1 → 2.23.0

package/lib/esm/structured-output/text-with-buttons.js

@@ -1,24 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.TextWithButtonsSchema = void 0;
-const tslib_1 = require("tslib");
-const zod_1 = tslib_1.__importDefault(require("zod"));
-exports.TextWithButtonsSchema = zod_1.default
-    .object({
-    type: zod_1.default.enum(['textWithButtons']),
-    content: zod_1.default.object({
-        text: zod_1.default.string(),
-        buttons: zod_1.default.array(zod_1.default.object({
-            text: zod_1.default.string(),
-            url: zod_1.default.string().nullable().optional(),
-            target: zod_1.default
-                .enum(['_blank', '_self'])
-                .default('_blank')
-                .nullable()
-                .optional()
-                .describe('The target of the button when it has an url. If not provided, it will default to _blank.'),
-        })),
-    }),
-})
-    .describe('A text message with buttons to allow the user to use quick replies');
-//# sourceMappingURL=text-with-buttons.js.map

package/lib/esm/structured-output/text-with-buttons.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"text-with-buttons.js","sourceRoot":"","sources":["../../../src/structured-output/text-with-buttons.ts"],"names":[],"mappings":";;;;AACA,sDAAmB;AAIN,QAAA,qBAAqB,GAAG,aAAC;KACnC,MAAM,CAAC;IACN,IAAI,EAAE,aAAC,CAAC,IAAI,CAAC,CAAC,iBAAiB,CAAC,CAAC;IACjC,OAAO,EAAE,aAAC,CAAC,MAAM,CAAC;QAChB,IAAI,EAAE,aAAC,CAAC,MAAM,EAAE;QAChB,OAAO,EAAE,aAAC,CAAC,KAAK,CACd,aAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,aAAC,CAAC,MAAM,EAAE;YAChB,GAAG,EAAE,aAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;YACrC,MAAM,EAAE,aAAC;iBACN,IAAI,CAAC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;iBACzB,OAAO,CAAC,QAAQ,CAAC;iBACjB,QAAQ,EAAE;iBACV,QAAQ,EAAE;iBACV,QAAQ,CACP,0FAA0F,CAC3F;SACJ,CAAC,CACH;KACF,CAAC;CACH,CAAC;KACD,QAAQ,CACP,oEAAoE,CACrE,CAAA"}

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

@@ -1,13 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.TextSchema = void 0;
-const zod_1 = require("zod");
-exports.TextSchema = zod_1.z
-    .object({
-    type: zod_1.z.enum(['text']),
-    content: zod_1.z.object({
-        text: zod_1.z.string(),
-    }),
-})
-    .describe('A text message');
-//# sourceMappingURL=text.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"text.js","sourceRoot":"","sources":["../../../src/structured-output/text.ts"],"names":[],"mappings":";;;AACA,6BAAuB;AAIV,QAAA,UAAU,GAAG,OAAC;KACxB,MAAM,CAAC;IACN,IAAI,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC;IACtB,OAAO,EAAE,OAAC,CAAC,MAAM,CAAC;QAChB,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;KACjB,CAAC;CACH,CAAC;KACD,QAAQ,CAAC,gBAAgB,CAAC,CAAA"}

package/lib/esm/tools/index.js

@@ -1,7 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.mandatoryTools = exports.retrieveKnowledge = void 0;
-var retrieve_knowledge_1 = require("./retrieve-knowledge");
-Object.defineProperty(exports, "retrieveKnowledge", { enumerable: true, get: function () { return retrieve_knowledge_1.retrieveKnowledge; } });
-exports.mandatoryTools = [];
-//# sourceMappingURL=index.js.map

package/lib/esm/tools/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/tools/index.ts"],"names":[],"mappings":";;;AAEA,2DAAwD;AAA/C,uHAAA,iBAAiB,OAAA;AAEb,QAAA,cAAc,GAAW,EAAE,CAAA"}

package/lib/esm/tools/retrieve-knowledge.d.ts

@@ -1,9 +0,0 @@
-import { z } from 'zod';
-import type { Context } from '../types';
-export declare const retrieveKnowledge: import("@openai/agents").FunctionTool<Context<import("@botonic/core").ResolvedPlugins, any>, z.ZodObject<{
-    query: z.ZodString;
-}, "strip", z.ZodTypeAny, {
-    query: string;
-}, {
-    query: string;
-}>, string>;

package/lib/esm/tools/retrieve-knowledge.js

@@ -1,32 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.retrieveKnowledge = void 0;
-const agents_1 = require("@openai/agents");
-const zod_1 = require("zod");
-const hubtype_api_client_1 = require("../services/hubtype-api-client");
-exports.retrieveKnowledge = (0, agents_1.tool)({
-    name: 'retrieve_knowledge',
-    description: 'Consult the knowledge base for information before answering. Use this tool to make sure the information you provide is faithful.',
-    parameters: zod_1.z.object({
-        query: zod_1.z.string().describe('The query to search the knowledge base for'),
-    }),
-    execute: async (input, runContext) => {
-        const context = runContext?.context;
-        const query = input.query;
-        if (!context) {
-            throw new Error('Context is required');
-        }
-        const sourceIds = context.sourceIds;
-        const client = new hubtype_api_client_1.HubtypeApiClient(context.authToken);
-        const chunks = await client.retrieveSimilarChunks(query, sourceIds);
-        const chunkTexts = chunks.map(chunk => chunk.text);
-        context.knowledgeUsed = {
-            query,
-            sourceIds,
-            chunksIds: chunks.map(chunk => chunk.id),
-            chunkTexts,
-        };
-        return chunkTexts;
-    },
-});
-//# sourceMappingURL=retrieve-knowledge.js.map

package/lib/esm/tools/retrieve-knowledge.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"retrieve-knowledge.js","sourceRoot":"","sources":["../../../src/tools/retrieve-knowledge.ts"],"names":[],"mappings":";;;AAAA,2CAAsD;AACtD,6BAAuB;AAEvB,uEAAiE;AAGpD,QAAA,iBAAiB,GAAG,IAAA,aAAI,EAAC;IACpC,IAAI,EAAE,oBAAoB;IAC1B,WAAW,EACT,kIAAkI;IACpI,UAAU,EAAE,OAAC,CAAC,MAAM,CAAC;QACnB,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;KACzE,CAAC;IACF,OAAO,EAAE,KAAK,EACZ,KAAwB,EACxB,UAAgC,EACb,EAAE;QACrB,MAAM,OAAO,GAAG,UAAU,EAAE,OAAO,CAAA;QACnC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAA;QACzB,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAA;QACxC,CAAC;QACD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAA;QACnC,MAAM,MAAM,GAAG,IAAI,qCAAgB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QACtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAA;QACnE,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAElD,OAAO,CAAC,aAAa,GAAG;YACtB,KAAK;YACL,SAAS;YACT,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;YACxC,UAAU;SACX,CAAA;QAED,OAAO,UAAU,CAAA;IACnB,CAAC;CACF,CAAC,CAAA"}

package/lib/esm/types.js

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

package/lib/esm/types.js.map

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

package/src/agent-builder.ts

@@ -1,183 +0,0 @@
-import type { CampaignV2, ContactInfo, ResolvedPlugins } from '@botonic/core'
-import {
-  Agent,
-  type AgentOutputType,
-  type InputGuardrail,
-} from '@openai/agents'
-import type { z } from 'zod'
-
-import { OPENAI_PROVIDER } from './constants'
-import type { DebugLogger } from './debug-logger'
-import { createInputGuardrail } from './guardrails'
-import type { GuardrailTrackingContext } from './guardrails/input'
-import type { LLMConfig } from './llm-config'
-import { getOutputSchema, type OutputSchema } from './structured-output'
-import { mandatoryTools, retrieveKnowledge } from './tools'
-import type { AIAgent, Context, GuardrailRule, Tool } from './types'
-
-interface AIAgentBuilderOptions<
-  TPlugins extends ResolvedPlugins = ResolvedPlugins,
-  TExtraData = any,
-> {
-  name: string
-  instructions: string
-  tools: Tool<TPlugins, TExtraData>[]
-  campaignsContext?: CampaignV2[]
-  contactInfo: ContactInfo[]
-  inputGuardrailRules: GuardrailRule[]
-  sourceIds: string[]
-  outputMessagesSchemas?: z.ZodObject<any>[]
-  llmConfig: LLMConfig
-  logger: DebugLogger
-  guardrailTrackingContext: GuardrailTrackingContext
-}
-
-export class AIAgentBuilder<
-  TPlugins extends ResolvedPlugins = ResolvedPlugins,
-  TExtraData = any,
-> {
-  private name: string
-  private instructions: string
-  private tools: Tool<TPlugins, TExtraData>[]
-  private externalOutputMessagesSchemas: z.ZodObject<any>[]
-  private inputGuardrails: InputGuardrail[]
-  public llmConfig: LLMConfig
-  private logger: DebugLogger
-
-  constructor(options: AIAgentBuilderOptions<TPlugins, TExtraData>) {
-    this.name = options.name
-    this.instructions = this.addExtraInstructions(
-      options.instructions,
-      options.contactInfo,
-      options.campaignsContext
-    )
-    this.tools = this.addHubtypeTools(options.tools, options.sourceIds)
-    this.externalOutputMessagesSchemas = options.outputMessagesSchemas || []
-    this.inputGuardrails = []
-    this.llmConfig = options.llmConfig
-    this.logger = options.logger
-    if (options.inputGuardrailRules.length > 0) {
-      const inputGuardrail = createInputGuardrail(
-        options.inputGuardrailRules,
-        options.llmConfig,
-        options.guardrailTrackingContext
-      )
-      this.inputGuardrails.push(inputGuardrail)
-    }
-  }
-
-  build(): AIAgent<TPlugins, TExtraData> {
-    // When using standard OpenAI API, we need to specify the model
-    // Azure OpenAI uses deployment name instead
-
-    const model = this.llmConfig.modelName
-    const hasRetrieveKnowledge = this.tools.includes(retrieveKnowledge)
-
-    this.logger.logModelSettings({
-      provider: OPENAI_PROVIDER,
-      model,
-      reasoning: this.llmConfig.modelSettings.reasoning as
-        | { effort: string }
-        | undefined,
-      text: this.llmConfig.modelSettings.text as
-        | { verbosity: string }
-        | undefined,
-      toolChoice: this.llmConfig.modelSettings.toolChoice as string | undefined,
-      hasRetrieveKnowledge,
-    })
-
-    return new Agent<
-      Context<TPlugins, TExtraData>,
-      AgentOutputType<typeof OutputSchema>
-    >({
-      name: this.name,
-      model,
-      instructions: this.instructions,
-      tools: this.tools,
-      outputType: getOutputSchema(this.externalOutputMessagesSchemas),
-      inputGuardrails: this.inputGuardrails,
-      outputGuardrails: [],
-    })
-  }
-
-  private addExtraInstructions(
-    initialInstructions: string,
-    contactInfo: ContactInfo[],
-    campaignsContext?: CampaignV2[]
-  ): string {
-    const instructions = `<instructions>\n${initialInstructions.trim()}\n</instructions>`
-    const metadataInstructions = this.getMetadataInstructions()
-    const contactInfoInstructions = this.getContactInfoInstructions(contactInfo)
-    const campaignInstructions = this.getCampaignInstructions(campaignsContext)
-    const outputInstructions = this.getOutputInstructions()
-    return `${instructions}\n\n${metadataInstructions}\n\n${contactInfoInstructions}\n\n${campaignInstructions}\n\n${outputInstructions}`
-  }
-
-  private getContactInfoInstructions(contactInfo: ContactInfo[]): string {
-    const structuredContactInfo = contactInfo
-      .map(
-        info =>
-          ` <contact_info>
-              <name>${info.name}</name>
-              <value>${info.value}</value>
-              <type>${info.type}</type>
-              ${
-                info.description
-                  ? `<description>${info.description}</description>`
-                  : ''
-              }
-            </contact_info>`
-      )
-      .join('\n')
-    return `<contact_info_fields>\n${structuredContactInfo}</contact_info_fields>`
-  }
-
-  private getMetadataInstructions(): string {
-    const metadata = `Current Date: ${new Date().toISOString()}`
-    return `<metadata>\n${metadata}\n</metadata>`
-  }
-
-  private getCampaignInstructions(campaignsContext?: CampaignV2[]): string {
-    if (!campaignsContext || campaignsContext.length === 0) {
-      return ''
-    }
-    const campaignsWithContext = campaignsContext.filter(
-      campaign => campaign.agent_context
-    )
-    if (campaignsWithContext.length === 0) {
-      return ''
-    }
-    return campaignsWithContext
-      .map(
-        (campaign, index) =>
-          `<campaign_context_${index + 1}>\n${campaign.agent_context}\n</campaign_context_${index + 1}>`
-      )
-      .join('\n')
-  }
-
-  private getOutputInstructions(): string {
-    const example = {
-      messages: [
-        {
-          type: 'text',
-          content: {
-            text: 'Hello, how can I help you today?',
-          },
-        },
-      ],
-    }
-    const output = `Return a JSON that follows the output schema provided. Never return multiple output schemas concatenated by a line break.\n<example>\n${JSON.stringify(example)}\n</example>`
-    return `<output>\n${output}\n</output>`
-  }
-
-  private addHubtypeTools(
-    tools: Tool<TPlugins, TExtraData>[],
-    sourceIds: string[]
-  ): Tool<TPlugins, TExtraData>[] {
-    const hubtypeTools: Tool[] = [...mandatoryTools]
-    if (sourceIds.length > 0) {
-      hubtypeTools.push(retrieveKnowledge)
-    }
-    return [...hubtypeTools, ...tools]
-  }
-}

package/src/bot-config-tools.ts

@@ -1,21 +0,0 @@
-import type { ToolConfigJSON } from '@botonic/core'
-import { zodToJsonSchema } from 'zod-to-json-schema'
-
-import type { CustomTool } from './types'
-
-/**
- * Maps custom AI agent tools to the format expected by bot config (e.g. for flow-builder).
- * Converts each tool's Zod schema to JSON Schema so it can be serialized in the config.
- */
-export function getToolsForBotConfig(
-  customTools: CustomTool[]
-): ToolConfigJSON[] {
-  return customTools.map(tool => ({
-    name: tool.name,
-    description: tool.description,
-    // Cast to avoid TS "Type instantiation is excessively deep" with zodToJsonSchema + ZodSchema
-    schema: zodToJsonSchema(tool.schema as never, {
-      $refStrategy: 'none',
-    }) as ToolConfigJSON['schema'],
-  }))
-}

package/src/constants.ts

@@ -1,21 +0,0 @@
-export const HUBTYPE_API_URL =
-  process.env.HUBTYPE_API_URL || 'https://api.hubtype.com'
-
-// OpenAI Provider Configuration
-export const OPENAI_API_KEY = process.env.OPENAI_API_KEY // pragma: allowlist secret
-export const OPENAI_MODEL = process.env.OPENAI_MODEL || 'gpt-4.1-mini'
-export const OPENAI_PROVIDER: 'openai' | 'azure' =
-  (process.env.OPENAI_PROVIDER as 'openai' | 'azure') || 'azure'
-
-// Azure OpenAI Configuration
-export const AZURE_OPENAI_API_KEY = process.env.AZURE_OPENAI_API_KEY // pragma: allowlist secret
-export const AZURE_OPENAI_API_BASE = process.env.AZURE_OPENAI_API_BASE
-export const AZURE_OPENAI_API_VERSION =
-  process.env.AZURE_OPENAI_API_VERSION || '2025-01-01-preview'
-
-export const isProd = process.env.NODE_ENV === 'production'
-
-export const MAX_MEMORY_LENGTH = 25
-
-export const DEFAULT_TIMEOUT_16_SECONDS = 16000
-export const DEFAULT_MAX_RETRIES = 2

package/src/debug-logger.ts

@@ -1,166 +0,0 @@
-import type { AiAgentArgs, ToolExecution } from '@botonic/core'
-import type { ModelSettings } from '@openai/agents'
-import { MAX_MEMORY_LENGTH, OPENAI_PROVIDER } from './constants'
-import type { AgenticInputMessage, MemoryOptions, RunResult } from './types'
-
-const PREFIX = '[BotonicPluginAiAgents]'
-
-export interface DebugLoggerConfig {
-  messageHistoryApiVersion: string
-  maxRetries: number
-  timeout: number
-  customToolNames: string[]
-  memory: MemoryOptions
-}
-
-export interface ModelSettingsInfo {
-  provider: string
-  model: string | undefined
-  reasoning?: { effort: string }
-  text?: { verbosity: string }
-  toolChoice?: string
-  hasRetrieveKnowledge: boolean
-}
-
-export interface DebugLogger {
-  logInitialConfig(config: DebugLoggerConfig): void
-  logAgentDebugInfo(
-    aiAgentArgs: AiAgentArgs,
-    toolNames: string[],
-    messages: AgenticInputMessage[]
-  ): void
-  logModelSettings(settings: ModelSettingsInfo): void
-  logRunnerStart(model: string, modelSettings: ModelSettings): void
-  logRunResult(runResult: RunResult, startTime: number): void
-  logGuardrailTriggered(): void
-  logRunnerError(startTime: number, error: unknown): void
-  logToolExecution(toolExecution: ToolExecution): void
-}
-
-class EnabledDebugLogger implements DebugLogger {
-  logInitialConfig(config: DebugLoggerConfig): void {
-    console.log(`${PREFIX} === Plugin Initialization ===`)
-    console.log(`${PREFIX} Provider: ${OPENAI_PROVIDER}`)
-    console.log(
-      `${PREFIX} Message History API Version: ${config.messageHistoryApiVersion}`
-    )
-    console.log(`${PREFIX} Max Retries: ${config.maxRetries}`)
-    console.log(`${PREFIX} Timeout: ${config.timeout}ms`)
-    console.log(
-      `${PREFIX} Custom Tools Registered: ${config.customToolNames.join(', ') || 'none'}`
-    )
-    console.log(`${PREFIX} Memory Options:`, {
-      maxMessages: config.memory.maxMessages ?? MAX_MEMORY_LENGTH,
-      includeToolCalls: config.memory.includeToolCalls ?? true,
-      maxFullToolResults: config.memory.maxFullToolResults ?? 1,
-      debugMode: config.memory.debugMode ?? false,
-    })
-    console.log(`${PREFIX} === End Plugin Initialization ===`)
-  }
-
-  logAgentDebugInfo(
-    aiAgentArgs: AiAgentArgs,
-    toolNames: string[],
-    messages: AgenticInputMessage[]
-  ): void {
-    console.log(`${PREFIX} === AI Agent Debug Info ===`)
-    console.log(`${PREFIX} Agent Name: ${aiAgentArgs.name}`)
-    console.log(`${PREFIX} Active Tools: ${JSON.stringify(toolNames)}`)
-    console.log(
-      `${PREFIX} Source IDs: ${JSON.stringify(aiAgentArgs.sourceIds || [])}`
-    )
-    console.log(`${PREFIX} Message History Count: ${messages.length}`)
-    console.log(
-      `${PREFIX} Input Guardrail Rules: ${aiAgentArgs.inputGuardrailRules?.length || 0}`
-    )
-    console.log(`${PREFIX} Instructions:`)
-    console.log(aiAgentArgs.instructions)
-    console.log(`${PREFIX} === End Debug Info ===`)
-  }
-
-  logModelSettings(settings: ModelSettingsInfo): void {
-    console.log(`${PREFIX} === Agent Model Settings ===`)
-    console.log(
-      `${PREFIX} Has Retrieve Knowledge Tool: ${settings.hasRetrieveKnowledge}`
-    )
-    if (settings.reasoning) {
-      console.log(`${PREFIX} Reasoning Effort: ${settings.reasoning.effort}`)
-    }
-    if (settings.text) {
-      console.log(`${PREFIX} Text Verbosity: ${settings.text.verbosity}`)
-    }
-    if (settings.toolChoice) {
-      console.log(`${PREFIX} Tool Choice: ${settings.toolChoice}`)
-    }
-    console.log(`${PREFIX} === End Model Settings ===`)
-  }
-
-  logRunnerStart(model: string, modelSettings: ModelSettings): void {
-    console.log(`${PREFIX} === Runner Execution Start ===`)
-    console.log(`${PREFIX} Model: ${model}`)
-    console.log(`${PREFIX} Model Settings: ${JSON.stringify(modelSettings)}`)
-  }
-
-  logRunResult(runResult: RunResult, startTime: number): void {
-    const elapsedMs = Date.now() - startTime
-
-    console.log(`${PREFIX} === Runner Execution Complete ===`)
-    console.log(`${PREFIX} Execution Time: ${elapsedMs}ms`)
-    console.log(`${PREFIX} Output Messages Count: ${runResult.messages.length}`)
-    console.log(`${PREFIX} Exit: ${runResult.exit}`)
-    console.log(`${PREFIX} Error: ${runResult.error}`)
-    console.log(
-      `${PREFIX} Tools Executed: ${JSON.stringify(runResult.toolsExecuted.map(t => t.toolName))}`
-    )
-    if (runResult.inputGuardrailsTriggered.length > 0) {
-      console.log(
-        `${PREFIX} Input Guardrails Triggered: ${runResult.inputGuardrailsTriggered.length}`
-      )
-    }
-    if (runResult.outputGuardrailsTriggered.length > 0) {
-      console.log(
-        `${PREFIX} Output Guardrails Triggered: ${runResult.outputGuardrailsTriggered.length}`
-      )
-    }
-    console.log(`${PREFIX} === End Runner Execution ===`)
-  }
-
-  logGuardrailTriggered(): void {
-    console.log(`${PREFIX} Input guardrail triggered`)
-  }
-
-  logRunnerError(startTime: number, error: unknown): void {
-    const elapsedMs = Date.now() - startTime
-    console.log(`${PREFIX} Runner execution failed after ${elapsedMs}ms`)
-    console.log(`${PREFIX} Error:`, error)
-  }
-
-  logToolExecution(toolExecution: ToolExecution): void {
-    console.log(`${PREFIX} Tool Execution: ${toolExecution.toolName}`)
-    console.log(
-      `${PREFIX} Tool Arguments: ${JSON.stringify(toolExecution.toolArguments)}`
-    )
-    console.log(`${PREFIX} Tool Results: ${toolExecution.toolResults}`)
-    console.log(
-      `${PREFIX} Knowledgebase Sources IDs: ${JSON.stringify(toolExecution.knowledgebaseSourcesIds)}`
-    )
-    console.log(
-      `${PREFIX} Knowledgebase Chunks IDs: ${JSON.stringify(toolExecution.knowledgebaseChunksIds)}`
-    )
-  }
-}
-
-class DisabledDebugLogger implements DebugLogger {
-  logInitialConfig(): void {}
-  logAgentDebugInfo(): void {}
-  logModelSettings(): void {}
-  logRunnerStart(_model: string, _modelSettings: ModelSettings): void {}
-  logRunResult(): void {}
-  logGuardrailTriggered(): void {}
-  logRunnerError(): void {}
-  logToolExecution(): void {}
-}
-
-export function createDebugLogger(enableDebug: boolean): DebugLogger {
-  return enableDebug ? new EnabledDebugLogger() : new DisabledDebugLogger()
-}

package/src/guardrails/index.ts

@@ -1 +0,0 @@
-export * from './input'

package/src/guardrails/input.ts

@@ -1,122 +0,0 @@
-import {
-  Agent,
-  type InputGuardrail,
-  Runner,
-  type UserMessageItem,
-} from '@openai/agents'
-import { z } from 'zod'
-import { AZURE_OPENAI_API_VERSION, isProd, OPENAI_PROVIDER } from '../constants'
-import type { LLMConfig } from '../llm-config'
-import { HubtypeApiClient } from '../services/hubtype-api-client'
-import { TrackFeature, TrackProductName } from '../services/types'
-import type { GuardrailRule, ResultRawResponse } from '../types'
-
-export interface GuardrailTrackingContext {
-  botId: string
-  isTest: boolean
-  authToken: string
-  inferenceId: string
-}
-
-export function createInputGuardrail(
-  rules: GuardrailRule[],
-  llmConfig: LLMConfig,
-  trackingContext: GuardrailTrackingContext
-): InputGuardrail {
-  const outputType = z.object(
-    Object.fromEntries(
-      rules.map(rule => [rule.name, z.boolean().describe(rule.description)])
-    )
-  )
-
-  const agent = new Agent({
-    name: 'InputGuardrail',
-    instructions:
-      'Check if the user triggers some of the following guardrails.',
-    outputType,
-  })
-
-  return {
-    name: 'InputGuardrail',
-    execute: async ({ input, context }) => {
-      const lastMessage = input[input.length - 1] as UserMessageItem
-      const modelProvider = llmConfig.modelProvider
-      const modelSettings = llmConfig.modelSettings
-      modelSettings.toolChoice = undefined
-      const runner = new Runner({
-        modelSettings,
-        modelProvider,
-        tracingDisabled: true,
-      })
-      const startTime = Date.now()
-      const result = await runner.run(agent, [lastMessage], { context })
-      const endTime = Date.now()
-
-      void sendGuardrailLlmRunTracking(
-        result,
-        trackingContext,
-        llmConfig,
-        startTime,
-        endTime
-      )
-
-      const finalOutput = result.finalOutput as Record<string, boolean>
-      if (finalOutput === undefined) {
-        throw new Error('Guardrail agent failed to produce output')
-      }
-      const triggered = Object.values(finalOutput).some(value => value === true)
-      const triggeredGuardrails = Object.keys(finalOutput).filter(
-        key => finalOutput[key] === true
-      )
-      return {
-        outputInfo: triggeredGuardrails,
-        tripwireTriggered: triggered,
-      }
-    },
-  }
-}
-
-async function sendGuardrailLlmRunTracking(
-  result: {
-    rawResponses?: ResultRawResponse[]
-  },
-  trackingContext: GuardrailTrackingContext,
-  llmConfig: LLMConfig,
-  startTime: number,
-  endTime: number
-): Promise<void> {
-  if (!isProd) {
-    return
-  }
-  const rawResponses = result.rawResponses ?? []
-  if (rawResponses.length === 0) {
-    return
-  }
-  const totalDuration = endTime - startTime
-  const durationPerCall = Math.round(totalDuration / rawResponses.length)
-  const temperature =
-    (llmConfig.modelSettings.temperature as number | undefined) ?? 0
-  const apiVersion = OPENAI_PROVIDER === 'azure' ? AZURE_OPENAI_API_VERSION : ''
-
-  const llmRuns = rawResponses.map(response => ({
-    inference_id: trackingContext.inferenceId,
-    is_test: trackingContext.isTest,
-    product_name: TrackProductName.AI_AGENT,
-    deployment_name: llmConfig.modelName,
-    model_name:
-      (response.providerData?.['model'] as string | undefined) ??
-      llmConfig.modelName,
-    feature: TrackFeature.AI_AGENT_GUARDRAIL,
-    api_version: apiVersion,
-    num_prompt_tokens: response.usage.inputTokens,
-    num_completion_tokens: response.usage.outputTokens,
-    duration_in_milliseconds: durationPerCall,
-    temperature,
-    error: null,
-  }))
-
-  const client = new HubtypeApiClient(trackingContext.authToken)
-  await client.trackLlmRuns(trackingContext.botId, {
-    llm_runs: llmRuns,
-  })
-}