@botonic/plugin-ai-agents 0.36.0-alpha.6 → 0.36.0-alpha.8

package/src/structured-output/carousel.ts

@@ -0,0 +1,36 @@
+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 const CarouselSchema = z
+  .object({
+    type: z.enum(['carousel']),
+    content: z.object({
+      elements: z.array(
+        z.object({
+          title: z.string(),
+          subtitle: z.string(),
+          image: z.string(),
+          button: z.object({
+            text: z.string(),
+            url: z.string(),
+          }),
+        })
+      ),
+    }),
+  })
+  .describe('A carousel message containing a list of elements')

package/src/structured-output/exit.ts

@@ -0,0 +1,15 @@
+import z from 'zod'
+
+import { BaseMessage } from './shared'
+
+export interface ExitMessage extends BaseMessage {
+  type: 'exit'
+}
+
+export const ExitSchema = z
+  .object({
+    type: z.enum(['exit']),
+  })
+  .describe(
+    'An exit message. This message should only be used to exit the agent due to out of context or the user asking to exit the conversation'
+  )

package/src/structured-output/index.ts

@@ -0,0 +1,27 @@
+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
+
+export interface Output {
+  messages: OutputMessage[]
+}
+
+export const OutputSchema = z
+  .object({
+    messages: z.array(
+      z.union([TextSchema, TextWithButtonsSchema, CarouselSchema, ExitSchema])
+    ),
+  })
+  .describe('The messages to be sent to the user')

package/src/structured-output/shared.ts

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

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

@@ -0,0 +1,23 @@
+import z from 'zod'
+
+import { BaseMessage } from './shared'
+
+export interface TextWithButtonsMessage extends BaseMessage {
+  type: 'textWithButtons'
+  content: {
+    text: string
+    buttons: string[]
+  }
+}
+
+export const TextWithButtonsSchema = z
+  .object({
+    type: z.enum(['textWithButtons']),
+    content: z.object({
+      text: z.string(),
+      buttons: z.array(z.string()),
+    }),
+  })
+  .describe(
+    'A text message with buttons to allow the user to use quick replies'
+  )

package/src/structured-output/text.ts

@@ -0,0 +1,19 @@
+import { z } from 'zod'
+
+import { BaseMessage } from './shared'
+
+export interface TextMessage extends BaseMessage {
+  type: 'text'
+  content: {
+    text: string
+  }
+}
+
+export const TextSchema = z
+  .object({
+    type: z.enum(['text']),
+    content: z.object({
+      text: z.string(),
+    }),
+  })
+  .describe('A text message')

package/src/types.ts

