[flytekitplugins] Rename ChatGPT module to OpenAI module (flyteorg#2263)

* refactor chagpt module to openai module Signed-off-by: Future-Outlier <[email protected]> * nit Signed-off-by: Future-Outlier <[email protected]> --------- Signed-off-by: Future-Outlier <[email protected]>
jasonlai1218 · Mar 26, 2024 · fa2aa0b · fa2aa0b
1 parent 133e8d5
commit fa2aa0b
Show file tree

Hide file tree

Showing 8 changed files with 10 additions and 10 deletions.
diff --git a/plugins/flytekit-openai/README.md b/plugins/flytekit-openai/README.md
@@ -4,7 +4,7 @@ ChatGPT plugin allows you to run ChatGPT tasks in the Flyte workflow without cha
 ## Example
 ```python
 from flytekit import task, workflow
-from flytekitplugins.chatgpt import ChatGPTTask, ChatGPTConfig
+from flytekitplugins.openai import ChatGPTTask, ChatGPTConfig
 
 chatgpt_small_job = ChatGPTTask(
     name="chatgpt gpt-3.5-turbo",
@@ -40,5 +40,5 @@ if __name__ == "__main__":
 To install the plugin, run the following command:
 
 ```bash
-pip install flytekitplugins-chatgpt
+pip install flytekitplugins-openai
 ```
diff --git a/...penai/flytekitplugins/chatgpt/__init__.py → ...openai/flytekitplugins/openai/__init__.py b/...penai/flytekitplugins/chatgpt/__init__.py → ...openai/flytekitplugins/openai/__init__.py
@@ -1,5 +1,5 @@
 """
-.. currentmodule:: flytekitplugins.chatgpt
+.. currentmodule:: flytekitplugins.openai
 This package contains things that are useful when extending Flytekit.
 .. autosummary::
    :template: custom.rst
@@ -8,5 +8,5 @@
    ChatGPTTask
 """
 
-from .agent import ChatGPTAgent
-from .task import ChatGPTTask
+from .chatgpt.agent import ChatGPTAgent
+from .chatgpt.task import ChatGPTTask
diff --git a/plugins/flytekit-openai/flytekitplugins/openai/chatgpt/__init__.py b/plugins/flytekit-openai/flytekitplugins/openai/chatgpt/__init__.py
diff --git a/...t-openai/flytekitplugins/chatgpt/agent.py → ...i/flytekitplugins/openai/chatgpt/agent.py b/...t-openai/flytekitplugins/chatgpt/agent.py → ...i/flytekitplugins/openai/chatgpt/agent.py
diff --git a/...it-openai/flytekitplugins/chatgpt/task.py → ...ai/flytekitplugins/openai/chatgpt/task.py b/...it-openai/flytekitplugins/chatgpt/task.py → ...ai/flytekitplugins/openai/chatgpt/task.py
diff --git a/plugins/flytekit-openai/setup.py b/plugins/flytekit-openai/setup.py
@@ -1,10 +1,10 @@
 from setuptools import setup
 
-PLUGIN_NAME = "chatgpt"
+PLUGIN_NAME = "openai"
 
 microlib_name = f"flytekitplugins-{PLUGIN_NAME}"
 
-plugin_requires = ["flytekit>1.10.7", "openai>=1.12.0", "flyteidl>=1.11.0b0"]
+plugin_requires = ["flytekit>1.10.7", "openai>=1.12.0", "flyteidl>=1.11.0"]
 
 __version__ = "0.0.0+develop"
 
@@ -13,9 +13,9 @@
     version=__version__,
     author="flyteorg",
     author_email="[email protected]",
-    description="This package holds the ChatGPT plugins for flytekit",
+    description="This package holds the openai plugins for flytekit",
     namespace_packages=["flytekitplugins"],
-    packages=[f"flytekitplugins.{PLUGIN_NAME}"],
+    packages=[f"flytekitplugins.{PLUGIN_NAME}", f"flytekitplugins.{PLUGIN_NAME}.chatgpt"],
     install_requires=plugin_requires,
     license="apache2",
     python_requires=">=3.8",

diff --git a/plugins/flytekit-openai/tests/__init__.py b/plugins/flytekit-openai/tests/__init__.py
diff --git a/plugins/flytekit-openai/tests/test_chatgpt.py b/plugins/flytekit-openai/tests/test_chatgpt.py
@@ -1,6 +1,6 @@
 from collections import OrderedDict
 
-from flytekitplugins.chatgpt import ChatGPTTask
+from flytekitplugins.openai import ChatGPTTask
 
 from flytekit.configuration import Image, ImageConfig, SerializationSettings
 from flytekit.extend import get_serializable