ййй
This commit is contained in:
@@ -10,12 +10,23 @@ _src = _agent_root / "src"
|
||||
if _src.exists() and str(_src) not in sys.path:
|
||||
sys.path.insert(0, str(_src))
|
||||
|
||||
from app.modules.rag.persistence.repository import RagRepository
|
||||
from tests.pipeline_setup_v3.shared.env_loader import load_pipeline_setup_env
|
||||
from tests.pipeline_setup_v3.shared.rag_indexer import RagSessionIndexer
|
||||
|
||||
|
||||
def main() -> int:
|
||||
argv = ["reindex", *sys.argv[1:]]
|
||||
from tests.pipeline_setup.suite_02_pipeline.pipeline_intent_rag.cli import main as cli_main
|
||||
import argparse
|
||||
|
||||
return cli_main(argv)
|
||||
parser = argparse.ArgumentParser(description="Index a repo for pipeline_setup_v3")
|
||||
parser.add_argument("--repo-path", required=True, help="Repository path to index")
|
||||
parser.add_argument("--project-id", default=None, help="Optional project id override")
|
||||
ns = parser.parse_args(sys.argv[1:])
|
||||
|
||||
load_pipeline_setup_env(start_dir=Path(__file__).resolve().parents[1])
|
||||
repo_path = Path(str(ns.repo_path)).expanduser().resolve()
|
||||
RagSessionIndexer(RagRepository()).index_repo(repo_path, project_id=ns.project_id)
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Reference in New Issue
Block a user