@@ -1,4 +1,7 @@
-import { ZodSchema } from 'zod'
+import { AgentInputItem } from '@openai/agents'
+import { ZodObject } from 'zod'
+
+import { OutputMessage } from './structured-output'
 export interface PluginAiAgentOptions {
   authToken?: string
   customTools?: CustomTool[]
@@ -7,8 +10,7 @@ export interface PluginAiAgentOptions {
 export interface CustomTool {
   name: string
   description: string
-  schema: ZodSchema
-  returnDirect?: boolean
+  schema: ZodObject<any>
   func: (input?: any) => Promise<any>
 }
 
@@ -18,34 +20,5 @@ export interface AiAgentArgs {
   activeTools?: { name: string }[]
 }
 
-export interface AgenticBaseMessage {
-  role: 'assistant' | 'user' | 'tool' | 'exit'
-}
-
-export interface AssistantMessage extends AgenticBaseMessage {
-  role: 'assistant'
-  content: string
-}
-
-export interface UserMessage extends AgenticBaseMessage {
-  role: 'user'
-  content: string
-}
-
-export interface ToolMessage extends AgenticBaseMessage {
-  role: 'tool'
-  toolName: string
-  toolOutput: string | null
-}
-
-export interface ExitMessage extends AgenticBaseMessage {
-  role: 'exit'
-}
-
-export type AgenticInputMessage = AssistantMessage | UserMessage
-export type AgenticOutputMessage = AssistantMessage | ToolMessage | ExitMessage
-export type AgenticMessage =
-  | AssistantMessage
-  | UserMessage
-  | ToolMessage
-  | ExitMessage
+export type AgenticInputMessage = AgentInputItem
+export type AgenticOutputMessage = OutputMessage

package/lib/cjs/ai-agent-client.d.ts

@@ -1,9 +0,0 @@
-import { BaseChatModel } from '@langchain/core/language_models/chat_models';
-import { StructuredTool } from '@langchain/core/tools';
-import { CompiledStateGraph } from '@langchain/langgraph';
-import { AgenticInputMessage, AgenticOutputMessage, AiAgentArgs } from './types';
-export declare class AiAgentClient {
-    agent: CompiledStateGraph<any, any>;
-    constructor(aiAgentArgs: AiAgentArgs, chatModel: BaseChatModel, tools?: StructuredTool[]);
-    runAgent(_messages: AgenticInputMessage[]): Promise<AgenticOutputMessage>;
-}

package/lib/cjs/ai-agent-client.js

@@ -1,49 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.AiAgentClient = void 0;
-const tslib_1 = require("tslib");
-const messages_1 = require("@langchain/core/messages");
-const prebuilt_1 = require("@langchain/langgraph/prebuilt");
-const default_1 = require("./tools/default");
-class AiAgentClient {
-    constructor(aiAgentArgs, chatModel, tools = []) {
-        this.agent = (0, prebuilt_1.createReactAgent)({
-            name: aiAgentArgs.name,
-            llm: chatModel,
-            tools: tools,
-            prompt: aiAgentArgs.instructions,
-        });
-    }
-    runAgent(_messages) {
-        return tslib_1.__awaiter(this, void 0, void 0, function* () {
-            const response = yield this.agent.invoke({
-                messages: _messages.map((message) => ({
-                    role: message.role,
-                    content: message.content,
-                })),
-            });
-            const lastMessage = response.messages.at(-1);
-            if (lastMessage instanceof messages_1.ToolMessage) {
-                if (lastMessage.name && default_1.EXIT_TOOLS.includes(lastMessage.name)) {
-                    return {
-                        role: 'exit',
-                    };
-                }
-                return {
-                    role: 'tool',
-                    toolName: lastMessage.name,
-                    toolOutput: lastMessage.content,
-                };
-            }
-            if (lastMessage instanceof messages_1.AIMessage) {
-                return {
-                    role: 'assistant',
-                    content: lastMessage.content,
-                };
-            }
-            throw new Error('Last message is not a valid message');
-        });
-    }
-}
-exports.AiAgentClient = AiAgentClient;
-//# sourceMappingURL=ai-agent-client.js.map

package/lib/cjs/ai-agent-client.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"ai-agent-client.js","sourceRoot":"","sources":["../../src/ai-agent-client.ts"],"names":[],"mappings":";;;;AACA,uDAGiC;AAGjC,4DAAgE;AAEhE,6CAA4C;AAU5C,MAAa,aAAa;IAGxB,YACE,WAAwB,EACxB,SAAwB,EACxB,QAA0B,EAAE;QAE5B,IAAI,CAAC,KAAK,GAAG,IAAA,2BAAgB,EAAC;YAC5B,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,GAAG,EAAE,SAAS;YACd,KAAK,EAAE,KAAK;YACZ,MAAM,EAAE,WAAW,CAAC,YAAY;SACjC,CAAC,CAAA;IACJ,CAAC;IAEK,QAAQ,CACZ,SAAgC;;YAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,QAAQ,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,OAA4B,EAAE,EAAE,CAAC,CAAC;oBACzD,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,OAAO,EAAE,OAAO,CAAC,OAAO;iBACzB,CAAC,CAAC;aACJ,CAAC,CAAA;YACF,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YAE5C,IAAI,WAAW,YAAY,sBAAoB,EAAE;gBAC/C,IAAI,WAAW,CAAC,IAAI,IAAI,oBAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE;oBAC7D,OAAO;wBACL,IAAI,EAAE,MAAM;qBACE,CAAA;iBACjB;gBACD,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,QAAQ,EAAE,WAAW,CAAC,IAAI;oBAC1B,UAAU,EAAE,WAAW,CAAC,OAAO;iBACjB,CAAA;aACjB;YAED,IAAI,WAAW,YAAY,oBAAkB,EAAE;gBAC7C,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,WAAW,CAAC,OAAO;iBACT,CAAA;aACtB;YAED,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;QACxD,CAAC;KAAA;CACF;AAjDD,sCAiDC"}

package/lib/cjs/errors.d.ts

@@ -1,3 +0,0 @@
-export declare class AiAgentError extends Error {
-    constructor(message: string);
-}

package/lib/cjs/errors.js

@@ -1,11 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.AiAgentError = void 0;
-class AiAgentError extends Error {
-    constructor(message) {
-        super(message);
-        this.name = 'AiAgentError';
-    }
-}
-exports.AiAgentError = AiAgentError;
-//# sourceMappingURL=errors.js.map

package/lib/cjs/errors.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../src/errors.ts"],"names":[],"mappings":";;;AAAA,MAAa,YAAa,SAAQ,KAAK;IACrC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAA;QACd,IAAI,CAAC,IAAI,GAAG,cAAc,CAAA;IAC5B,CAAC;CACF;AALD,oCAKC"}

package/lib/cjs/tools/custom.d.ts

@@ -1,3 +0,0 @@
-import { DynamicStructuredTool } from '@langchain/core/tools';
-import { CustomTool } from '../types';
-export declare function createCustomTool(customTool: CustomTool): DynamicStructuredTool;

package/lib/cjs/tools/custom.js

@@ -1,14 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createCustomTool = void 0;
-const tools_1 = require("@langchain/core/tools");
-function createCustomTool(customTool) {
-    return (0, tools_1.tool)(customTool.func, {
-        name: customTool.name,
-        description: customTool.description,
-        schema: customTool.schema,
-        returnDirect: customTool.returnDirect || false,
-    });
-}
-exports.createCustomTool = createCustomTool;
-//# sourceMappingURL=custom.js.map

package/lib/cjs/tools/custom.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"custom.js","sourceRoot":"","sources":["../../../src/tools/custom.ts"],"names":[],"mappings":";;;AAAA,iDAAmE;AAInE,SAAgB,gBAAgB,CAC9B,UAAsB;IAEtB,OAAO,IAAA,YAAI,EAAC,UAAU,CAAC,IAAI,EAAE;QAC3B,IAAI,EAAE,UAAU,CAAC,IAAI;QACrB,WAAW,EAAE,UAAU,CAAC,WAAW;QACnC,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,YAAY,EAAE,UAAU,CAAC,YAAY,IAAI,KAAK;KAC/C,CAAC,CAAA;AACJ,CAAC;AATD,4CASC"}

