Rename module

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 6e7f448cf8
2 changed files with 3 additions and 3 deletions
--- a/init.py
+++ b/init.py
@ -1,5 +1,5 @@
-from .openai_chat import OpenAI
+from .openai_chat import OpenaiChat


 def OpenAI(*args, **kwargs):
-    return OpenAI(*args, **kwargs)
+    return OpenaiChat(*args, **kwargs)
--- a/openai_chat.py
+++ b/openai_chat.py
@ -19,7 +19,7 @@ import openai
 from towhee.operator.base import PyOperator


-class OpenAI(PyOperator):
+class OpenaiChat(PyOperator):
    '''Wrapper of OpenAI Chat API'''
    def __init__(self,
                 model_name: str = 'gpt-3.5-turbo',