diff --git a/openai_qa.py b/openai_qa.py index 3b9826c..1aafcc1 100644 --- a/openai_qa.py +++ b/openai_qa.py @@ -18,7 +18,7 @@ from tenacity import retry, stop_after_attempt, wait_random_exponential from towhee.operator.base import PyOperator -class OpenaiEmbeding(PyOperator): +class OpenaiQA(PyOperator): def __init__(self, model_name='text-davinci-003', api_key=None): self._engine = model_name self._api_key = api_key