package/lib/cjs/tools/default.d.ts

@@ -1,6 +0,0 @@
-import { z } from 'zod';
-export declare const handoffToHumanAgent: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>;
-export declare const outOfContext: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>;
-export declare const finishConversation: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>;
-export declare const MANDATORY_TOOLS: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>[];
-export declare const EXIT_TOOLS: string[];

package/lib/cjs/tools/default.js

@@ -1,41 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.EXIT_TOOLS = exports.MANDATORY_TOOLS = exports.finishConversation = exports.outOfContext = exports.handoffToHumanAgent = void 0;
-const tslib_1 = require("tslib");
-const tools_1 = require("@langchain/core/tools");
-const zod_1 = require("zod");
-exports.handoffToHumanAgent = (0, tools_1.tool)(() => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
-    return null;
-}), {
-    name: 'handoffToHumanAgent',
-    schema: zod_1.z.object({}),
-    description: 'Use this tool strictly when the user wants to talk with a human agent.',
-    returnDirect: true,
-});
-exports.outOfContext = (0, tools_1.tool)(() => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
-    return null;
-}), {
-    name: 'outOfContext',
-    schema: zod_1.z.object({}),
-    description: 'Use this when the user talks about something that is clearly out of your context. Example: "I want to buy a car" when you are a travel agent.',
-    returnDirect: true,
-});
-exports.finishConversation = (0, tools_1.tool)(() => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
-    return null;
-}), {
-    name: 'finishConversation',
-    schema: zod_1.z.object({}),
-    description: 'Use this when the user wants to end the conversation.',
-    returnDirect: true,
-});
-exports.MANDATORY_TOOLS = [
-    exports.handoffToHumanAgent,
-    exports.outOfContext,
-    exports.finishConversation,
-];
-exports.EXIT_TOOLS = [
-    exports.handoffToHumanAgent.name,
-    exports.outOfContext.name,
-    exports.finishConversation.name,
-];
-//# sourceMappingURL=default.js.map

