Skip to content

Commit

Permalink
remove duplicate indexing (#103)
Browse files Browse the repository at this point in the history
remove duplicate cli
  • Loading branch information
darthtrevino authored Apr 5, 2024
1 parent 0a357b5 commit 0149d78
Showing 1 changed file with 0 additions and 62 deletions.
62 changes: 0 additions & 62 deletions graphrag/index/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,7 @@

"""The Indexing Engine package root."""

import argparse

from .cache import PipelineCache
from .cli import index_cli
from .config import (
PipelineBlobCacheConfig,
PipelineBlobReportingConfig,
Expand Down Expand Up @@ -120,62 +117,3 @@
"run_pipeline",
"run_pipeline_with_config",
]


if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument(
"--config",
help="The configuration yaml file to use when running the pipeline",
required=False,
type=str,
)
parser.add_argument(
"-v",
"--verbose",
help="Runs the pipeline with verbose logging",
action="store_true",
)
parser.add_argument(
"--memprofile",
help="Runs the pipeline with memory profiling",
action="store_true",
)
parser.add_argument(
"--root",
help="If no configuration is defined, the root directory to use for input data and output data",
# Only required if config is not defined
required=False,
type=str,
)
parser.add_argument(
"--resume",
help="Resume a given data run leveraging Parquet output files.",
# Only required if config is not defined
required=False,
type=str,
)
parser.add_argument(
"--reporter",
help="The progress reporter to use. Valid values are 'rich', 'print', or 'none'",
type=str,
)
parser.add_argument(
"--emit",
help="The data formats to emit, comma-separated. Valid values are 'parquet' and 'csv'. default='parquet,csv'",
type=str,
)
parser.add_argument("--nocache", help="Disable LLM cache.", action="store_true")
args = parser.parse_args()

index_cli(
root=args.root,
verbose=args.verbose,
resume=args.resume,
memprofile=args.memprofile,
nocache=args.nocache,
reporter=args.reporter,
config=args.config,
emit=args.emit,
cli=True,
)

0 comments on commit 0149d78

Please sign in to comment.