Make openai_organization Optional in OpenAI plugins (#2585)

* use openai_organization: Optional[str] = in OpenAI plugins Signed-off-by: Future-Outlier <[email protected]> * change default value to None Signed-off-by: Future-Outlier <[email protected]> --------- Signed-off-by: Future-Outlier <[email protected]>
flyteorg · Jul 17, 2024 · 262f6d6 · 262f6d6
1 parent 785b41c
commit 262f6d6
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/plugins/flytekit-openai/flytekitplugins/openai/batch/task.py b/plugins/flytekit-openai/flytekitplugins/openai/batch/task.py
@@ -32,8 +32,8 @@ class BatchEndpointTask(AsyncAgentExecutorMixin, PythonTask):
     def __init__(
         self,
         name: str,
-        openai_organization: str,
         config: Dict[str, Any],
+        openai_organization: Optional[str] = None,
         **kwargs,
     ):
         super().__init__(
@@ -70,8 +70,8 @@ class OpenAIFileDefaultImages(DefaultImages):
 
 @dataclass
 class OpenAIFileConfig:
-    openai_organization: str
     secret: Secret
+    openai_organization: Optional[str] = None
 
     def _secret_to_dict(self) -> Dict[str, Optional[str]]:
         return {

diff --git a/plugins/flytekit-openai/flytekitplugins/openai/batch/workflow.py b/plugins/flytekit-openai/flytekitplugins/openai/batch/workflow.py
@@ -16,8 +16,8 @@
 
 def create_batch(
     name: str,
-    openai_organization: str,
     secret: Secret,
+    openai_organization: Optional[str] = None,
     config: Optional[Dict[str, Any]] = None,
     is_json_iterator: bool = True,
     file_upload_mem: str = "700Mi",

diff --git a/plugins/flytekit-openai/flytekitplugins/openai/chatgpt/task.py b/plugins/flytekit-openai/flytekitplugins/openai/chatgpt/task.py
@@ -1,4 +1,4 @@
-from typing import Any, Dict
+from typing import Any, Dict, Optional
 
 from flytekit.configuration import SerializationSettings
 from flytekit.core.base_task import PythonTask
@@ -13,7 +13,7 @@ class ChatGPTTask(SyncAgentExecutorMixin, PythonTask):
 
     _TASK_TYPE = "chatgpt"
 
-    def __init__(self, name: str, openai_organization: str, chatgpt_config: Dict[str, Any], **kwargs):
+    def __init__(self, name: str, chatgpt_config: Dict[str, Any], openai_organization: Optional[str] = None, **kwargs):
         """
         Args:
             name: Name of this task, should be unique in the project