Перейти к основному содержимому

prompt_adapter

PromptAdapter Objects

class PromptAdapter()

Адаптер для работы с промптами, хранящимися в CSV-файле.

Attributes:

  • _prompts dict - Словарь для хранения промптов.
  • file_path str - Путь к файлу с промптами.

__init__

def __init__(file_path: str, file_dir='prompts')

Инициализирует экземпляр PromptAdapter.

Arguments:

  • file_path str - Имя файла с промптами, который находится в папке 'prompts'.

get_prompt

def get_prompt(doc_class: str, question_type: str) -> str

Возвращает промпт для заданного класса документа и типа вопроса.

Arguments:

  • doc_class str - Класс документа.
  • question_type str - Тип вопроса.

Returns:

  • str - Строка с промптом.

Raises:

  • ValueError - Если промпт для заданного класса документа и типа вопроса не найден.