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