feat: 提交审批单的时候通过AI核验报销是否符合报销规范

2025-09-08 21:27:27 +08:00
parent 48529c38c1
commit f320682c49
3 changed files with 96 additions and 0 deletions
@@ -0,0 +1,90 @@
+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