third_codes/eland
1
0
Fork 0
mirror of https://github.com/elastic/eland.git synced 2025-07-11 00:02:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add --ca-certs and --insecure option for configuring TLS

Browse Source
This commit is contained in:
David Kyle 2022-03-08 21:44:13 +00:00 committed by GitHub
parent abd05df50b
commit 5c5e5af54d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
bin/eland_import_hub_model
View File

@ -29,6 +29,7 @@ import argparse
import tempfile
import elasticsearch
from elastic_transport.client_utils import DEFAULT
from eland.ml.pytorch import PyTorchModel
from eland.ml.pytorch.transformers import SUPPORTED_TASK_TYPES, TransformerModel
@ -81,9 +82,21 @@ def main():
default=False,
help="Should the model previously stored with `elasticsearch-model-id` be deleted"
)
parser.add_argument(
"--insecure",
action="store_false",
default=True,
help="Do not verify SSL certificates"
)
parser.add_argument(
"--ca-certs",
required=False,
default=DEFAULT,
help="Path to CA bundle"
)
args = parser.parse_args()
es = elasticsearch.Elasticsearch(args.url, timeout=300) # 5 minute timeout
es = elasticsearch.Elasticsearch(args.url, request_timeout=300, verify_certs=args.insecure, ca_certs=args.ca_certs) # 5 minute timeout
# trace and save model, then upload it from temp file
with tempfile.TemporaryDirectory() as tmp_dir: