@botonic/plugin-ai-agents 0.36.0-alpha.5 → 0.36.0-alpha.7

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')
-  }
-}

package/src/errors.ts

@@ -1,6 +0,0 @@
-export class AiAgentError extends Error {
-  constructor(message: string) {
-    super(message)
-    this.name = 'AiAgentError'
-  }
-}

package/src/hubtype-client.ts

@@ -1,43 +0,0 @@
-import { BotContext } from '@botonic/core'
-import axios from 'axios'
-
-import { HUBTYPE_API_URL } from './constants'
-import { AgenticInputMessage } from './types'
-
-export class HubtypeApiClient {
-  private readonly authToken: string
-
-  constructor(authToken: string) {
-    this.authToken = authToken
-  }
-
-  async getMessages(
-    request: BotContext,
-    memoryLength: number
-  ): Promise<AgenticInputMessage[]> {
-    const url = `${HUBTYPE_API_URL}/external/v1/ai/agent/message_history/`
-    const headers = {
-      'Content-Type': 'application/json',
-      Authorization: `Bearer ${this.authToken}`,
-    }
-    const params = {
-      last_message_id: request.input.message_id,
-      num_messages: memoryLength,
-    }
-
-    try {
-      const messages = await axios.get<{ messages: AgenticInputMessage[] }>(
-        url,
-        {
-          headers,
-          params,
-        }
-      )
-
-      return messages.data.messages
-    } catch (error) {
-      console.error(error)
-      throw new Error('Failed to get messages from Hubtype')
-    }
-  }
-}

package/src/tools/custom.ts

@@ -1,14 +0,0 @@
-import { DynamicStructuredTool, tool } from '@langchain/core/tools'
-
-import { CustomTool } from '../types'
-
-export function createCustomTool(
-  customTool: CustomTool
-): DynamicStructuredTool {
-  return tool(customTool.func, {
-    name: customTool.name,
-    description: customTool.description,
-    schema: customTool.schema,
-    returnDirect: customTool.returnDirect || false,
-  })
-}

package/src/tools/default.ts

@@ -1,52 +0,0 @@
-import { tool } from '@langchain/core/tools'
-import { z } from 'zod'
-
-export const handoffToHumanAgent = tool(
-  async () => {
-    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(
-  async () => {
-    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(
-  async () => {
-    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,
-]

package/src/utils.ts

@@ -1,28 +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 enum AiProvider {
-  AzureOpenAI = 'azureOpenAI',
-}
-
-export function loadChatModel(provider: AiProvider) {
-  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`)
-  }
-}