DB-GPT/web/components/chat/ob-editor/service.ts

import type { IModelOptions } from '@oceanbase-odc/monaco-plugin-ob/dist/type';
import { ISession } from '../monaco-editor';

export function getModelService(
  { _modelId, delimiter }: { _modelId: string; delimiter: string },
  session?: () => ISession | null,
): IModelOptions {
  return {
    delimiter,
    async getTableList(schemaName?: string) {
      return session?.()?.getTableList(schemaName) || [];
    },
    async getTableColumns(tableName: string, _dbName?: string) {
      return session?.()?.getTableColumns(tableName) || [];
    },
    async getSchemaList() {
      return session?.()?.getSchemaList() || [];
    },
  };
}