main
HuangHai 2 weeks ago
parent 1ecd82f96e
commit 64312d7220

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

Loading…
Cancel
Save