from clean.reparagraph import reparagraph_file
from config import ConfigFactory

OPENAI_BASE_URL = "http://8.218.238.241:17935/v1"
OPENAI_API_KEY = "sk-urFGAQRThR6pysea0aC93bD27fA34bA69811A9254aAaD8B2"
MODEL_NAME = "deepseek-chat"

class ReparagraphConfig(ConfigFactory):
    """配置类"""
    def __init__(self):
        self.remove_refs = True
        self.max_file_size = 50 * 1024 * 1024  # 50MB
        self.dry_run = False
        self.parallel = True
        self.task_name = "result_reparagraph"
        self.openai_base_url = OPENAI_BASE_URL
        self.openai_api_key = OPENAI_API_KEY
        self.model_name = MODEL_NAME
        self.max_retries = 5

config = ReparagraphConfig()
reparagraph_file("/home/ubuntu/50T/LYT/datapipe/result/dify-sty-agent", config)
# reparagraph_file("/home/ubuntu/50T/LYT/datapipe/result/Biocatalytic CsPbX3 Perovskite Nanocrystals A Self-Reporting Nanoprobe for Metabolism Analysis.md", config)