textPreprocessing.py

def text2prompt(text: str):
  """
    Применяет инструкцию в формате, который распознается
    Mistral Instruct и выводит на выходе входной текст для модели
  """

  instruction = 'Ниже тебе дан текст, содержащий новость на тему экономики. ' \
     'На основе текста ты должен дать наиболее подходящую финансовую рекомендацию из списка ["buy", "sell", "long", "short"]. ' \
     'Твой ответ должен содержать только одно слово, которое будет подходящей рекомендацией из списка.'

  bos_token = "<s>"
  eos_token = "</s>"

  full_prompt = ""
  full_prompt += bos_token
  full_prompt += "### Instruction:"
  full_prompt += "\n" + instruction
  full_prompt += "\n\n### Input:"
  full_prompt += "\n" + text
  full_prompt += "\n\n### Response:"
  full_prompt += eos_token