Update azure_openai.py

azure_openai.py

@@ -1,7 +1,7 @@
 import httpx
 import logging
 import os
-from pipecat.frames.frames import TextFrame, LLMResponseFrame
+from pipecat.frames import TextFrame  # Updated import
 from pipecat.processors.frame_processor import FrameProcessor, FrameDirection
 
 logger = logging.getLogger(__name__)
@@ -17,7 +17,7 @@ class AzureOpenAILLMService(FrameProcessor):
         self.endpoint = endpoint
         self.client = httpx.AsyncClient()
 
-    async def process_frame(self, frame: Frame, direction: FrameDirection):
+    async def process_frame(self, frame, direction: FrameDirection):
         if isinstance(frame, TextFrame) and direction == FrameDirection.UPSTREAM:
             try:
                 messages = []
@@ -45,7 +45,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(LLMResponseFrame(content=content, continue_flag=continue_flag))
+                    await self.push_frame(TextFrame(content=content))  # Corrected this line
                 else:
                     logger.error("No valid content in API response")
                     await self.push_frame(TextFrame("Error: No valid response from LLM"))