com.alibaba.dashscope.threads.runs.StepDetailBase Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dashscope-sdk-java Show documentation

The newest version!

package com.alibaba.dashscope.threads.runs;

import com.alibaba.dashscope.common.TypeRegistry;
import lombok.Data;

@Data
public class StepDetailBase {
  private static final TypeRegistry stepDetailRegistry = new TypeRegistry<>();

  protected static synchronized void registerStepDetail(
      String type, Class clazz) {
    stepDetailRegistry.register(type, clazz);
  }

  public static synchronized Class getStepDetailClass(String type) {
    return stepDetailRegistry.get(type);
  }

  static {
    registerStepDetail("message_creation", StepMessageCreation.class);
    registerStepDetail("tool_calls", StepToolCalls.class);
  }

  private String type;
}