package/lib/cjs/tools/default.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"default.js","sourceRoot":"","sources":["../../../src/tools/default.ts"],"names":[],"mappings":";;;;AAAA,iDAA4C;AAC5C,6BAAuB;AAEV,QAAA,mBAAmB,GAAG,IAAA,YAAI,EACrC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,qBAAqB;IAC3B,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,wEAAwE;IAC1E,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAEY,QAAA,YAAY,GAAG,IAAA,YAAI,EAC9B,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,cAAc;IACpB,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,+IAA+I;IACjJ,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAEY,QAAA,kBAAkB,GAAG,IAAA,YAAI,EACpC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,oBAAoB;IAC1B,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EAAE,uDAAuD;IACpE,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAEY,QAAA,eAAe,GAAG;IAC7B,2BAAmB;IACnB,oBAAY;IACZ,0BAAkB;CACnB,CAAA;AAEY,QAAA,UAAU,GAAG;IACxB,2BAAmB,CAAC,IAAI;IACxB,oBAAY,CAAC,IAAI;IACjB,0BAAkB,CAAC,IAAI;CACxB,CAAA"}

package/lib/cjs/utils.d.ts

@@ -1,5 +0,0 @@
-import { AzureChatOpenAI } from '@langchain/openai';
-export declare enum AiProvider {
-    AzureOpenAI = "azureOpenAI"
-}
-export declare function loadChatModel(provider: AiProvider): AzureChatOpenAI;

package/lib/cjs/utils.js

@@ -1,26 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.loadChatModel = exports.AiProvider = void 0;
-const openai_1 = require("@langchain/openai");
-const constants_1 = require("./constants");
-var AiProvider;
-(function (AiProvider) {
-    AiProvider["AzureOpenAI"] = "azureOpenAI";
-})(AiProvider = exports.AiProvider || (exports.AiProvider = {}));
-function loadChatModel(provider) {
-    switch (provider) {
-        case AiProvider.AzureOpenAI:
-            return new openai_1.AzureChatOpenAI({
-                azureOpenAIApiVersion: constants_1.AZURE_OPENAI_API_VERSION,
-                azureOpenAIApiKey: constants_1.AZURE_OPENAI_API_KEY,
-                azureOpenAIApiDeploymentName: constants_1.AZURE_OPENAI_API_DEPLOYMENT_NAME,
-                azureOpenAIEndpoint: constants_1.AZURE_OPENAI_API_BASE,
-                temperature: 0,
-                // other params...
-            });
-        default:
-            throw new Error(`Provider ${provider} not supported`);
-    }
-}
-exports.loadChatModel = loadChatModel;
-//# sourceMappingURL=utils.js.map

package/lib/cjs/utils.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/utils.ts"],"names":[],"mappings":";;;AAAA,8CAAmD;AAEnD,2CAKoB;AAEpB,IAAY,UAEX;AAFD,WAAY,UAAU;IACpB,yCAA2B,CAAA;AAC7B,CAAC,EAFW,UAAU,GAAV,kBAAU,KAAV,kBAAU,QAErB;AAED,SAAgB,aAAa,CAAC,QAAoB;IAChD,QAAQ,QAAQ,EAAE;QAChB,KAAK,UAAU,CAAC,WAAW;YACzB,OAAO,IAAI,wBAAe,CAAC;gBACzB,qBAAqB,EAAE,oCAAwB;gBAC/C,iBAAiB,EAAE,gCAAoB;gBACvC,4BAA4B,EAAE,4CAAgC;gBAC9D,mBAAmB,EAAE,iCAAqB;gBAC1C,WAAW,EAAE,CAAC;gBACd,kBAAkB;aACnB,CAAC,CAAA;QACJ;YACE,MAAM,IAAI,KAAK,CAAC,YAAY,QAAQ,gBAAgB,CAAC,CAAA;KACxD;AACH,CAAC;AAdD,sCAcC"}

package/lib/esm/ai-agent-client.d.ts

@@ -1,9 +0,0 @@
-import { BaseChatModel } from '@langchain/core/language_models/chat_models';
-import { StructuredTool } from '@langchain/core/tools';
-import { CompiledStateGraph } from '@langchain/langgraph';
-import { AgenticInputMessage, AgenticOutputMessage, AiAgentArgs } from './types';
-export declare class AiAgentClient {
-    agent: CompiledStateGraph<any, any>;
-    constructor(aiAgentArgs: AiAgentArgs, chatModel: BaseChatModel, tools?: StructuredTool[]);
-    runAgent(_messages: AgenticInputMessage[]): Promise<AgenticOutputMessage>;
-}

package/lib/esm/ai-agent-client.js

