/**
 * 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.
 */

import { Plugin } from '@nocobase/client';
import PluginAIClient from '@nocobase/plugin-ai/client';
import { customLLMProviderOptions } from './llm-providers/custom-llm';

export class PluginCustomLLMClient extends Plugin {
  async afterAdd() {}

  async beforeLoad() {}

  async load() {
    this.aiPlugin.aiManager.registerLLMProvider('custom-llm', customLLMProviderOptions);
  }

  private get aiPlugin(): PluginAIClient {
    return this.app.pm.get('ai');
  }
}

export default PluginCustomLLMClient;