Update app.py

Author: mrm9084

examples/Python/ChatApp/app.py

@@ -10,7 +10,7 @@
 """
 
 import os
-from typing import List, Dict, Any
+from typing import Dict, Any
 from azure.core.credentials import TokenCredential
 from azure.identity import DefaultAzureCredential, get_bearer_token_provider
 from azure.appconfiguration.provider import load, SettingSelector, WatchKey
@@ -19,6 +19,7 @@
 
 
 def main():
+    global appconfig, azure_openai_config, chat_completion_config, azure_client, chat_messages, chat_conversation, credential
     # Create a credential using DefaultAzureCredential
     credential = DefaultAzureCredential()
 
@@ -29,84 +30,79 @@ def main():
             "The environment variable 'AZURE_APPCONFIGURATION_ENDPOINT' is not set or is empty."
         )
 
-    try:
-        azure_openai_config = None
-        azure_client = None
-
-        def on_refresh_success():
-            azure_openai_config, azure_client = _create_ai_client(appconfig)
-
-        global appconfig
-        # Create the configuration provider with refresh settings
-        appconfig = load(
-            endpoint=app_config_endpoint,
-            selects=[SettingSelector(key_filter="ChatApp:*")],
-            credential=credential,
-            keyvault_credential=credential,  # Use the same credential for Key Vault references
-            trim_prefixes=["ChatApp:"],
-            refresh_on=[WatchKey(key="ChatApp:ChatCompletion:Messages")],
-            on_refresh_success=on_refresh_success,
-        )
-
-        azure_openai_config, azure_client = _create_ai_client(appconfig, credential)
+    # Initialize chat conversation
+    chat_messages = []
+    chat_conversation = []
 
-        # Initialize chat conversation
-        chat_conversation = []
-        print("Chat started! What's on your mind?")
+    def configure_app():
+        global appconfig, azure_openai_config, chat_completion_config, azure_client, chat_messages, credential
+        azure_openai_config = _extract_openai_config(appconfig)
+        azure_client = _create_ai_client(azure_openai_config, credential)
 
-        while True:
-            # Refresh the configuration from Azure App Configuration
-            appconfig.refresh()
+        # Configure chat completion with AI configuration
+        chat_completion_config = ChatCompletionConfiguration(
+            **appconfig["ChatCompletion"]
+        )
+        chat_messages = chat_completion_config.messages
+
+    # Create the configuration provider with refresh settings
+    appconfig = load(
+        endpoint=app_config_endpoint,
+        selects=[SettingSelector(key_filter="ChatApp:*")],
+        credential=credential,
+        keyvault_credential=credential,  # Use the same credential for Key Vault references
+        trim_prefixes=["ChatApp:"],
+        refresh_on=[WatchKey(key="ChatApp:ChatCompletion")],
+        on_refresh_success=configure_app,
+    )
+    configure_app()
 
-            # Configure chat completion with AI configuration
-            chat_completion_config = ChatCompletionConfiguration(**appconfig["ChatCompletion"])
+    print("Chat started! What's on your mind?")
 
-            # Get user input
-            user_input = input("You: ")
+    while True:
+        # Refresh the configuration from Azure App Configuration
+        appconfig.refresh()
 
-            # Exit if user input is empty
-            if not user_input.strip():
-                print("Exiting chat. Goodbye!")
-                break
+        # Get user input
+        user_input = input("You: ")
 
-            # Add user message to chat conversation
-            chat_conversation.append({"role": "user", "content": user_input})
+        # Exit if user input is empty
+        if not user_input.strip():
+            print("Exiting chat. Goodbye!")
+            break
 
-            # Get latest system message from AI configuration
-            chat_messages = _get_chat_messages(chat_completion_config)
-            chat_messages.extend(chat_conversation)
+        # Add user message to chat conversation
+        chat_conversation.append({"role": "user", "content": user_input})
 
-            # Get AI response and add it to chat conversation
-            response = azure_client.chat.completions.create(
-                model=azure_openai_config.deployment_name,
-                messages=chat_messages,
-                max_tokens=chat_completion_config.max_tokens,
-                temperature=chat_completion_config.temperature,
-                top_p=chat_completion_config.top_p,
-            )
+        # Get latest system message from AI configuration
+        chat_messages.extend(chat_conversation)
 
-            ai_response = response.choices[0].message.content
-            chat_conversation.append({"role": "assistant", "content": ai_response})
-            print(f"AI: {ai_response}")
+        # Get AI response and add it to chat conversation
+        response = azure_client.chat.completions.create(
+            model=azure_openai_config.deployment_name,
+            messages=chat_messages,
+            max_tokens=chat_completion_config.max_tokens,
+            temperature=chat_completion_config.temperature,
+            top_p=chat_completion_config.top_p,
+        )
 
-    except Exception as e:
-        print(f"Error: {e}")
+        ai_response = response.choices[0].message.content
+        chat_conversation.append({"role": "assistant", "content": ai_response})
+        print(f"AI: {ai_response}")
 
 
 def _create_ai_client(
     azure_openai_config: AzureOpenAIConfiguration, credential: TokenCredential = None
 ) -> AzureOpenAI:
-    # Extract Azure OpenAI configuration
-    azure_openai_config = _extract_openai_config(appconfig)
     # Create an Azure OpenAI client
     if azure_openai_config.api_key:
-        return azure_openai_config, AzureOpenAI(
+        return AzureOpenAI(
             azure_endpoint=azure_openai_config.endpoint,
             api_key=azure_openai_config.api_key,
             api_version=azure_openai_config.api_version,
         )
     else:
-        return azure_openai_config, AzureOpenAI(
+        return AzureOpenAI(
             azure_endpoint=azure_openai_config.endpoint,
             azure_ad_token_provider=get_bearer_token_provider(
                 credential or DefaultAzureCredential(),
@@ -131,25 +127,6 @@ def _extract_openai_config(config_data: Dict[str, Any]) -> AzureOpenAIConfigurat
         api_version=config_data.get(f"{prefix}ApiVersion", "2023-05-15"),
     )
 
-def _get_chat_messages(
-    chat_completion_config: ChatCompletionConfiguration,
-) -> List[Dict[str, str]]:
-    """
-    Convert configuration messages to chat message dictionaries.
-
-    :param chat_completion_config: The chat completion configuration
-    :return: A list of chat message dictionaries
-    """
-    chat_messages = []
-
-    for message in chat_completion_config.messages:
-        if message["role"] in ["system", "user", "assistant"]:
-            chat_messages.append({"role": message["role"], "content": message["content"]})
-        else:
-            raise ValueError(f"Unknown role: {message.role}")
-
-    return chat_messages
-
 
 if __name__ == "__main__":
     main()