Skip to content

Commit

Permalink
update model name constant
Browse files Browse the repository at this point in the history
  • Loading branch information
maxhniebergall committed Jan 9, 2025
1 parent 7a26c92 commit 18dd742
Showing 1 changed file with 4 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -74,11 +74,10 @@ public class ElasticInferenceService extends SenderService {

private static final EnumSet<TaskType> supportedTaskTypes = EnumSet.of(TaskType.SPARSE_EMBEDDING, TaskType.COMPLETION);

public static final String V1_EIS_COMPLETION_MODEL_ID = "temp1";

public static final String DEFAULT_EIS_COMPLETION_ENDPOINT_ID = "eis-alpha-1";
public static final String DEFAULT_EIS_COMPLETION_ENDPOINT_ID_V1 = ".eis-alpha-1";

public static final List<String> DEFAULT_EIS_ENDPOINT_IDS = List.of(DEFAULT_EIS_COMPLETION_ENDPOINT_ID);
public static final List<String> DEFAULT_EIS_ENDPOINT_IDS = List.of(DEFAULT_EIS_COMPLETION_ENDPOINT_ID_V1);

public ElasticInferenceService(
HttpRequestSender.Factory factory,
Expand Down Expand Up @@ -231,7 +230,7 @@ public EnumSet<TaskType> supportedTaskTypes() {

@Override
public List<DefaultConfigId> defaultConfigIds() {
return List.of(new DefaultConfigId(DEFAULT_EIS_COMPLETION_ENDPOINT_ID, TaskType.COMPLETION, this));
return List.of(new DefaultConfigId(DEFAULT_EIS_COMPLETION_ENDPOINT_ID_V1, TaskType.COMPLETION, this));
}

@Override
Expand All @@ -245,7 +244,7 @@ private static ElasticInferenceServiceCompletionModel firstDefaultCompletionMode
serviceSettings.put(MODEL_ID, "elastic-model"); // TODO

return new ElasticInferenceServiceCompletionModel(
DEFAULT_EIS_COMPLETION_ENDPOINT_ID,
DEFAULT_EIS_COMPLETION_ENDPOINT_ID_V1,
TaskType.COMPLETION,
NAME,
serviceSettings,
Expand Down

0 comments on commit 18dd742

Please sign in to comment.