|
1 | 1 | import json |
2 | 2 | import os |
3 | | -from argparse import ArgumentParser, Namespace |
4 | | -from typing import Any, Mapping |
| 3 | +from argparse import ArgumentParser, Namespace, _ArgumentGroup |
| 4 | +from typing import Any, Mapping, Union |
5 | 5 | from urllib.parse import quote, urlparse, urlunparse |
6 | 6 |
|
7 | 7 | from redisvl.redis.constants import REDIS_URL_ENV_VAR |
@@ -75,7 +75,9 @@ def create_redis_url(args: Namespace) -> str: |
75 | 75 | return _build_redis_url(args) |
76 | 76 |
|
77 | 77 |
|
78 | | -def _add_redis_connection_args(parser_or_group: ArgumentParser) -> None: |
| 78 | +def _add_redis_connection_args( |
| 79 | + parser_or_group: Union[ArgumentParser, _ArgumentGroup], |
| 80 | +) -> None: |
79 | 81 | """Add Redis connection flags to a parser or argument group.""" |
80 | 82 | parser_or_group.add_argument( |
81 | 83 | "-u", |
@@ -103,9 +105,7 @@ def _add_redis_connection_args(parser_or_group: ArgumentParser) -> None: |
103 | 105 | type=str, |
104 | 106 | default=None, |
105 | 107 | ) |
106 | | - parser_or_group.add_argument( |
107 | | - "--ssl", help="Use SSL for Redis", action="store_true" |
108 | | - ) |
| 108 | + parser_or_group.add_argument("--ssl", help="Use SSL for Redis", action="store_true") |
109 | 109 | parser_or_group.add_argument( |
110 | 110 | "-a", |
111 | 111 | "--password", |
|
0 commit comments