• Home
• com.alibaba
• dashscope-sdk-java
• 2.16.9
• source code
• ToolBase.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.alibaba.dashscope.tools.ToolBase Maven / Gradle / Ivy

package com.alibaba.dashscope.tools;

import com.alibaba.dashscope.common.TypeRegistry;
import com.alibaba.dashscope.tools.T2Image.Text2Image;
import com.alibaba.dashscope.tools.codeinterpretertool.ToolCodeInterpreter;
import com.alibaba.dashscope.tools.search.ToolQuarkSearch;
import com.alibaba.dashscope.tools.wanx.ToolWanX;
import lombok.experimental.SuperBuilder;

/** tool request base */
@SuperBuilder
public abstract class ToolBase implements ToolInterface {
  private static final TypeRegistry toolRegistry = new TypeRegistry<>();

  protected static synchronized void registerTool(
      String toolType, Class clazz) {
    toolRegistry.register(toolType, clazz);
  }

  public static synchronized Class getToolClass(String toolType) {
    return toolRegistry.get(toolType);
  }

  // register official tools for list.
  static {
    registerTool("function", ToolFunction.class);
    registerTool("quark_search", ToolQuarkSearch.class);
    registerTool("code_interpreter", ToolCodeInterpreter.class);
    registerTool("wanx", ToolWanX.class);
    registerTool("text_to_image", Text2Image.class);
  }

  @Override
  public abstract String getType();
}