fix: executor push on windows

This commit is contained in:
Florian Hönicke
2023-04-22 22:34:15 +02:00
parent 202239af2a
commit ba75e2786e

View File

@@ -7,15 +7,15 @@ import threading
import time import time
import webbrowser import webbrowser
from pathlib import Path from pathlib import Path
from typing import Dict
import click import click
import hubble import hubble
import requests import requests
from hubble.executor.helper import upload_file, archive_package, get_request_header from hubble.executor.helper import upload_file, archive_package, get_full_version
from jcloud.flow import CloudFlow from jcloud.flow import CloudFlow
from jina import Flow from jina import Flow
from src.apis.gpt import configure_openai_api_key
from src.constants import DEMO_TOKEN from src.constants import DEMO_TOKEN
from src.utils.io import suppress_stdout, is_docker_running from src.utils.io import suppress_stdout, is_docker_running
from src.utils.string_tools import print_colored from src.utils.string_tools import print_colored
@@ -71,6 +71,21 @@ def push_executor(dir_path):
print(f'connection error - retrying in 5 seconds...') print(f'connection error - retrying in 5 seconds...')
time.sleep(5) time.sleep(5)
def get_request_header() -> Dict:
"""Return the header of request with an authorization token.
:return: request header
"""
metas, envs = get_full_version()
headers = {
**{f'jinameta-{k}': str(v) for k, v in metas.items()},
**envs,
}
headers['Authorization'] = f'token {DEMO_TOKEN}'
return headers
def _push_executor(dir_path): def _push_executor(dir_path):
dir_path = Path(dir_path) dir_path = Path(dir_path)
md5_hash = hashlib.md5() md5_hash = hashlib.md5()
@@ -88,7 +103,6 @@ def _push_executor(dir_path):
} }
with suppress_stdout(): with suppress_stdout():
headers = get_request_header() headers = get_request_header()
headers['Authorization'] = f'token {DEMO_TOKEN}'
resp = upload_file( resp = upload_file(
'https://api.hubble.jina.ai/v2/rpc/executor.push', 'https://api.hubble.jina.ai/v2/rpc/executor.push',