Added --verify-ssl #664

This commit is contained in:
Paul Gauthier 2024-06-21 07:46:34 -07:00
parent b03d739065
commit d73fddccb9
2 changed files with 10 additions and 0 deletions

View file

@ -158,6 +158,12 @@ def get_parser(default_config_files, git_root):
env_var="OPENAI_ORGANIZATION_ID", env_var="OPENAI_ORGANIZATION_ID",
help="Specify the OpenAI organization ID", help="Specify the OpenAI organization ID",
) )
group.add_argument(
"--verify-ssl",
action=argparse.BooleanOptionalAction,
default=True,
help="Verify the SSL cert when connecting to models (default: True)",
)
group.add_argument( group.add_argument(
"--model-metadata-file", "--model-metadata-file",
metavar="MODEL_FILE", metavar="MODEL_FILE",

View file

@ -5,6 +5,7 @@ import sys
from pathlib import Path from pathlib import Path
import git import git
import httpx
from dotenv import load_dotenv from dotenv import load_dotenv
from prompt_toolkit.enums import EditingMode from prompt_toolkit.enums import EditingMode
from streamlit.web import cli from streamlit.web import cli
@ -240,6 +241,9 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
parser = get_parser(default_config_files, git_root) parser = get_parser(default_config_files, git_root)
args = parser.parse_args(argv) args = parser.parse_args(argv)
if not args.verify_ssl:
litellm.client_session = httpx.Client(verify=False)
if args.gui and not return_coder: if args.gui and not return_coder:
launch_gui(argv) launch_gui(argv)
return return