async def foo(body: JinaRequestModel):
from jina.enums import DataInputType
bd = body.dict() if body else {'data': None}
bd['exec_endpoint'] = exec_endpoint
req_generator_input = bd
req_generator_input['data_type'] = DataInputType.DICT
if bd['data'] is not None and 'docs' in bd['data']:
req_generator_input['data'] = req_generator_input['data'][
'docs']
result = await _get_singleton_result(
request_generator(**req_generator_input))
return result
def test_models_fail(bad_docs):
import pydantic
with pytest.raises(pydantic.error_wrappers.ValidationError):
_ = JinaRequestModel(data=bad_docs)
def test_models_with_docs():
_ = JinaRequestModel(data={'docs': [{'text': 'abc'}]})
def test_model_with_top_k(top_k):
m = JinaRequestModel(data=[{'text': 'abc'}], parameters={'top_k': top_k})
assert m.parameters['top_k'] == top_k
m = JinaRequestModel(parameters={'top_k': top_k})
assert m.parameters['top_k'] == top_k