|
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 |
|
|
