From 7f909332fcb7042c71b39870d6892583fd6b7a10 Mon Sep 17 00:00:00 2001 From: Vanshika Vanshika Date: Fri, 6 Feb 2026 00:21:54 +0530 Subject: [PATCH] feat(cli): Add --scenario option to feast init for RAG template Signed-off-by: Vanshika Vanshika rh-pre-commit.version: 2.3.2 rh-pre-commit.check-secrets: ENABLED --- sdk/python/feast/cli/cli.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/sdk/python/feast/cli/cli.py b/sdk/python/feast/cli/cli.py index 4eff2707af9..d187c13d349 100644 --- a/sdk/python/feast/cli/cli.py +++ b/sdk/python/feast/cli/cli.py @@ -447,6 +447,7 @@ def materialize_incremental_command(ctx: click.Context, end_ts: str, views: List "milvus", "ray", "ray_rag", + "rag", "pytorch_nlp", ], case_sensitive=False, @@ -458,13 +459,23 @@ def materialize_incremental_command(ctx: click.Context, end_ts: str, views: List "--repo-path", help="Directory path where the repository will be created (default: create subdirectory with project name)", ) -def init_command(project_directory, minimal: bool, template: str, repo_path: str): +@click.option( + "--scenario", + type=click.Choice(["driver", "rag"], case_sensitive=False), + default="driver", + help="Scenario/demo to create: driver (classic driver stats) or rag (RAG with vector search).", +) +def init_command( + project_directory, minimal: bool, template: str, repo_path: str, scenario: str +): """Create a new Feast repository""" if not project_directory: project_directory = generate_project_name() if minimal: template = "minimal" + elif scenario == "rag": + template = "rag" init_repo(project_directory, template, repo_path)