Chauhan commited on
Commit
7ec877b
·
verified ·
1 Parent(s): 20a124c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,14 +43,14 @@ def main():
43
  rag_subparser = rag_parser.add_subparsers(title='subcommand', dest='subcommand', description='Choose a sub command to start the app.')
44
  rag_start_parser = rag_subparser.add_parser("start", help="Start the rag app")
45
  rag_start_parser.add_argument('--assistant-type', type=str,
46
- choices=["healthcare"],
47
  default="default",
48
  help='Assistant Type')
49
 
50
  args = parser.parse_args()
51
  args.command = 'rag'
52
  args.subcommand = 'start'
53
- args.assistant_type = 'default'
54
 
55
  if args.command == 'rag' and args.subcommand == 'start':
56
  _start_vuln_rag(args)
 
43
  rag_subparser = rag_parser.add_subparsers(title='subcommand', dest='subcommand', description='Choose a sub command to start the app.')
44
  rag_start_parser = rag_subparser.add_parser("start", help="Start the rag app")
45
  rag_start_parser.add_argument('--assistant-type', type=str,
46
+ choices=["healthcare", "diabetes"],
47
  default="default",
48
  help='Assistant Type')
49
 
50
  args = parser.parse_args()
51
  args.command = 'rag'
52
  args.subcommand = 'start'
53
+ args.assistant_type = 'healthcare'
54
 
55
  if args.command == 'rag' and args.subcommand == 'start':
56
  _start_vuln_rag(args)