io.codemodder.plugins.llm.EnvironmentBasedModelMapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of codemodder-plugin-llm Show documentation

Codemod plugin for augmenting transformation with LLM assisted analysis and fixes

There is a newer version: 0.97.6

Show newest version

package io.codemodder.plugins.llm;

import java.util.HashMap;
import java.util.Map;

/** Mapper that maps models to their deployment names based on environment variables. */
final class EnvironmentBasedModelMapper implements ModelMapper {

  private final HashMap map = new HashMap<>();

  EnvironmentBasedModelMapper() {
    this(System.getenv());
  }

  EnvironmentBasedModelMapper(final Map environment) {
    for (final Model model : StandardModel.values()) {
      final var name = String.format(DEPLOYMENT_TEMPLATE, toEnvironmentVariableCase(model.id()));
      final var deployment = environment.getOrDefault(name, model.id());
      map.put(model, deployment);
    }
  }

  @Override
  public String getModelName(Model model) {
    return map.getOrDefault(model, model.id());
  }

  /** Converts a model ID to environment variable casing. */
  private static String toEnvironmentVariableCase(String input) {
    return input.toUpperCase().replace('-', '_').replace('.', '_');
  }

  private static final String DEPLOYMENT_TEMPLATE = "CODEMODDER_AZURE_OPENAI_%s_DEPLOYMENT";
}