From 2d58a2c667a74199fce9883c2a4ce3ac524b1965 Mon Sep 17 00:00:00 2001 From: pavanmantha Date: Sun, 28 Jul 2024 11:10:45 +0530 Subject: [PATCH] -changed the cli options for qdrant --- bootstraprag/cli.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/bootstraprag/cli.py b/bootstraprag/cli.py index 4957363..5ff0e08 100644 --- a/bootstraprag/cli.py +++ b/bootstraprag/cli.py @@ -17,8 +17,9 @@ def create_zip(project_name): @click.command() @click.argument('project_name') -@click.option('--framework', type=click.Choice(['llamaindex', 'langchain', 'haystack', 'None']), - prompt="Which technology would you like to use?") +@click.option('--framework', type=click.Choice(['llamaindex', 'langchain', 'haystack']), + prompt="Which technology would you like to use (leave blank, if you want to use qdrant direct search)?", + default='', required=False) @click.option('--template', type=click.Choice([]), prompt=False) @click.option('--observability', type=click.Choice([]), prompt=False) def create(project_name, framework, template, observability): @@ -27,7 +28,7 @@ def create(project_name, framework, template, observability): if framework == 'llamaindex' or framework == 'langchain' or framework == 'haystack': template_choices = ['simple-rag', 'self-rag', 'rag-with-cot', 'rag-with-react', 'rag-with-hyde'] - elif framework == 'None': + elif framework == '': framework = 'qdrant' template_choices = ['simple-search', 'hybrid-search']