logo
OpenAI
repo-copy-icon

copied

Browse Source

Rename module

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 1 year ago
parent
commit
6e7f448cf8
  1. 4
      __init__.py
  2. 2
      openai_chat.py

4
__init__.py

@ -1,5 +1,5 @@
from .openai_chat import OpenAI
from .openai_chat import OpenaiChat
def OpenAI(*args, **kwargs): def OpenAI(*args, **kwargs):
return OpenAI(*args, **kwargs)
return OpenaiChat(*args, **kwargs)

2
openai_chat.py

@ -19,7 +19,7 @@ import openai
from towhee.operator.base import PyOperator from towhee.operator.base import PyOperator
class OpenAI(PyOperator):
class OpenaiChat(PyOperator):
'''Wrapper of OpenAI Chat API''' '''Wrapper of OpenAI Chat API'''
def __init__(self, def __init__(self,
model_name: str = 'gpt-3.5-turbo', model_name: str = 'gpt-3.5-turbo',

Loading…
Cancel
Save