@@ -1,45 +0,0 @@
-import { __awaiter } from "tslib";
-import { AIMessage as LangchainAIMessage, ToolMessage as LangchainToolMessage, } from '@langchain/core/messages';
-import { createReactAgent } from '@langchain/langgraph/prebuilt';
-import { EXIT_TOOLS } from './tools/default';
-export class AiAgentClient {
-    constructor(aiAgentArgs, chatModel, tools = []) {
-        this.agent = createReactAgent({
-            name: aiAgentArgs.name,
-            llm: chatModel,
-            tools: tools,
-            prompt: aiAgentArgs.instructions,
-        });
-    }
-    runAgent(_messages) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const response = yield this.agent.invoke({
-                messages: _messages.map((message) => ({
-                    role: message.role,
-                    content: message.content,
-                })),
-            });
-            const lastMessage = response.messages.at(-1);
-            if (lastMessage instanceof LangchainToolMessage) {
-                if (lastMessage.name && EXIT_TOOLS.includes(lastMessage.name)) {
-                    return {
-                        role: 'exit',
-                    };
-                }
-                return {
-                    role: 'tool',
-                    toolName: lastMessage.name,
-                    toolOutput: lastMessage.content,
-                };
-            }
-            if (lastMessage instanceof LangchainAIMessage) {
-                return {
-                    role: 'assistant',
-                    content: lastMessage.content,
-                };
-            }
-            throw new Error('Last message is not a valid message');
-        });
-    }
-}
-//# sourceMappingURL=ai-agent-client.js.map

package/lib/esm/ai-agent-client.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"ai-agent-client.js","sourceRoot":"","sources":["../../src/ai-agent-client.ts"],"names":[],"mappings":";AACA,OAAO,EACL,SAAS,IAAI,kBAAkB,EAC/B,WAAW,IAAI,oBAAoB,GACpC,MAAM,0BAA0B,CAAA;AAGjC,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAA;AAEhE,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAA;AAU5C,MAAM,OAAO,aAAa;IAGxB,YACE,WAAwB,EACxB,SAAwB,EACxB,QAA0B,EAAE;QAE5B,IAAI,CAAC,KAAK,GAAG,gBAAgB,CAAC;YAC5B,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,GAAG,EAAE,SAAS;YACd,KAAK,EAAE,KAAK;YACZ,MAAM,EAAE,WAAW,CAAC,YAAY;SACjC,CAAC,CAAA;IACJ,CAAC;IAEK,QAAQ,CACZ,SAAgC;;YAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,QAAQ,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,OAA4B,EAAE,EAAE,CAAC,CAAC;oBACzD,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,OAAO,EAAE,OAAO,CAAC,OAAO;iBACzB,CAAC,CAAC;aACJ,CAAC,CAAA;YACF,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YAE5C,IAAI,WAAW,YAAY,oBAAoB,EAAE;gBAC/C,IAAI,WAAW,CAAC,IAAI,IAAI,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE;oBAC7D,OAAO;wBACL,IAAI,EAAE,MAAM;qBACE,CAAA;iBACjB;gBACD,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,QAAQ,EAAE,WAAW,CAAC,IAAI;oBAC1B,UAAU,EAAE,WAAW,CAAC,OAAO;iBACjB,CAAA;aACjB;YAED,IAAI,WAAW,YAAY,kBAAkB,EAAE;gBAC7C,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,WAAW,CAAC,OAAO;iBACT,CAAA;aACtB;YAED,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;QACxD,CAAC;KAAA;CACF"}

package/lib/esm/errors.d.ts

@@ -1,3 +0,0 @@
-export declare class AiAgentError extends Error {
-    constructor(message: string);
-}

package/lib/esm/errors.js

@@ -1,7 +0,0 @@
-export class AiAgentError extends Error {
-    constructor(message) {
-        super(message);
-        this.name = 'AiAgentError';
-    }
-}
-//# sourceMappingURL=errors.js.map

package/lib/esm/errors.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../src/errors.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,YAAa,SAAQ,KAAK;IACrC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAA;QACd,IAAI,CAAC,IAAI,GAAG,cAAc,CAAA;IAC5B,CAAC;CACF"}

package/lib/esm/tools/custom.d.ts

@@ -1,3 +0,0 @@
-import { DynamicStructuredTool } from '@langchain/core/tools';
-import { CustomTool } from '../types';
-export declare function createCustomTool(customTool: CustomTool): DynamicStructuredTool;

package/lib/esm/tools/custom.js

