main
HuangHai 2 weeks ago
parent 1ecd82f96e
commit 64312d7220

@ -2,8 +2,8 @@ import asyncio
import os
import shutil
from raganything import RAGAnything, RAGAnythingConfig
from Util.LightRagUtil import create_llm_model_func, create_embedding_func, create_vision_model_func
from Util.LightRagUtil import create_llm_model_func, create_embedding_func, create_vision_model_func, \
format_exam_content
import logging
# 在程序开始时添加以下配置
@ -90,8 +90,7 @@ async def main():
f.write(content)
# 新增:使用大模型整理内容
from Util.LightRagUtil import format_exam_content
formatted_content = format_exam_content(
format_exam_content(
client=llm_model_func(),
raw_text=content,
output_path=os.path.join(output_dir, "整理后的结果.md")

Loading…
Cancel
Save