diff --git a/src/pipecat/utils/context/llm_context_summarization.py b/src/pipecat/utils/context/llm_context_summarization.py index f7536cbaf..259d0bae5 100644 --- a/src/pipecat/utils/context/llm_context_summarization.py +++ b/src/pipecat/utils/context/llm_context_summarization.py @@ -382,7 +382,7 @@ class LLMContextSummarizationUtil: return total @staticmethod - def _get_function_calls_in_progress_index( + def _get_earliest_function_call_not_resolved_in_range( messages: List[dict], start_idx: int, summary_end: int ) -> int: """Find the earliest message index with incomplete function calls. @@ -491,8 +491,10 @@ class LLMContextSummarizationUtil: return LLMMessagesToSummarize(messages=[], last_summarized_index=-1) # Check for function calls in progress in the range we want to summarize - function_call_start = LLMContextSummarizationUtil._get_function_calls_in_progress_index( - messages, summary_start, summary_end + function_call_start = ( + LLMContextSummarizationUtil._get_earliest_function_call_not_resolved_in_range( + messages, summary_start, summary_end + ) ) if function_call_start >= 0 and function_call_start < summary_end: # Stop summarization before the function call diff --git a/tests/test_context_summarization.py b/tests/test_context_summarization.py index 454535bc0..f0d4ffc19 100644 --- a/tests/test_context_summarization.py +++ b/tests/test_context_summarization.py @@ -1168,7 +1168,7 @@ class TestLLMSpecificMessageHandling(unittest.TestCase): {"role": "tool", "tool_call_id": "call_123", "content": '{"time": "10:30 AM"}'}, ] - result = LLMContextSummarizationUtil._get_function_calls_in_progress_index( + result = LLMContextSummarizationUtil._get_earliest_function_call_not_resolved_in_range( messages, 0, len(messages) ) self.assertEqual(result, -1)