@@ -1,10 +0,0 @@
-import { tool } from '@langchain/core/tools';
-export function createCustomTool(customTool) {
-    return tool(customTool.func, {
-        name: customTool.name,
-        description: customTool.description,
-        schema: customTool.schema,
-        returnDirect: customTool.returnDirect || false,
-    });
-}
-//# sourceMappingURL=custom.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"custom.js","sourceRoot":"","sources":["../../../src/tools/custom.ts"],"names":[],"mappings":"AAAA,OAAO,EAAyB,IAAI,EAAE,MAAM,uBAAuB,CAAA;AAInE,MAAM,UAAU,gBAAgB,CAC9B,UAAsB;IAEtB,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE;QAC3B,IAAI,EAAE,UAAU,CAAC,IAAI;QACrB,WAAW,EAAE,UAAU,CAAC,WAAW;QACnC,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,YAAY,EAAE,UAAU,CAAC,YAAY,IAAI,KAAK;KAC/C,CAAC,CAAA;AACJ,CAAC"}

package/lib/esm/tools/default.d.ts

@@ -1,6 +0,0 @@
-import { z } from 'zod';
-export declare const handoffToHumanAgent: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>;
-export declare const outOfContext: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>;
-export declare const finishConversation: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>;
-export declare const MANDATORY_TOOLS: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>[];
-export declare const EXIT_TOOLS: string[];

package/lib/esm/tools/default.js

@@ -1,38 +0,0 @@
-import { __awaiter } from "tslib";
-import { tool } from '@langchain/core/tools';
-import { z } from 'zod';
-export const handoffToHumanAgent = tool(() => __awaiter(void 0, void 0, void 0, function* () {
-    return null;
-}), {
-    name: 'handoffToHumanAgent',
-    schema: z.object({}),
-    description: 'Use this tool strictly when the user wants to talk with a human agent.',
-    returnDirect: true,
-});
-export const outOfContext = tool(() => __awaiter(void 0, void 0, void 0, function* () {
-    return null;
-}), {
-    name: 'outOfContext',
-    schema: z.object({}),
-    description: 'Use this when the user talks about something that is clearly out of your context. Example: "I want to buy a car" when you are a travel agent.',
-    returnDirect: true,
-});
-export const finishConversation = tool(() => __awaiter(void 0, void 0, void 0, function* () {
-    return null;
-}), {
-    name: 'finishConversation',
-    schema: z.object({}),
-    description: 'Use this when the user wants to end the conversation.',
-    returnDirect: true,
-});
-export const MANDATORY_TOOLS = [
-    handoffToHumanAgent,
-    outOfContext,
-    finishConversation,
-];
-export const EXIT_TOOLS = [
-    handoffToHumanAgent.name,
-    outOfContext.name,
-    finishConversation.name,
-];
-//# sourceMappingURL=default.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"default.js","sourceRoot":"","sources":["../../../src/tools/default.ts"],"names":[],"mappings":";AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,uBAAuB,CAAA;AAC5C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,MAAM,CAAC,MAAM,mBAAmB,GAAG,IAAI,CACrC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,qBAAqB;IAC3B,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,wEAAwE;IAC1E,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,IAAI,CAC9B,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,cAAc;IACpB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,+IAA+I;IACjJ,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,IAAI,CACpC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,oBAAoB;IAC1B,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EAAE,uDAAuD;IACpE,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAED,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,mBAAmB;IACnB,YAAY;IACZ,kBAAkB;CACnB,CAAA;AAED,MAAM,CAAC,MAAM,UAAU,GAAG;IACxB,mBAAmB,CAAC,IAAI;IACxB,YAAY,CAAC,IAAI;IACjB,kBAAkB,CAAC,IAAI;CACxB,CAAA"}

package/lib/esm/utils.d.ts

@@ -1,5 +0,0 @@
-import { AzureChatOpenAI } from '@langchain/openai';
-export declare enum AiProvider {
-    AzureOpenAI = "azureOpenAI"
-}
-export declare function loadChatModel(provider: AiProvider): AzureChatOpenAI;

package/lib/esm/utils.js

