@botonic/plugin-ai-agents 0.45.0 → 0.46.0-alpha.0

package/src/llm-config.ts

@@ -0,0 +1,96 @@
+import type { VerbosityLevel } from '@botonic/core'
+import {
+  type ModelProvider,
+  type ModelSettings,
+  OpenAIProvider,
+} from '@openai/agents'
+import OpenAI, { AzureOpenAI } from 'openai'
+import {
+  AZURE_OPENAI_API_BASE,
+  AZURE_OPENAI_API_KEY,
+  AZURE_OPENAI_API_VERSION,
+  isProd,
+  OPENAI_API_KEY,
+  OPENAI_MODEL,
+  OPENAI_PROVIDER,
+} from './constants'
+
+export class LLMConfig {
+  private readonly maxRetries: number
+  private readonly timeout: number
+  public readonly modelName: string
+  public readonly modelSettings: ModelSettings
+  public readonly modelProvider: ModelProvider
+
+  constructor(
+    maxRetries: number,
+    timeout: number,
+    modelName: string,
+    verbosity: VerbosityLevel
+  ) {
+    this.maxRetries = maxRetries
+    this.timeout = timeout
+    this.modelName = OPENAI_PROVIDER === 'openai' ? OPENAI_MODEL : modelName
+    this.modelProvider = this.getModelProvider()
+    this.modelSettings = this.getModelSettings(modelName, verbosity)
+  }
+
+  private getModelProvider(): ModelProvider {
+    const client = this.getClient()
+    return new OpenAIProvider({
+      openAIClient: client,
+      useResponses: false,
+    })
+  }
+
+  private getClient(): OpenAI | AzureOpenAI {
+    if (OPENAI_PROVIDER === 'openai') {
+      return this.getOpenAIClient()
+    }
+
+    return this.getAzureClient()
+  }
+
+  private getOpenAIClient(): OpenAI {
+    return new OpenAI({
+      apiKey: OPENAI_API_KEY,
+      timeout: this.timeout,
+      maxRetries: this.maxRetries,
+      dangerouslyAllowBrowser: !isProd,
+    })
+  }
+
+  private getAzureClient(): AzureOpenAI {
+    return new AzureOpenAI({
+      apiKey: AZURE_OPENAI_API_KEY,
+      apiVersion: AZURE_OPENAI_API_VERSION,
+      deployment: this.modelName,
+      baseURL: AZURE_OPENAI_API_BASE,
+      timeout: this.timeout,
+      maxRetries: this.maxRetries,
+      dangerouslyAllowBrowser: !isProd,
+    })
+  }
+
+  private getModelSettings(
+    model: string,
+    verbosity: VerbosityLevel
+  ): ModelSettings {
+    if (model.includes('gpt-5')) {
+      return {
+        reasoning: { effort: 'none' },
+        temperature: 1,
+        text: { verbosity },
+      }
+    }
+
+    if (model.includes('gpt-4')) {
+      return {
+        temperature: 0,
+        text: { verbosity },
+      }
+    }
+
+    throw new Error(`Unsupported model: ${model}`)
+  }
+}

package/src/runner.ts

@@ -9,8 +9,9 @@ import {
   RunToolCallItem,
   RunToolCallOutputItem,
 } from '@openai/agents'
-
+import { OPENAI_PROVIDER } from './constants'
 import type { DebugLogger } from './debug-logger'
