73db1a5a53
<!-- .github/pull_request_template.md --> ## Description Introducing scructlog. ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import argparse
|
|
import asyncio
|
|
from cognee.shared.logging_utils import get_logger, ERROR
|
|
|
|
from cognee.api.v1.cognify.code_graph_pipeline import run_code_graph_pipeline
|
|
|
|
|
|
async def main(repo_path, include_docs):
|
|
run_status = False
|
|
async for run_status in run_code_graph_pipeline(repo_path, include_docs=include_docs):
|
|
run_status = run_status
|
|
|
|
return run_status
|
|
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--repo_path", type=str, required=True, help="Path to the repository")
|
|
parser.add_argument(
|
|
"--include_docs",
|
|
type=lambda x: x.lower() in ("true", "1"),
|
|
default=True,
|
|
help="Whether or not to process non-code files",
|
|
)
|
|
parser.add_argument(
|
|
"--time",
|
|
type=lambda x: x.lower() in ("true", "1"),
|
|
default=True,
|
|
help="Whether or not to time the pipeline run",
|
|
)
|
|
return parser.parse_args()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logger = get_logger(level=ERROR)
|
|
|
|
args = parse_args()
|
|
|
|
if args.time:
|
|
import time
|
|
|
|
start_time = time.time()
|
|
asyncio.run(main(args.repo_path, args.include_docs))
|
|
end_time = time.time()
|
|
print("\n" + "=" * 50)
|
|
print(f"Pipeline Execution Time: {end_time - start_time:.2f} seconds")
|
|
print("=" * 50 + "\n")
|
|
else:
|
|
asyncio.run(main(args.repo_path, args.include_docs))
|