vespa.py

def get_features_from_vespa(row):
  id = row['product_uid']
  query = row['search_term']
  query_tensor = row['query_embedding']
  query_body = {
    'hits': 1,
    'ranking.profile': 'dump',
    'rankfeatures': 'true',
    'type': 'any',
    'recall': '+id:%i' % id, 
    'ranking.features.query(tensor)': query_tensor.tolist(),
    'query': query + " id:>0"
  } 
  url = 'http://localhost:8080/search/'
  r = requests.post(url, json=query_body, headers={'Content-Type':'application/json'})
  json_response = r.json()
  if json_response["root"]["fields"]["totalCount"] != 1:
    return {} 
  hit = json_response["root"]["children"][0]
  rankfeatures = hit['fields']['rankfeatures']
  rankfeatures.update(hit['fields']['summaryfeatures'])
  return rankfeatures