google: updgrade OpenAILLMContext to GoogleLLMContext

2025-02-24 15:35:18 -08:00
parent 30cdd1b71a
commit 40c2452d6e
2 changed files with 5 additions and 0 deletions
--- a/src/pipecat/services/google/google.py
+++ b/src/pipecat/services/google/google.py
@@ -1176,6 +1176,8 @@ class GoogleLLMService(LLMService):
    def create_context_aggregator(
        context: OpenAILLMContext, *, assistant_expect_stripped_words: bool = True
    ) -> GoogleContextAggregatorPair:
+        if isinstance(context, OpenAILLMContext):
+            context = GoogleLLMContext.upgrade_to_google(context)
        user = GoogleUserContextAggregator(context)
        assistant = GoogleAssistantContextAggregator(
            context, expect_stripped_words=assistant_expect_stripped_words