|
@ -3,10 +3,7 @@ from typing import List, Tuple, Dict, Optional |
|
|
from towhee.operator import PyOperator |
|
|
from towhee.operator import PyOperator |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class QAPrompt(PyOperator): |
|
|
|
|
|
def __init__(self): |
|
|
|
|
|
super().__init__() |
|
|
|
|
|
self._template = """Use the following pieces of context to answer the question at the end. |
|
|
|
|
|
|
|
|
gpt_prompt = """Use the following pieces of context to answer the question at the end. |
|
|
If you don't know the answer, just say that you don't know, don't try to make up an answer. |
|
|
If you don't know the answer, just say that you don't know, don't try to make up an answer. |
|
|
|
|
|
|
|
|
{context} |
|
|
{context} |
|
@ -16,6 +13,24 @@ Question: {question} |
|
|
Helpful Answer: |
|
|
Helpful Answer: |
|
|
""" |
|
|
""" |
|
|
|
|
|
|
|
|
|
|
|
dolly_prompt = """{question} |
|
|
|
|
|
|
|
|
|
|
|
Input: |
|
|
|
|
|
{context} |
|
|
|
|
|
""" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class QAPrompt(PyOperator): |
|
|
|
|
|
def __init__(self, temp: str = None, llm_name: str = None): |
|
|
|
|
|
super().__init__() |
|
|
|
|
|
if temp: |
|
|
|
|
|
self._template = temp |
|
|
|
|
|
else: |
|
|
|
|
|
if llm_name.lower() == 'dolly': |
|
|
|
|
|
self._template = dolly_prompt |
|
|
|
|
|
else: |
|
|
|
|
|
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, docs: List[str], history=Optional[List[Tuple]]) -> List[Dict[str, str]]: |
|
|
""" |
|
|
""" |
|
|
history: |
|
|
history: |
|
|