|
@ -38,12 +38,11 @@ class QAPrompt(PyOperator): |
|
|
logger.warning('Unkown llm_name, use default prompt') |
|
|
logger.warning('Unkown llm_name, use default prompt') |
|
|
self._template = gpt_prompt |
|
|
self._template = gpt_prompt |
|
|
|
|
|
|
|
|
def __call__(self, question: str, docs: List[str], history=Optional[List[Tuple]]) -> List[Dict[str, str]]: |
|
|
|
|
|
|
|
|
def __call__(self, question: str, context: str, history=Optional[List[Tuple]]) -> List[Dict[str, str]]: |
|
|
""" |
|
|
""" |
|
|
history: |
|
|
history: |
|
|
List[Tuple]: [(question1, answer1), (question2, answer2)] |
|
|
List[Tuple]: [(question1, answer1), (question2, answer2)] |
|
|
""" |
|
|
""" |
|
|
context = '\n'.join(docs) |
|
|
|
|
|
prompt_str = self._template.format(context=context, question=question) |
|
|
prompt_str = self._template.format(context=context, question=question) |
|
|
ret = [{'question': prompt_str}] |
|
|
ret = [{'question': prompt_str}] |
|
|
if not isinstance(history, list): |
|
|
if not isinstance(history, list): |
|
|