Browse Source
Add files
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
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 |
|
|
|
|
