74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
|
"""MinerU File转Markdown服务的命令行界面。"""
|
|||
|
|
|
|||
|
|
import sys
|
|||
|
|
import argparse
|
|||
|
|
|
|||
|
|
from . import config
|
|||
|
|
from . import server
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
"""命令行界面的入口点。"""
|
|||
|
|
parser = argparse.ArgumentParser(description="MinerU File转Markdown转换服务")
|
|||
|
|
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--output-dir", "-o", type=str, help="保存转换后文件的目录 (默认: ./downloads)"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--transport",
|
|||
|
|
"-t",
|
|||
|
|
type=str,
|
|||
|
|
default="stdio",
|
|||
|
|
help="协议类型 (默认: stdio,可选: sse,streamable-http)",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--port",
|
|||
|
|
"-p",
|
|||
|
|
type=int,
|
|||
|
|
default=8001,
|
|||
|
|
help="服务器端口 (默认: 8001, 仅在使用HTTP协议时有效)",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--host",
|
|||
|
|
type=str,
|
|||
|
|
default="127.0.0.1",
|
|||
|
|
help="服务器主机地址 (默认: 127.0.0.1, 仅在使用HTTP协议时有效)",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
# 检查参数有效性
|
|||
|
|
if args.transport == "stdio" and (args.host != "127.0.0.1" or args.port != 8001):
|
|||
|
|
print("警告: 在STDIO模式下，--host和--port参数将被忽略", file=sys.stderr)
|
|||
|
|
|
|||
|
|
# 验证API密钥 - 移动到这里，以便 --help 等参数可以无密钥运行
|
|||
|
|
if not config.MINERU_API_KEY:
|
|||
|
|
print(
|
|||
|
|
"错误: 启动服务需要 MINERU_API_KEY 环境变量。"
|
|||
|
|
"\\n请检查是否已设置该环境变量，例如："
|
|||
|
|
"\\n export MINERU_API_KEY='your_actual_api_key'"
|
|||
|
|
"\\n或者，确保在项目根目录的 `.env` 文件中定义了该变量。"
|
|||
|
|
"\\n\\n您可以使用 --help 查看可用的命令行选项。",
|
|||
|
|
file=sys.stderr, # 将错误消息输出到 stderr
|
|||
|
|
)
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# 如果提供了输出目录，则进行设置
|
|||
|
|
if args.output_dir:
|
|||
|
|
server.set_output_dir(args.output_dir)
|
|||
|
|
|
|||
|
|
# 打印配置信息
|
|||
|
|
print("MinerU File转Markdown转换服务启动...")
|
|||
|
|
if args.transport in ["sse", "streamable-http"]:
|
|||
|
|
print(f"服务器地址: {args.host}:{args.port}")
|
|||
|
|
print("按 Ctrl+C 可以退出服务")
|
|||
|
|
|
|||
|
|
server.run_server(mode=args.transport, port=args.port, host=args.host)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|