def predict()

in retail/recommendation-system/bqml-scann/index_server/main.py [0:0]

• 18 lines of code
• 4 McCabe index (conditional complexity)

def predict(model, version):
  result = 'predictions'
  try:
    data = request.get_json()['instances'][0]
    query = data.get('query', None)
    show = data.get('show', 10)
    if not str(show).isdigit(): show = 10

    is_valid, error = validate_request(query, show)

    if not is_valid: 
      value = error
    else:
      vector = embedding_lookup.lookup([query])[0]
      value = scann_matcher.match(vector, int(show))

  except Exception as error:
    value = 'Unexpected error: {}'.format(error)
    result = 'error'

  response = jsonify({result: value})
  return response