Update azure_openai.py

azure_openai.py

@@ -1,7 +1,8 @@
 import httpx
 import logging
 import os
-from pipecat.frames import TextFrame  # Updated import
+from pipecat.frames import TextFrame, LLMResponseFrame  # Updated import
+from pipecat.services.openai.llm import OpenAILLMService  # Updated import
 from pipecat.processors.frame_processor import FrameProcessor, FrameDirection
 
 logger = logging.getLogger(__name__)
@@ -11,7 +12,7 @@ class AzureOpenAILLMService(FrameProcessor):
         super().__init__()
         self.api_key = os.environ.get("azure_openai")
         if not self.api_key:
-            logger.error("Missing Hugging Face Secret: azure_openai")
+            logger.error("Missing Azure OpenAI API key: azure_openai")
             raise ValueError("Azure OpenAI API key not found in environment variable 'azure_openai'")
         self.preprompt = preprompt
         self.endpoint = endpoint
@@ -45,7 +46,7 @@ class AzureOpenAILLMService(FrameProcessor):
                 if "choices" in result and len(result["choices"]) > 0:
                     content = result["choices"][0]["message"]["content"]
                     continue_flag = len(content) >= 4000
-                    await self.push_frame(TextFrame(content=content))  # Corrected this line
+                    await self.push_frame(LLMResponseFrame(content=content, continue_flag=continue_flag))
                 else:
                     logger.error("No valid content in API response")
                     await self.push_frame(TextFrame("Error: No valid response from LLM"))