com.alibaba.dashscope.threads.ContentBase 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;

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

/** tool request base */
@Data
public abstract class ContentBase {
  private static final TypeRegistry contentRegistry = new TypeRegistry<>();

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

  public static synchronized Class getContentClass(String type) {
    return contentRegistry.get(type);
  }

  // register official tools for list.
  static {
    registerContent("image_file", ContentImageFile.class);
    registerContent("text", ContentText.class);
  }

  public abstract String getType();
}