githubmirror/UFO | Gitverse

learner/learn.py

36 строк976 B

vyokky

rag dev

04 мар 2024, 16:06

04 мар 2024, 16:06f921a86

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

import argparse
from . import indexer



# configs = load_config()

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()