parse.py

import argparse
from utils.log import log


@log
def init_parse() -> argparse.Namespace:
    parser = argparse.ArgumentParser(description='Baize: 将 LLM 集成至你的命令行！')

    parser.add_argument("prompt", nargs="*", type=str, help="命令行 Prompt")

    # 模型参数
    parser.add_argument('-s', '--stream', action='store_true', help='流式输出模式')
    parser.add_argument('-m', '--model', nargs=1, metavar='<model name>', type=str, help='设置要使用的配置名')
    parser.add_argument('-t', '--template', nargs=1, metavar='<template name>', type=str, help='Prompt 模板')
    parser.add_argument('-f', '--file', nargs="*", metavar='<file path>', type=str, help='上传文件')
    parser.add_argument('-i', '--img', nargs="*", metavar='<img path>', type=str, help='上传图片文件')
    parser.add_argument('--set', nargs=1, metavar='<model name>', help='设置默认模型')

    # 基本配置
    parser.add_argument('--init', nargs=1, metavar='<bash>', type=str, help='初始化路径环境变量')
    parser.add_argument('--version', action='store_true', help='查看版本')
    parser.add_argument('--setup', action='store_true', help='配置新模型')
    parser.add_argument('--modellist', action='store_true', help='查看现有模型列表')
    parser.add_argument('--deletemodel', nargs="*", metavar='<model name>', type=str, help='删除已配置模型')
    parser.add_argument('-l', '--list', action='store_true', help='查看现有 Prompt 模板列表')
    parser.add_argument('--context', action='store_true', help='查看 Context Prompt')
    parser.add_argument('--resetcontext', action='store_true', help='删除 Context Prompt')
    parser.add_argument('--setcontext', nargs=1, metavar='Context Prompt', type=str, help='设置 Context Prompt')
    parser.add_argument('--update', action='store_true', help='查看更新')

    # 执行模式
    parser.add_argument('--log', action='store_true', help='调试模式')
    parser.add_argument('-c', '--cli', action='store_true', help='命令行模式')
    parser.add_argument('--clidetail', action='store_true', help='命令行详细模式')
    parser.add_argument('--clikey', action='store_true', help='命令行模式快捷键调用')

    # 输入输出
    parser.add_argument('-C', '--copy', action='store_true', help='输出复制到剪贴板')
    parser.add_argument('-P', '--paste', action='store_true', help='剪贴板作为输入')
    parser.add_argument('-M', '--markdown', action='store_true', help='输出渲染 Markdown')
    parser.add_argument('-H', '--history', action='store_true', help='查看历史上下文')
    parser.add_argument('-o', '--output', nargs=1, metavar='<file path>', help='输出到文件')
    parser.add_argument('-p', '--previous', action='store_true', help='将上次对话记录引入本次对话')

    # Prompt 配置
    parser.add_argument('--createtemplate', nargs="?", metavar='<template path>', const='__default__', type=str, help='创建 Prompt Template')
    parser.add_argument('--deletetemplate', nargs="*", metavar='<template>', type=str, help='删除 Prompt Template')
    parser.add_argument('--showtemplate', nargs=1, metavar='<template>', type=str, help='查看 Prompt Template')

    # Tool 配置
    parser.add_argument('--tool', nargs=1, metavar='<tool name>', help='执行 Tool')
    parser.add_argument('--toollist', action='store_true', help='查看现有工具列表')
    parser.add_argument('--deletetool', nargs="*", metavar='<tool>', type=str, help='删除 Tool')
    parser.add_argument('--createtool', action='store_true', help='创建 Tool')

    # Workflow 配置
    parser.add_argument('-w', '--workflow', nargs=1, metavar='<workflow name>', help='执行 Workflow')
    parser.add_argument('--showworkflow', nargs=1, metavar='<workflow name>', help='查看 Workflow')
    parser.add_argument('--workflowlist', action='store_true', help='查看现有 Workflow 列表')
    parser.add_argument('--createworkflow', action='store_true', help='创建 Workflow')

    args = parser.parse_args()

    return args