use union for acceptable type

Author: manvkaur

azure/functions/decorators/function_app.py

@@ -3218,7 +3218,7 @@ def text_completion_input(self,
                               arg_name: str,
                               prompt: str,
                               chat_model: Optional
-                              [OpenAIModels]
+                              [Union[str, OpenAIModels]]
                               = OpenAIModels.DefaultChatModel,
                               temperature: Optional[str] = "0.5",
                               top_p: Optional[str] = None,
@@ -3376,7 +3376,7 @@ def assistant_post_input(self, arg_name: str,
                              id: str,
                              user_message: str,
                              chat_model: Optional
-                             [OpenAIModels]
+                             [Union[str, OpenAIModels]]
                              = OpenAIModels.DefaultChatModel,
                              chat_storage_connection_setting: Optional[str] = "AzureWebJobsStorage",       # noqa: E501
                              collection_name: Optional[str] = "ChatState",      # noqa: E501
@@ -3455,7 +3455,7 @@ def embeddings_input(self,
                          input: str,
                          input_type: InputType,
                          embeddings_model: Optional
-                         [OpenAIModels]
+                         [Union[str, OpenAIModels]]
                          = OpenAIModels.DefaultEmbeddingsModel,
                          max_chunk_length: Optional[int] = 8 * 1024,
                          max_overlap: Optional[int] = 128,
@@ -3513,10 +3513,10 @@ def semantic_search_input(self,
                               collection: str,
                               query: Optional[str] = None,
                               embeddings_model: Optional
-                              [OpenAIModels]
+                              [Union[str, OpenAIModels]]
                               = OpenAIModels.DefaultEmbeddingsModel,
                               chat_model: Optional
-                              [OpenAIModels]
+                              [Union[str, OpenAIModels]]
                               = OpenAIModels.DefaultChatModel,
                               system_prompt: Optional[str] = semantic_search_system_prompt,  # NoQA
                               max_knowledge_count: Optional[int] = 1,
@@ -3609,7 +3609,7 @@ def embeddings_store_output(self,
                                 store_connection_name: str,
                                 collection: str,
                                 embeddings_model: Optional
-                                [OpenAIModels]
+                                [Union[str, OpenAIModels]]
                                 = OpenAIModels.DefaultEmbeddingsModel,
                                 max_chunk_length: Optional[int] = 8 * 1024,
                                 max_overlap: Optional[int] = 128,

azure/functions/decorators/openai.py

@@ -52,7 +52,7 @@ def __init__(self,
                  name: str,
                  prompt: str,
                  chat_model: Optional
-                 [OpenAIModels]
+                 [Union[str, OpenAIModels]]
                  = OpenAIModels.DefaultChatModel,
                  temperature: Optional[str] = "0.5",
                  top_p: Optional[str] = None,