org.dromara.pdf.pdfbox.core.ext.parser.ai

Class AbstractAIParser

java.lang.Object

org.dromara.pdf.pdfbox.core.ext.AbstractExpander

org.dromara.pdf.pdfbox.core.ext.parser.AbstractParser

org.dromara.pdf.pdfbox.core.ext.parser.ai.AbstractAIParser

Direct Known Subclasses:

ALiAIParser, BaiDuAIParser, ByteDanceAIParser, DeepSeekAIParser, KunLunWanWeiAIParser, MoonshotAIParser, OSChinaAIParser, TencentAIParser, XunFeiAIParser, ZhiPuAIParser

public abstract class AbstractAIParser
extends AbstractParser

抽象AI解析器

Since:

1.8

Copyright (c) 2020 xsx All Rights Reserved. x-easypdf-pdfbox is licensed under Mulan PSL v2. You can use this software according to the terms and conditions of the Mulan PSL v2. You may obtain a copy of Mulan PSL v2 at: http://license.coscl.org.cn/MulanPSL2 THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE. See the Mulan PSL v2 for more details.

Author:

xsx

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
protected static class	AbstractAIParser.ImageContent 图像消息内容
protected static class	AbstractAIParser.ImageUrl 图像地址
protected static class	AbstractAIParser.Message 消息
protected static class	AbstractAIParser.TextContent 文本消息内容

Field Summary

Fields

Modifier and Type	Field and Description
protected AIParserConfig	config ai解析配置
protected String	taskIdKey 任务id key

Fields inherited from class org.dromara.pdf.pdfbox.core.ext.AbstractExpander

document, log

Constructor Summary

Constructors

Constructor and Description
AbstractAIParser(Document document, String ak, String sk, boolean isJsonResponse) 有参构造

Method Summary

Modifier and Type	Method and Description
protected List<Object>	createImageMessageContents(String prompt, BufferedImage image) 创建图像消息内容
protected List<AbstractAIParser.Message>	createImageMessages(String prompt, BufferedImage image) 创建图像消息
protected List<AbstractAIParser.Message>	createTextMessages(String prompt, String text) 创建消息
protected String	getAuthorization() 获取授权
protected BufferedImage	getPageImage(int pageIndex) 获取页面图像
protected BufferedImage	getPageImage(int pageIndex, int imageIndex) 获取页面图像
protected String	getPageText(int beginPageIndex, int endPageIndex) 获取页面文本
protected AIParseInfo	getParseInfo(com.alibaba.fastjson2.JSONObject response) 获取解析信息
protected com.alibaba.fastjson2.JSONObject	getRequestBody(String model, List<AbstractAIParser.Message> messages) 获取请求体
protected okhttp3.Headers	getRequestHeaders() 获取请求头
protected com.alibaba.fastjson2.JSONObject	getResponse(String url, String model, List<AbstractAIParser.Message> messages) 获取响应
protected AIParseInfo	parseImageInfo(List<AbstractAIParser.Message> messages) 解析图像信息
AIParseInfo	parseImageWithPage(String prompt, int pageIndex, int imageIndex) 解析图像（页面中的图像）
protected AIParseInfo	parseInfo(String url, String model, List<AbstractAIParser.Message> messages) 解析
AIParseInfo	parsePageWithImage(String prompt, int pageIndex) 解析页面（整个页面）
protected AIParseInfo	parseTexInfo(List<AbstractAIParser.Message> messages) 解析文本信息
AIParseInfo	parseTextWithDocument(String prompt) 解析文本（文档）
AIParseInfo	parseTextWithPage(String prompt, int beginPageIndex, int endPageIndex) 解析文本（页面）
void	setHttpClient(okhttp3.OkHttpClient client) 设置http客户端
void	setImageModel(String model) 设置图像模型
void	setImageSystemPrompt(String prompt) 设置图像系统提示词
void	setIsJsonResponse(boolean flag) 是否json响应
void	setTextModel(String model) 设置文本模型
void	setTextSystemPrompt(String prompt) 设置文本系统提示词

Methods inherited from class org.dromara.pdf.pdfbox.core.ext.AbstractExpander

getDocument

Methods inherited from class java.lang.Object

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

Field Detail

taskIdKey

protected String taskIdKey

任务id key

config

protected AIParserConfig config

ai解析配置

Constructor Detail

AbstractAIParser

public AbstractAIParser(Document document,
                        String ak,
                        String sk,
                        boolean isJsonResponse)

有参构造

Parameters:

document - 文档

ak - api密钥

sk - 密钥

isJsonResponse - 是否json响应

Method Detail

setHttpClient

