91 lines
3.5 KiB
Python
91 lines
3.5 KiB
Python
import json
|
|
|
|
from langchain_core.output_parsers import StrOutputParser
|
|
from langchain_core.prompts import ChatPromptTemplate
|
|
from pydantic import BaseModel, Field
|
|
|
|
from app.model.KnowledgeDoc import KnowledgeDocServiceWithCreator, KnowledgeDocModel
|
|
from app.model.ReimburseModel import ReimburseModel
|
|
from app.utils.db_utils import AsyncSessionDep
|
|
from app.utils.llm_utils import create_llm
|
|
|
|
|
|
class LlmApproveResult(BaseModel):
|
|
llm_flag: bool = Field(..., description='AI审核是否通过')
|
|
llm_reason: str | None = Field(default=None, description="审核不通过时的原因")
|
|
|
|
|
|
async def get_llm_approve_result(session: AsyncSessionDep, reimburse_cls: ReimburseModel):
|
|
try:
|
|
reimburse_document: KnowledgeDocModel = await KnowledgeDocServiceWithCreator.query_item(session, {"code": "REIMBURSE"})
|
|
except Exception as e:
|
|
raise Exception(f"获取报销规范文档失败: {e}")
|
|
|
|
chain = ChatPromptTemplate.from_template("""
|
|
你是一名专业的报销单审核专家，你需要根据报销规范文档审核报销单信息，报销规范如<document/>标签中的内容所示：
|
|
<document>
|
|
{document_content}
|
|
</document>
|
|
报销单信息如下<reimburse/>标签中的内容所示：
|
|
<reimburse>
|
|
{reimburse_content}
|
|
</reimburse>
|
|
你需要返回一个json对象，类型为：{response_description}，其中llm_flag表示审核是否通过，为布尔值；llm_reason为审核不通过的原因，审核通过reason为空；
|
|
""") | create_llm() | StrOutputParser()
|
|
|
|
print(":::::::::::::::报销规则:::::::::::::::::::")
|
|
print(reimburse_document.content)
|
|
|
|
print(":::::::::::::::报销单描述::::::::::::::::::::::::")
|
|
reimburse_description = get_reimburse_description(reimburse_cls)
|
|
print(reimburse_description)
|
|
|
|
llm_response_content = await chain.ainvoke({
|
|
"document_content": reimburse_document.content,
|
|
"reimburse_content": reimburse_description,
|
|
"response_description": '{"llm_flag":true, "llm_reason":""}'
|
|
})
|
|
|
|
print("::::::::::::::LLM审核结果:::::::::::::::::::::::")
|
|
print(llm_response_content)
|
|
|
|
llm_response_dict = json.loads(llm_response_content)
|
|
return llm_response_dict
|
|
|
|
|
|
def get_reimburse_description(reimrbuse_cls: ReimburseModel):
|
|
if not reimrbuse_cls.project:
|
|
raise Exception("报销单关联项目不存在")
|
|
|
|
reimbruse_content = f"""
|
|
报销申请人：{reimrbuse_cls.user.full_name}
|
|
报销单标题：{str(reimrbuse_cls.title)}
|
|
报销单金额：{str(reimrbuse_cls.amount)}
|
|
报销单备注信息：{str(reimrbuse_cls.remarks)}
|
|
报销项目：{str(reimrbuse_cls.project.name)}
|
|
"""
|
|
|
|
for index, travel_item in enumerate(reimrbuse_cls.travel_list):
|
|
reimbruse_content += f"""
|
|
|
|
第{index + 1}条差旅费用信息：
|
|
差旅费用名称：{travel_item.title}
|
|
差旅费用类型：{travel_item.type}
|
|
差旅出发时间：{travel_item.depart_time}
|
|
差旅出发地点：{travel_item.depart_city}
|
|
差旅到达时间：{travel_item.arrive_time}
|
|
差旅到达地点：{travel_item.arrive_city}
|
|
差旅报销金额：{travel_item.amount}
|
|
"""
|
|
|
|
for index, other_item in enumerate(reimrbuse_cls.other_list):
|
|
reimbruse_content += f"""
|
|
|
|
第{index + 1}条费用信息：
|
|
费用名称：{other_item.title}
|
|
费用类型：{other_item.type}
|
|
报销金额：{other_item.amount}
|
|
票据类型：{other_item.recipe_type}
|
|
"""
|
|
return reimbruse_content
|