@@ -1,22 +0,0 @@
-import { AzureChatOpenAI } from '@langchain/openai';
-import { AZURE_OPENAI_API_BASE, AZURE_OPENAI_API_DEPLOYMENT_NAME, AZURE_OPENAI_API_KEY, AZURE_OPENAI_API_VERSION, } from './constants';
-export var AiProvider;
-(function (AiProvider) {
-    AiProvider["AzureOpenAI"] = "azureOpenAI";
-})(AiProvider || (AiProvider = {}));
-export function loadChatModel(provider) {
-    switch (provider) {
-        case AiProvider.AzureOpenAI:
-            return new AzureChatOpenAI({
-                azureOpenAIApiVersion: AZURE_OPENAI_API_VERSION,
-                azureOpenAIApiKey: AZURE_OPENAI_API_KEY,
-                azureOpenAIApiDeploymentName: AZURE_OPENAI_API_DEPLOYMENT_NAME,
-                azureOpenAIEndpoint: AZURE_OPENAI_API_BASE,
-                temperature: 0,
-                // other params...
-            });
-        default:
-            throw new Error(`Provider ${provider} not supported`);
-    }
-}
-//# sourceMappingURL=utils.js.map

package/lib/esm/utils.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAA;AAEnD,OAAO,EACL,qBAAqB,EACrB,gCAAgC,EAChC,oBAAoB,EACpB,wBAAwB,GACzB,MAAM,aAAa,CAAA;AAEpB,MAAM,CAAN,IAAY,UAEX;AAFD,WAAY,UAAU;IACpB,yCAA2B,CAAA;AAC7B,CAAC,EAFW,UAAU,KAAV,UAAU,QAErB;AAED,MAAM,UAAU,aAAa,CAAC,QAAoB;IAChD,QAAQ,QAAQ,EAAE;QAChB,KAAK,UAAU,CAAC,WAAW;YACzB,OAAO,IAAI,eAAe,CAAC;gBACzB,qBAAqB,EAAE,wBAAwB;gBAC/C,iBAAiB,EAAE,oBAAoB;gBACvC,4BAA4B,EAAE,gCAAgC;gBAC9D,mBAAmB,EAAE,qBAAqB;gBAC1C,WAAW,EAAE,CAAC;gBACd,kBAAkB;aACnB,CAAC,CAAA;QACJ;YACE,MAAM,IAAI,KAAK,CAAC,YAAY,QAAQ,gBAAgB,CAAC,CAAA;KACxD;AACH,CAAC"}

package/src/ai-agent-client.ts

@@ -1,69 +0,0 @@
-import { BaseChatModel } from '@langchain/core/language_models/chat_models'
-import {
-  AIMessage as LangchainAIMessage,
-  ToolMessage as LangchainToolMessage,
-} from '@langchain/core/messages'
-import { StructuredTool } from '@langchain/core/tools'
-import { CompiledStateGraph } from '@langchain/langgraph'
-import { createReactAgent } from '@langchain/langgraph/prebuilt'
-
-import { EXIT_TOOLS } from './tools/default'
-import {
-  AgenticInputMessage,
-  AgenticOutputMessage,
-  AiAgentArgs,
-  AssistantMessage,
-  ExitMessage,
-  ToolMessage,
-} from './types'
-
-export class AiAgentClient {
-  public agent: CompiledStateGraph<any, any> // TODO: apply RunInput, RunOutput, etc.
-
-  constructor(
-    aiAgentArgs: AiAgentArgs,
-    chatModel: BaseChatModel,
-    tools: StructuredTool[] = []
-  ) {
-    this.agent = createReactAgent({
-      name: aiAgentArgs.name,
-      llm: chatModel,
-      tools: tools,
-      prompt: aiAgentArgs.instructions,
-    })
-  }
-
-  async runAgent(
-    _messages: AgenticInputMessage[]
-  ): Promise<AgenticOutputMessage> {
-    const response = await this.agent.invoke({
-      messages: _messages.map((message: AgenticInputMessage) => ({
-        role: message.role,
-        content: message.content,
-      })),
-    })
-    const lastMessage = response.messages.at(-1)
-
-    if (lastMessage instanceof LangchainToolMessage) {
-      if (lastMessage.name && EXIT_TOOLS.includes(lastMessage.name)) {
-        return {
-          role: 'exit',
-        } as ExitMessage
-      }
-      return {
-        role: 'tool',
-        toolName: lastMessage.name,
-        toolOutput: lastMessage.content,
-      } as ToolMessage
-    }
-
-    if (lastMessage instanceof LangchainAIMessage) {
-      return {
-        role: 'assistant',
-        content: lastMessage.content,
-      } as AssistantMessage
-    }
-
-    throw new Error('Last message is not a valid message')
-  }
-}