pangukitsappdev.skill.doc package¶
Submodules¶
pangukitsappdev.skill.doc.ask module¶
- class pangukitsappdev.skill.doc.ask.DocAskMapReduceSkill(llm_api: LLMApi, reduce_max_token: int = 2000)¶
-
文档问答refine策略
pangukitsappdev.skill.doc.base module¶
- class pangukitsappdev.skill.doc.base.AbstractDocSkill¶
基类:
Skill
,ABC
文档处理Skill
- execute(inputs: Dict[str, Any]) str ¶
执行Executor :param inputs: 输入的参数,基本上都是用来渲染prompt_template的 :return: 执行结果,LLM的返回结果
- abstract execute_with_docs(docs: List[Document], question: Optional[str] = None) str ¶
文档摘要/问答 Args:
docs: 文档 question: (Optional)问题,如果不传递question,则对文档进行摘要,否则就是文档文档
- Returns:
回答或者摘要
- execute_with_texts(texts: List[str], question: Optional[str] = None) str ¶
文档摘要/问答 Args:
texts: 文档内容 question: (Optional)问题,如果不传递question,则对文档进行摘要,否则就是文档文档
- Returns:
回答或者摘要
- class pangukitsappdev.skill.doc.base.DocMapReduceSkill(map_prompt: BasePromptTemplate, reduce_prompt: BasePromptTemplate, llm_api: LLMApi, reduce_max_token: int = 2000)¶
-
MapReduce策略 Attributes:
map_prompt: Map阶段使用的prompt模板 reduce_prompt: reduce阶段使用的prompt模板 llm_api: 大语言模型的接口 reduce_max_token: (Optional) 默认每次组合reduce最大的token数量,防止超过LLM的参数限制
- class pangukitsappdev.skill.doc.base.DocRefineSkill(qa_prompt: BasePromptTemplate, refine_prompt: BasePromptTemplate, llm_api: LLMApi)¶
-
refine策略
- Attributes:
qa_prompt: 问答的prompt模板 refine_prompt: 精炼的prompt模板 llm_api: llm
pangukitsappdev.skill.doc.summary module¶
- class pangukitsappdev.skill.doc.summary.DocSummaryMapReduceSkill(llm_api: LLMApi, reduce_max_token: int = 2000)¶
-
文档问答MapReduce策略