From f1947d7d383302204ce676695165da0f76b73948 Mon Sep 17 00:00:00 2001 From: Mark Backman Date: Sat, 11 Jan 2025 16:16:53 -0500 Subject: [PATCH] Update Anthropic and Gemini to allow overriding run_llm --- src/pipecat/services/anthropic.py | 9 ++++++++- src/pipecat/services/google.py | 7 ++++++- 2 files changed, 14 insertions(+), 2 deletions(-) diff --git a/src/pipecat/services/anthropic.py b/src/pipecat/services/anthropic.py index 108e5f854..5189a47d5 100644 --- a/src/pipecat/services/anthropic.py +++ b/src/pipecat/services/anthropic.py @@ -775,7 +775,14 @@ class AnthropicAssistantContextAggregator(LLMAssistantContextAggregator): ], } ) - run_llm = True + if frame.override_run_llm: + # Explicit override + print("Explicit override") + run_llm = frame.run_llm + else: + # Default behavior + print("Default behavior") + run_llm = True elif aggregation: self._context.add_message({"role": "assistant", "content": aggregation}) diff --git a/src/pipecat/services/google.py b/src/pipecat/services/google.py index 643d19332..c25f14a6f 100644 --- a/src/pipecat/services/google.py +++ b/src/pipecat/services/google.py @@ -282,7 +282,12 @@ class GoogleAssistantContextAggregator(OpenAIAssistantContextAggregator): ], ) ) - run_llm = not bool(self._function_calls_in_progress) + if frame.override_run_llm: + # Explicit override + run_llm = frame.run_llm + else: + # Default behavior + run_llm = not bool(self._function_calls_in_progress) else: if aggregation.strip(): self._context.add_message(