/**
 * This file is part of the NocoBase (R) project.
 * Copyright (c) 2020-2024 NocoBase Co., Ltd.
 * Authors: NocoBase Team.
 *
 * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
 * For more information, please refer to: https://www.nocobase.com/agreement.
 */

var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
  for (var name in all)
    __defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
  if (from && typeof from === "object" || typeof from === "function") {
    for (let key of __getOwnPropNames(from))
      if (!__hasOwnProp.call(to, key) && key !== except)
        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
  }
  return to;
};
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
var ai_manager_exports = {};
__export(ai_manager_exports, {
  AIManager: () => AIManager,
  SupportedModel: () => SupportedModel
});
module.exports = __toCommonJS(ai_manager_exports);
var import_tool_manager = require("./tool-manager");
var SupportedModel = /* @__PURE__ */ ((SupportedModel2) => {
  SupportedModel2["LLM"] = "LLM";
  SupportedModel2["EMBEDDING"] = "EMBEDDING";
  return SupportedModel2;
})(SupportedModel || {});
class AIManager {
  constructor(plugin) {
    this.plugin = plugin;
  }
  llmProviders = /* @__PURE__ */ new Map();
  toolManager = new import_tool_manager.ToolManager();
  registerLLMProvider(name, meta) {
    this.llmProviders.set(name, meta);
  }
  listLLMProviders() {
    const providers = this.llmProviders.entries();
    return Array.from(providers).map(([name, { title, supportedModel, supportWebSearch }]) => ({
      name,
      title,
      supportedModel: supportedModel ?? ["LLM" /* LLM */],
      supportWebSearch: supportWebSearch ?? false
    }));
  }
  getSupportedProvider(model) {
    return Array.from(this.llmProviders.entries()).filter(([_2, { supportedModel }]) => supportedModel && supportedModel.includes(model)).map(([name]) => name);
  }
}
// Annotate the CommonJS export names for ESM import in node:
0 && (module.exports = {
  AIManager,
  SupportedModel
});
