import { CreateIndexCodeParam } from './Types';

export const getCreateIndexJSCode = (params: CreateIndexCodeParam) => {
  const { collectionName, fieldName, extraParams } = params;

  const jsCode = `import { MilvusClient } from '@zilliz/milvus2-sdk-node';
const client = new MilvusClient(milvus_address);

client.indexManager.createIndex({
  collection_name: '${collectionName}',
  field_name: '${fieldName}',
  extra_params: ${JSON.stringify(extraParams, null, 2)},
});`;

  return jsCode;
};