org.dromara.hutool.ai.model.hutool

Class HutoolServiceImpl

java.lang.Object

org.dromara.hutool.ai.core.BaseAIService

org.dromara.hutool.ai.model.hutool.HutoolServiceImpl

All Implemented Interfaces:

AIService, HutoolService

public class HutoolServiceImpl
extends BaseAIService
implements HutoolService

Hutool服务，AI具体功能的实现

Since:

6.0.0

Author:

elichow

Field Summary

Fields inherited from class org.dromara.hutool.ai.core.BaseAIService

config

Constructor Summary

Constructors

Constructor and Description
HutoolServiceImpl(AIConfig config)

Method Summary

Modifier and Type	Method and Description
String	chat(List<Message> messages) 对话
void	chat(List<Message> messages, Consumer<String> callback) 对话-SSE流式输出
String	chatVision(String prompt, List<String> images, String detail) 图像理解：模型会依据传入的图片信息以及问题，给出回复。
void	chatVision(String prompt, List<String> images, String detail, Consumer<String> callback) 图像理解-SSE流式输出
String	embeddingVision(String text, String image) 图文向量化：仅支持单一文本、单张图片或文本与图片的组合输入（即一段文本 + 一张图片），暂不支持批量文本 / 图片的同时处理
String	getVideoTasksInfo(String taskId) 查询视频生成任务信息
String	imagesGenerations(String prompt) 文生图
String	stt(File file) STT音频转文本
String	tokenizeText(String text) 分词：可以将文本转换为模型可理解的 token 信息
InputStream	tts(String input, HutoolCommon.HutoolSpeech voice) TTS文本转语音
String	videoTasks(String text, String image, List<HutoolCommon.HutoolVideo> videoParams) 创建视频生成任务

Methods inherited from class org.dromara.hutool.ai.core.BaseAIService

sendFormData, sendGet, sendPost, sendPostStream

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.dromara.hutool.ai.model.hutool.HutoolService

chatVision, chatVision, tts, videoTasks

Methods inherited from interface org.dromara.hutool.ai.core.AIService

chat, chat

Constructor Detail

HutoolServiceImpl

public HutoolServiceImpl(AIConfig config)

Method Detail

chat

public String chat(List<Message> messages)

Description copied from interface: AIService

对话

Specified by:

chat in interface AIService

Parameters:

messages - 由目前为止的对话组成的消息列表，可以设置role，content。详细参考官方文档

Returns:

AI回答

chat

public void chat(List<Message> messages,
                 Consumer<String> callback)

Description copied from interface: AIService

对话-SSE流式输出

Specified by:

chat in interface AIService

Parameters:

messages - 由目前为止的对话组成的消息列表，可以设置role，content。详细参考官方文档

callback - 流式数据回调函数

chatVision

public String chatVision(String prompt,
                         List<String> images,
                         String detail)

Description copied from interface: HutoolService

图像理解：模型会依据传入的图片信息以及问题，给出回复。

Specified by:

chatVision in interface HutoolService

Parameters:

prompt - 题词

images - 图片列表/或者图片Base64编码图片列表(URI形式)

detail - 手动设置图片的质量，取值范围high、low、auto,默认为auto

Returns:

AI回答

chatVision

public void chatVision(String prompt,
                       List<String> images,
                       String detail,
                       Consumer<String> callback)

Description copied from interface: HutoolService

图像理解-SSE流式输出

Specified by:

chatVision in interface HutoolService

Parameters:

prompt - 题词

images - 图片列表/或者图片Base64编码图片列表(URI形式)

detail - 手动设置图片的质量，取值范围high、low、auto,默认为auto

callback - 流式数据回调函数

tokenizeText

public String tokenizeText(String text)

Description copied from interface: HutoolService

分词：可以将文本转换为模型可理解的 token 信息

Specified by:

tokenizeText in interface HutoolService

Parameters:

text - 需要分词的内容

Returns:

分词结果

imagesGenerations

public String imagesGenerations(String prompt)

Description copied from interface: HutoolService

文生图

Specified by:

imagesGenerations in interface HutoolService

Parameters:

prompt - 题词

Returns:

包含生成图片的url

embeddingVision

public String embeddingVision(String text,
                              String image)

Description copied from interface: HutoolService

图文向量化：仅支持单一文本、单张图片或文本与图片的组合输入（即一段文本 + 一张图片），暂不支持批量文本 / 图片的同时处理

Specified by:

embeddingVision in interface HutoolService

Parameters:

text - 需要向量化的内容

image - 需要向量化的图片地址/或者图片Base64编码图片(URI形式)

Returns:

处理后的向量信息

tts

public InputStream tts(String input,
                       HutoolCommon.HutoolSpeech voice)

Description copied from interface: HutoolService

TTS文本转语音

Specified by:

tts in interface HutoolService

Parameters:

input - 需要转成语音的文本

voice - AI的音色

Returns:

返回的音频mp3文件流

stt

public String stt(File file)

Description copied from interface: HutoolService

STT音频转文本

Specified by:

stt in interface HutoolService

Parameters:

file - 需要转成文本的音频文件

Returns:

返回的文本内容

videoTasks

public String videoTasks(String text,
                         String image,
                         List<HutoolCommon.HutoolVideo> videoParams)

Description copied from interface: HutoolService

创建视频生成任务

Specified by:

videoTasks in interface HutoolService

Parameters:

text - 文本提示词

image - 图片/或者图片Base64编码图片(URI形式)

videoParams - 视频参数列表

Returns:

生成任务id

getVideoTasksInfo

public String getVideoTasksInfo(String taskId)

Description copied from interface: HutoolService

查询视频生成任务信息

Specified by:

getVideoTasksInfo in interface HutoolService

Parameters:

taskId - 通过创建生成视频任务返回的生成任务id

Returns:

生成任务信息