com.huaweicloud.pangu.dev.sdk.skill.document

Class DocMapReduceSkill

java.lang.Object

com.huaweicloud.pangu.dev.sdk.skill.AbstractDocSkill

com.huaweicloud.pangu.dev.sdk.skill.document.DocMapReduceSkill

All Implemented Interfaces:

Skill, DocSkill

Direct Known Subclasses:

DocAskMapReduceSkill, DocSummarizeMapReduceSkill

public class DocMapReduceSkill
extends AbstractDocSkill

MapReduce策略

Since:

2023-07-08

Nested Class Summary

Nested classes/interfaces inherited from class com.huaweicloud.pangu.dev.sdk.skill.AbstractDocSkill

AbstractDocSkill.PromptParam

Constructor Summary

Constructors

Constructor and Description
DocMapReduceSkill(LLM llm) 初始化

Method Summary

Modifier and Type	Method and Description
String	execute(List<Document> docs, String question, int reduceMaxToken) 文档处理（MapRefine策略），支持配置每次组合reduce的token数限制；值需小于模型最大支持token数
String	executeWithDocs(List<Document> docs, String question) 文档问答
void	setMapPrompt(String promptTemplate) map提示词
void	setReducePrompt(String promptTemplate) reduce提示词

Methods inherited from class com.huaweicloud.pangu.dev.sdk.skill.AbstractDocSkill

execute, execute, executeWithDocs, executeWithTexts, executeWithTexts

Methods inherited from class java.lang.Object

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

Constructor Detail

DocMapReduceSkill

public DocMapReduceSkill(LLM llm)

初始化

Parameters:

llm - 模型

Method Detail

setMapPrompt

public void setMapPrompt(String promptTemplate)

map提示词

Parameters:

promptTemplate - 提示词

setReducePrompt

public void setReducePrompt(String promptTemplate)

reduce提示词

Parameters:

promptTemplate - 提示词

executeWithDocs

public String executeWithDocs(List<Document> docs,
                              String question)

Description copied from interface: DocSkill

文档问答

Parameters:

docs - 文档

question - 问题

Returns:

回答

execute

public String execute(List<Document> docs,
                      String question,
                      int reduceMaxToken)

文档处理（MapRefine策略），支持配置每次组合reduce的token数限制；值需小于模型最大支持token数

Parameters:

docs - 文档

question - 问题

reduceMaxToken - 每次组合reduce的最大token数

Returns:

回答