def test_timeout_repr():
timeout = httpx.TimeoutConfig(timeout=5.0)
assert repr(timeout) == "TimeoutConfig(timeout=5.0)"
timeout = httpx.TimeoutConfig(read_timeout=5.0)
assert repr(timeout) == (
"TimeoutConfig(connect_timeout=None, read_timeout=5.0, "
"write_timeout=None, pool_timeout=None)")
def __init__(self, model: str, model_settings: Dict[str, Any],
server_url: str) -> None:
self.server_url = getenv('ORANGE_EMBEDDING_API_URL', server_url)
self._model = model
self._im_size = model_settings['target_image_size']
# attribute that offers support for cancelling the embedding
# if ran in another thread
self.cancelled = False
self.machine_id = \
QSettings().value('error-reporting/machine-id', '', type=str) \
or str(uuid.getnode())
self.session_id = str(random.randint(1, 1e10))
self._image_loader = ImageLoader()
self._cache = EmbedderCache(model)
self.batch_size = 10000
# default embedding timeouts are too small we need to increase them
self.timeouts = httpx.TimeoutConfig(timeout=60)
self.num_parallel_requests = 0
self.max_parallel = model_settings['batch_size']
def __init__(self, node_manager=None, **kwargs):
if node_manager is None:
node_manager = NodeManager(network='eos')
self.node_manager = node_manager
if node_manager.node_count == 0:
log.info("Node manager database is empty. Adding DEFAULT_NODES.")
node_manager.bulk_insert(*self.DEFAULT_NODES)
# self.current_node = node_manager.weighted_node
while node_manager.weighted_node is None:
log.info(
"[__init__] current_node is None. Waiting a few seconds for last_fail's to get older."
)
time.sleep(3)
# self.current_node = node_manager.weighted_node
self.client = httpx.Client(timeout=30)
httpx.TimeoutConfig(30)
# self.url = self.current_node.url.strip().strip('/')
# self.client = client = httpx.client.Client()
# client.headers['Content-Type'] = 'application/json'
self.max_retries = int(kwargs.pop('max_retries', 10))
self.retry_wait = float(kwargs.pop('retry_wait', 2.0))
def test_timeout_from_config_instance():
timeout = httpx.TimeoutConfig(timeout=5.0)
assert httpx.TimeoutConfig(timeout) == httpx.TimeoutConfig(timeout=5.0)
def test_timeout_from_tuple():
timeout = httpx.TimeoutConfig(timeout=(5.0, 5.0, 5.0))
assert timeout == httpx.TimeoutConfig(timeout=5.0)
def test_timeout_eq():
timeout = httpx.TimeoutConfig(timeout=5.0)
assert timeout == httpx.TimeoutConfig(timeout=5.0)
def test_timeout_from_one_none_value():
timeout = httpx.TimeoutConfig(read_timeout=None)
assert timeout == httpx.TimeoutConfig()
def test_timeout_from_none():
timeout = httpx.TimeoutConfig(timeout=None)
assert timeout == httpx.TimeoutConfig()
def test_timeout_from_nothing():
timeout = httpx.TimeoutConfig()
assert timeout.connect_timeout is None
assert timeout.read_timeout is None
assert timeout.write_timeout is None
