githubmirror/UFO | Gitverse

learner/learn.py

47 строк966 B

vyokky

re-format

06 май 2024, 12:09

06 май 2024, 12:098c9fe21

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.

import argparse
from . import indexer


args = argparse.ArgumentParser()
args.add_argument(
    "--app", help="The name of application to learn.", type=str, default="./"
)
args.add_argument(
    "--docs", help="The help application of the app.", type=str, default="./"
)
args.add_argument(
    "--format", help="The format of the help doc.", type=str, default="xml"
)
args.add_argument(
    "--incremental", action="store_true", help="Enable incremental update."
)
args.add_argument(
    "--save_path",
    help="The format of the help doc.",
    type=str,
    default="./vectordb/docs/",
)


parsed_args = args.parse_args()


def main():
    """
    Main function.
    """

    indexer.create_indexer(
        parsed_args.app,
        parsed_args.docs,
        parsed_args.format,
        parsed_args.incremental,
        parsed_args.save_path,
    )


if __name__ == "__main__":
    main()