23 lines
938 B
Python
23 lines
938 B
Python
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) |