diff --git a/src/pipecat/processors/aggregators/llm_response.py b/src/pipecat/processors/aggregators/llm_response.py index 568314f77..b72190e48 100644 --- a/src/pipecat/processors/aggregators/llm_response.py +++ b/src/pipecat/processors/aggregators/llm_response.py @@ -36,6 +36,7 @@ from pipecat.frames.frames import ( FunctionCallsStartedFrame, InputAudioRawFrame, InterimTranscriptionFrame, + LLMContextAssistantTimestampFrame, LLMFullResponseEndFrame, LLMFullResponseStartFrame, LLMMessagesAppendFrame, @@ -44,7 +45,6 @@ from pipecat.frames.frames import ( LLMSetToolChoiceFrame, LLMSetToolsFrame, LLMTextFrame, - OpenAILLMContextAssistantTimestampFrame, SpeechControlParamsFrame, StartFrame, StartInterruptionFrame, @@ -864,7 +864,7 @@ class LLMAssistantContextAggregator(LLMContextResponseAggregator): await self.push_context_frame() # Push timestamp frame with current time - timestamp_frame = OpenAILLMContextAssistantTimestampFrame(timestamp=time_now_iso8601()) + timestamp_frame = LLMContextAssistantTimestampFrame(timestamp=time_now_iso8601()) await self.push_frame(timestamp_frame) async def _handle_llm_messages_append(self, frame: LLMMessagesAppendFrame): diff --git a/tests/test_context_aggregators.py b/tests/test_context_aggregators.py index 232fa59cc..3731fd3bf 100644 --- a/tests/test_context_aggregators.py +++ b/tests/test_context_aggregators.py @@ -17,9 +17,9 @@ from pipecat.frames.frames import ( FunctionCallResultFrame, FunctionCallResultProperties, InterimTranscriptionFrame, + LLMContextAssistantTimestampFrame, LLMFullResponseEndFrame, LLMFullResponseStartFrame, - OpenAILLMContextAssistantTimestampFrame, SpeechControlParamsFrame, StartInterruptionFrame, TextFrame, @@ -738,7 +738,7 @@ class TestAnthropicAssistantContextAggregator( ): CONTEXT_CLASS = AnthropicLLMContext AGGREGATOR_CLASS = AnthropicAssistantContextAggregator - EXPECTED_CONTEXT_FRAMES = [OpenAILLMContextFrame, OpenAILLMContextAssistantTimestampFrame] + EXPECTED_CONTEXT_FRAMES = [OpenAILLMContextFrame, LLMContextAssistantTimestampFrame] def check_message_multi_content( self, context: OpenAILLMContext, content_index: int, index: int, content: str @@ -773,7 +773,7 @@ class TestAWSBedrockAssistantContextAggregator( ): CONTEXT_CLASS = AWSBedrockLLMContext AGGREGATOR_CLASS = AWSBedrockAssistantContextAggregator - EXPECTED_CONTEXT_FRAMES = [OpenAILLMContextFrame, OpenAILLMContextAssistantTimestampFrame] + EXPECTED_CONTEXT_FRAMES = [OpenAILLMContextFrame, LLMContextAssistantTimestampFrame] def check_message_multi_content( self, context: OpenAILLMContext, content_index: int, index: int, content: str @@ -814,7 +814,7 @@ class TestGoogleAssistantContextAggregator( ): CONTEXT_CLASS = GoogleLLMContext AGGREGATOR_CLASS = GoogleAssistantContextAggregator - EXPECTED_CONTEXT_FRAMES = [OpenAILLMContextFrame, OpenAILLMContextAssistantTimestampFrame] + EXPECTED_CONTEXT_FRAMES = [OpenAILLMContextFrame, LLMContextAssistantTimestampFrame] def check_message_content(self, context: OpenAILLMContext, index: int, content: str): obj = context.messages[index].to_json_dict() @@ -848,4 +848,4 @@ class TestOpenAIAssistantContextAggregator( ): CONTEXT_CLASS = OpenAILLMContext AGGREGATOR_CLASS = OpenAIAssistantContextAggregator - EXPECTED_CONTEXT_FRAMES = [OpenAILLMContextFrame, OpenAILLMContextAssistantTimestampFrame] + EXPECTED_CONTEXT_FRAMES = [OpenAILLMContextFrame, LLMContextAssistantTimestampFrame]