def rank()

in experiments/legacy/backend/category.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def rank(desc: str, candidates: list[list[str]]) -> list[list[str]]:
  """Use an LLM to rank candidates by description.
  
  Args:
    desc: user provided description of product
    candidates: list of categories. Each category is in list form 
      e.g. ['level 1 category', 'level 2 category'] so it's a list of lists
  
  Returns:
    The candidates ranked by the LLM from most to least relevant. If there are
    duplicate candidates the list is deduped prior to returning
  """
  try:
    return _rank(desc, candidates)
  except ValueError as e:
    logging.error(e)
    logging.error('Falling back to original candidate ranking.')
    return list(dict.fromkeys([tuple(l) for l in candidates]))