def parse_arguments()

in rally-custom/custom_tracks/elasticsearch/openai_vector/_tools/parse_documents.py [0:0]


def parse_arguments():
    if len(sys.argv) < 2:
        print(f"Usage: {sys.argv[0]} <input_file_path> [<max_initial_indexing_docs> <max_parallel_indexing_docs>]")
        exit(1)

    if len(sys.argv) == 2:
        return (sys.argv[1], DEFAULT_MAX_INITIAL_INDEXING_DOCS, DEFAULT_MAX_PARALLEL_INDEXING_DOCS)
    elif len(sys.argv) == 3:
        return (sys.argv[1], int(sys.argv[2]), DEFAULT_MAX_PARALLEL_INDEXING_DOCS)
    elif len(sys.argv) >= 4:
        return (sys.argv[1], int(sys.argv[2]), int(sys.argv[3]))