+import type { LLMConfig } from './llm-config'
 import { retrieveKnowledge } from './tools'
 import type {
   AgenticInputMessage,
@@ -35,9 +36,15 @@ export class AIAgentRunner<
 > {
   private agent: AIAgent<TPlugins, TExtraData>
   private logger: DebugLogger
+  private llmConfig: LLMConfig
 
-  constructor(agent: AIAgent<TPlugins, TExtraData>, logger: DebugLogger) {
+  constructor(
+    agent: AIAgent<TPlugins, TExtraData>,
+    openAiClient: LLMConfig,
+    logger: DebugLogger
+  ) {
     this.agent = agent
+    this.llmConfig = openAiClient
     this.logger = logger
   }
 
@@ -47,11 +54,24 @@ export class AIAgentRunner<
   ): Promise<RunResult> {
     const startTime = Date.now()
 
-    this.logger.logRunnerStart()
+    this.logger.logRunnerStart(
+      this.llmConfig.modelName,
+      this.llmConfig.modelSettings
+    )
 
     try {
+      const modelProvider = this.llmConfig.modelProvider
+      const modelSettings = this.llmConfig.modelSettings
+
+      const hasRetrieveKnowledge = this.agent.tools.includes(retrieveKnowledge)
+      if (hasRetrieveKnowledge && OPENAI_PROVIDER === 'azure') {
+        modelSettings.toolChoice = retrieveKnowledge.name
+      }
+
       const runner = new Runner({
-        modelSettings: { temperature: 0 },
+        modelSettings,
+        modelProvider,
+        tracingDisabled: true,
       })
       // Type assertion to bypass strict type checking - the actual return type from runner.run()
       // doesn't perfectly match our interface, but the properties we access are compatible

package/lib/cjs/openai.d.ts

@@ -1 +0,0 @@
-export declare function setUpOpenAI(maxRetries?: number, timeout?: number): void;

package/lib/cjs/openai.js

@@ -1,39 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.setUpOpenAI = setUpOpenAI;
-const tslib_1 = require("tslib");
-const agents_1 = require("@openai/agents");
-const openai_1 = tslib_1.__importStar(require("openai"));
-const constants_1 = require("./constants");
-function setUpOpenAI(maxRetries, timeout) {
-    if (constants_1.OPENAI_PROVIDER === 'azure') {
-        setAzureOpenAIClient(maxRetries, timeout);
-    }
-    else {
-        setOpenAIClient(maxRetries, timeout);
-    }
-    (0, agents_1.setOpenAIAPI)('chat_completions');
-    (0, agents_1.setTracingDisabled)(true);
-}
-function setOpenAIClient(maxRetries, timeout) {
-    const client = new openai_1.default({
-        apiKey: constants_1.OPENAI_API_KEY,
-        timeout: timeout || 16000, // 16 seconds
-        maxRetries: maxRetries || 2,
-        dangerouslyAllowBrowser: !constants_1.isProd,
-    });
-    (0, agents_1.setDefaultOpenAIClient)(client);
-}
-function setAzureOpenAIClient(maxRetries, timeout) {
-    const client = new openai_1.AzureOpenAI({
-        apiKey: constants_1.AZURE_OPENAI_API_KEY,
-        apiVersion: constants_1.AZURE_OPENAI_API_VERSION,
-        deployment: constants_1.AZURE_OPENAI_API_DEPLOYMENT_NAME,
-        baseURL: constants_1.AZURE_OPENAI_API_BASE,
-        timeout: timeout || 16000, // 16 seconds
-        maxRetries: maxRetries || 2,
-        dangerouslyAllowBrowser: !constants_1.isProd,
-    });
-    (0, agents_1.setDefaultOpenAIClient)(client);
-}
-//# sourceMappingURL=openai.js.map

package/lib/cjs/openai.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/openai.ts"],"names":[],"mappings":";;AAiBA,kCAQC;;AAzBD,2CAIuB;AACvB,yDAA4C;AAE5C,2CAQoB;AAEpB,SAAgB,WAAW,CAAC,UAAmB,EAAE,OAAgB;IAC/D,IAAI,2BAAe,KAAK,OAAO,EAAE,CAAC;QAChC,oBAAoB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAA;IAC3C,CAAC;SAAM,CAAC;QACN,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,CAAA;IACtC,CAAC;IACD,IAAA,qBAAY,EAAC,kBAAkB,CAAC,CAAA;IAChC,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAA;AAC1B,CAAC;AAED,SAAS,eAAe,CAAC,UAAmB,EAAE,OAAgB;IAC5D,MAAM,MAAM,GAAG,IAAI,gBAAM,CAAC;QACxB,MAAM,EAAE,0BAAc;QACtB,OAAO,EAAE,OAAO,IAAI,KAAK,EAAE,aAAa;QACxC,UAAU,EAAE,UAAU,IAAI,CAAC;QAC3B,uBAAuB,EAAE,CAAC,kBAAM;KACjC,CAAC,CAAA;IACF,IAAA,+BAAsB,EAAC,MAAM,CAAC,CAAA;AAChC,CAAC;AAED,SAAS,oBAAoB,CAAC,UAAmB,EAAE,OAAgB;IACjE,MAAM,MAAM,GAAG,IAAI,oBAAW,CAAC;QAC7B,MAAM,EAAE,gCAAoB;QAC5B,UAAU,EAAE,oCAAwB;QACpC,UAAU,EAAE,4CAAgC;QAC5C,OAAO,EAAE,iCAAqB;QAC9B,OAAO,EAAE,OAAO,IAAI,KAAK,EAAE,aAAa;QACxC,UAAU,EAAE,UAAU,IAAI,CAAC;QAC3B,uBAAuB,EAAE,CAAC,kBAAM;KACjC,CAAC,CAAA;IACF,IAAA,+BAAsB,EAAC,MAAM,CAAC,CAAA;AAChC,CAAC"}

package/lib/esm/openai.d.ts

@@ -1 +0,0 @@
-export declare function setUpOpenAI(maxRetries?: number, timeout?: number): void;

package/lib/esm/openai.js

@@ -1,39 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.setUpOpenAI = setUpOpenAI;
-const tslib_1 = require("tslib");
-const agents_1 = require("@openai/agents");
-const openai_1 = tslib_1.__importStar(require("openai"));
-const constants_1 = require("./constants");
-function setUpOpenAI(maxRetries, timeout) {
-    if (constants_1.OPENAI_PROVIDER === 'azure') {
-        setAzureOpenAIClient(maxRetries, timeout);
-    }
-    else {
-        setOpenAIClient(maxRetries, timeout);
-    }
-    (0, agents_1.setOpenAIAPI)('chat_completions');
-    (0, agents_1.setTracingDisabled)(true);
-}
-function setOpenAIClient(maxRetries, timeout) {
-    const client = new openai_1.default({
-        apiKey: constants_1.OPENAI_API_KEY,
-        timeout: timeout || 16000, // 16 seconds
-        maxRetries: maxRetries || 2,
-        dangerouslyAllowBrowser: !constants_1.isProd,
-    });
-    (0, agents_1.setDefaultOpenAIClient)(client);
-}
-function setAzureOpenAIClient(maxRetries, timeout) {
-    const client = new openai_1.AzureOpenAI({
-        apiKey: constants_1.AZURE_OPENAI_API_KEY,
-        apiVersion: constants_1.AZURE_OPENAI_API_VERSION,
-        deployment: constants_1.AZURE_OPENAI_API_DEPLOYMENT_NAME,
-        baseURL: constants_1.AZURE_OPENAI_API_BASE,
-        timeout: timeout || 16000, // 16 seconds
-        maxRetries: maxRetries || 2,
-        dangerouslyAllowBrowser: !constants_1.isProd,
-    });
-    (0, agents_1.setDefaultOpenAIClient)(client);
-}
-//# sourceMappingURL=openai.js.map

package/lib/esm/openai.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/openai.ts"],"names":[],"mappings":";;AAiBA,kCAQC;;AAzBD,2CAIuB;AACvB,yDAA4C;AAE5C,2CAQoB;AAEpB,SAAgB,WAAW,CAAC,UAAmB,EAAE,OAAgB;IAC/D,IAAI,2BAAe,KAAK,OAAO,EAAE,CAAC;QAChC,oBAAoB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAA;IAC3C,CAAC;SAAM,CAAC;QACN,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,CAAA;IACtC,CAAC;IACD,IAAA,qBAAY,EAAC,kBAAkB,CAAC,CAAA;IAChC,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAA;AAC1B,CAAC;AAED,SAAS,eAAe,CAAC,UAAmB,EAAE,OAAgB;IAC5D,MAAM,MAAM,GAAG,IAAI,gBAAM,CAAC;QACxB,MAAM,EAAE,0BAAc;QACtB,OAAO,EAAE,OAAO,IAAI,KAAK,EAAE,aAAa;QACxC,UAAU,EAAE,UAAU,IAAI,CAAC;QAC3B,uBAAuB,EAAE,CAAC,kBAAM;KACjC,CAAC,CAAA;IACF,IAAA,+BAAsB,EAAC,MAAM,CAAC,CAAA;AAChC,CAAC;AAED,SAAS,oBAAoB,CAAC,UAAmB,EAAE,OAAgB;IACjE,MAAM,MAAM,GAAG,IAAI,oBAAW,CAAC;QAC7B,MAAM,EAAE,gCAAoB;QAC5B,UAAU,EAAE,oCAAwB;QACpC,UAAU,EAAE,4CAAgC;QAC5C,OAAO,EAAE,iCAAqB;QAC9B,OAAO,EAAE,OAAO,IAAI,KAAK,EAAE,aAAa;QACxC,UAAU,EAAE,UAAU,IAAI,CAAC;QAC3B,uBAAuB,EAAE,CAAC,kBAAM;KACjC,CAAC,CAAA;IACF,IAAA,+BAAsB,EAAC,MAAM,CAAC,CAAA;AAChC,CAAC"}

package/src/openai.ts

@@ -1,49 +0,0 @@
-import {
-  setDefaultOpenAIClient,
-  setOpenAIAPI,
-  setTracingDisabled,
-} from '@openai/agents'
-import OpenAI, { AzureOpenAI } from 'openai'
-
-import {
-  AZURE_OPENAI_API_BASE,
-  AZURE_OPENAI_API_DEPLOYMENT_NAME,
-  AZURE_OPENAI_API_KEY,
-  AZURE_OPENAI_API_VERSION,
-  isProd,
-  OPENAI_API_KEY,
-  OPENAI_PROVIDER,
-} from './constants'
-
-export function setUpOpenAI(maxRetries?: number, timeout?: number) {
-  if (OPENAI_PROVIDER === 'azure') {
-    setAzureOpenAIClient(maxRetries, timeout)
-  } else {
-    setOpenAIClient(maxRetries, timeout)
-  }
-  setOpenAIAPI('chat_completions')
-  setTracingDisabled(true)
-}
-
-function setOpenAIClient(maxRetries?: number, timeout?: number) {
-  const client = new OpenAI({
-    apiKey: OPENAI_API_KEY,
-    timeout: timeout || 16000, // 16 seconds
-    maxRetries: maxRetries || 2,
-    dangerouslyAllowBrowser: !isProd,
-  })
-  setDefaultOpenAIClient(client)
-}
-
-function setAzureOpenAIClient(maxRetries?: number, timeout?: number) {
-  const client = new AzureOpenAI({
-    apiKey: AZURE_OPENAI_API_KEY,
-    apiVersion: AZURE_OPENAI_API_VERSION,
-    deployment: AZURE_OPENAI_API_DEPLOYMENT_NAME,
-    baseURL: AZURE_OPENAI_API_BASE,
-    timeout: timeout || 16000, // 16 seconds
-    maxRetries: maxRetries || 2,
-    dangerouslyAllowBrowser: !isProd,
-  })
-  setDefaultOpenAIClient(client)
-}