public void setHttpClient(okhttp3.OkHttpClient client)

设置http客户端

Parameters:

client - 客户端

setImageModel

public void setImageModel(String model)

设置图像模型

Parameters:

model - 模型

setImageSystemPrompt

public void setImageSystemPrompt(String prompt)

设置图像系统提示词

Parameters:

prompt - 提示词

setTextModel

public void setTextModel(String model)

设置文本模型

Parameters:

model - 模型

setTextSystemPrompt

public void setTextSystemPrompt(String prompt)

设置文本系统提示词

Parameters:

prompt - 提示词

setIsJsonResponse

public void setIsJsonResponse(boolean flag)

是否json响应

Parameters:

flag - 标记

parsePageWithImage

public AIParseInfo parsePageWithImage(String prompt,
                                      int pageIndex)

解析页面（整个页面）

注：访问大模型超时时间为1分钟

Parameters:

prompt - 提示词

pageIndex - 页面索引

Returns:

返回解析内容

parseImageWithPage

public AIParseInfo parseImageWithPage(String prompt,
                                      int pageIndex,
                                      int imageIndex)

解析图像（页面中的图像）

注：访问大模型超时时间为1分钟

Parameters:

prompt - 提示词

pageIndex - 页面索引

imageIndex - 图像索引

Returns:

返回解析信息

parseTextWithDocument

public AIParseInfo parseTextWithDocument(String prompt)

解析文本（文档）

注：访问大模型超时时间为1分钟

Parameters:

prompt - 提示词

Returns:

返回解析信息

parseTextWithPage

public AIParseInfo parseTextWithPage(String prompt,
                                     int beginPageIndex,
                                     int endPageIndex)

解析文本（页面）

注：访问大模型超时时间为1分钟

Parameters:

prompt - 提示词

beginPageIndex - 起始页面索引

endPageIndex - 结束页面索引

Returns:

返回解析信息

parseImageInfo

protected AIParseInfo parseImageInfo(List<AbstractAIParser.Message> messages)

解析图像信息

Parameters:

messages - 消息

Returns:

返回解析信息

parseTexInfo

protected AIParseInfo parseTexInfo(List<AbstractAIParser.Message> messages)

解析文本信息

Parameters:

messages - 消息

Returns:

返回解析信息

parseInfo

protected AIParseInfo parseInfo(String url,
                                String model,
                                List<AbstractAIParser.Message> messages)

解析

Parameters:

url - 请求路径

model - 模型

messages - 消息

Returns:

返回解析信息

createImageMessages

protected List<AbstractAIParser.Message> createImageMessages(String prompt,
                                                             BufferedImage image)

创建图像消息

Parameters:

prompt - 提示词

image - 图像

Returns:

返回消息

createImageMessageContents

protected List<Object> createImageMessageContents(String prompt,
                                                  BufferedImage image)

创建图像消息内容

Parameters:

prompt - 提示词

image - 图像

Returns:

返回消息内容

createTextMessages

protected List<AbstractAIParser.Message> createTextMessages(String prompt,
                                                            String text)

创建消息

Parameters:

prompt - 提示词

text - 文本

Returns:

返回消息

getAuthorization

protected String getAuthorization()

获取授权

Returns:

返回授权

getRequestHeaders

protected okhttp3.Headers getRequestHeaders()

获取请求头

Returns:

返回请求头

getRequestBody

protected com.alibaba.fastjson2.JSONObject getRequestBody(String model,
                                                          List<AbstractAIParser.Message> messages)

获取请求体

Parameters:

model - 模型

messages - 消息

Returns:

返回请求体

getResponse

protected com.alibaba.fastjson2.JSONObject getResponse(String url,
                                                       String model,
                                                       List<AbstractAIParser.Message> messages)

获取响应

Parameters:

url - 地址

model - 模型

messages - 消息

Returns:

返回响应

getParseInfo

protected AIParseInfo getParseInfo(com.alibaba.fastjson2.JSONObject response)

获取解析信息

Parameters:

response - 响应

Returns:

返回解析信息

getPageImage

protected BufferedImage getPageImage(int pageIndex)

获取页面图像

注：页面转图像

Parameters:

pageIndex - 页面索引

Returns:

返回页面图像

getPageImage

protected BufferedImage getPageImage(int pageIndex,
                                     int imageIndex)

获取页面图像

注：页面中指定索引的图像

Parameters:

pageIndex - 页面索引

Returns:

返回页面图像

getPageText

protected String getPageText(int beginPageIndex,
                             int endPageIndex)

获取页面文本

Parameters:

beginPageIndex - 起始页面索引

endPageIndex - 结束页面索引

Returns:

返回文本