niuyazhe
feature(nyz): add level1 questions
f18dd2a
raw
history blame
1.2 kB
from typing import Tuple
from .question import Question
from ..llms import get_llm_fn
class QuestionExecutor:
def __init__(self, question: Question, lang: str = 'cn', llm: str = 'chatgpt', llm_cfgs=None):
self.question = question
self.lang = lang
self.llm = llm
self.llm_cfgs = dict(llm_cfgs or {})
@property
def question_text(self):
return self.question.texts[self.lang]
def check(self, qs_text: str) -> Tuple[str, bool, str]:
answer_text = get_llm_fn(self.llm)(qs_text, **self.llm_cfgs)
correct, explanation = self.check_answer(qs_text, answer_text)
return answer_text, correct, explanation
def check_answer(self, user_text: str, answer_text: str) -> Tuple[bool, str]:
correct, explanation = self.question.checker(self.question_text, user_text, answer_text, self.lang)
if explanation is None:
if correct:
explanation = 'LLM的回答满足要求' if self.lang == 'cn' else 'Correct Answer From LLM'
else:
explanation = 'LLM的回答不满足要求' if self.lang == 'cn' else 'Wrong Answer From LLM'
return correct, explanation