createChatModel method

BaseChatModel<ChatModelOptions> createChatModel(
  1. List<ToolSpec> tools
)

Implementation

BaseChatModel createChatModel(List<ToolSpec> tools) {
  switch (type) {
    case AgentAIConnectorType.openai:
      return ChatOpenAI(
        apiKey: apiKey,
        baseUrl: baseUrl ?? "https://api.openai.com/v1",
        defaultOptions: ChatOpenAIOptions(model: model, tools: tools),
      );
    case AgentAIConnectorType.google:
      return ChatGoogleGenerativeAI(
        apiKey: apiKey,
        baseUrl:
            baseUrl ?? 'https://generativelanguage.googleapis.com/v1beta',
        defaultOptions: ChatGoogleGenerativeAIOptions(
          model: model,
          tools: tools,
        ),
      );
  }
}