def test_flow_rest_reload(): f = Flow().add() f.use_rest_gateway() with f: r = requests.post(f'http://0.0.0.0:{f.port_expose}/reload', json={'targets': ['pod0']}) assert r.status_code == 200
def test_change_gateway(restful, changeto_gateway, mocker): f = Flow(restful=restful).add().add().add(needs='pod1').needs_all() with f: mock = mocker.Mock() f.post('', random_docs(10), on_done=mock) mock.assert_called() mock = mocker.Mock() if changeto_gateway == 'RESTGateway': f.use_rest_gateway() if changeto_gateway == 'GRPCGateway': f.use_grpc_gateway() f.post('', random_docs(10), on_done=mock) mock.assert_called()
def query_restful(): flow = Flow(workspace="workspace", port_expose=os.environ.get('JINA_PORT', str(45678)))\ .add(uses={"jtype": "ImageCrafter", "with": {"target_size": 96, "img_mean": [0.485, 0.456, 0.406], "img_std": [0.229, 0.224, 0.225]}})\ .add(uses=BigTransferEncoder)\ .add(uses={"jtype": "EmbeddingIndexer", "with": {"index_file_name": "image.json"}, "metas": {"name": "vec_idx"}}, name="vec_idx")\ .add(uses={"jtype": "KeyValueIndexer", "metas": {"name": "kv_idx"}}, name="kv_idx")\ .add(uses={"jtype": "MatchImageReader"}) flow.use_rest_gateway() with flow: flow.block()
def query_restful(): f = Flow().load_config('flow-query.yml') f.use_rest_gateway() with f: f.block()