|
|
|
@ -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")
|
|
|
|
|