コード例 #1
0
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
コード例 #2
0
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()
コード例 #3
0
ファイル: app_py.py プロジェクト: jina-ai/legacy-examples
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()
コード例 #4
0
ファイル: app.py プロジェクト: ankitcdry1612/examples
def query_restful():
    f = Flow().load_config('flow-query.yml')
    f.use_rest_gateway()
    with f:
        f.block()