diff --git a/src/pipecat_flows/adapters.py b/src/pipecat_flows/adapters.py index 0aee1ed..1d5f550 100644 --- a/src/pipecat_flows/adapters.py +++ b/src/pipecat_flows/adapters.py @@ -298,8 +298,9 @@ def create_adapter(llm) -> LLMAdapter: # Try Google try: from pipecat.services.google import GoogleLLMService + from pipecat.services.gemini_multimodal_live.gemini import GeminiMultimodalLiveLLMService - if isinstance(llm, GoogleLLMService): + if isinstance(llm, GoogleLLMService) or isinstance(llm, GeminiMultimodalLiveLLMService): logger.debug("Creating Google adapter") return GeminiAdapter() except ImportError as e: diff --git a/src/pipecat_flows/manager.py b/src/pipecat_flows/manager.py index 7caada4..44ba414 100644 --- a/src/pipecat_flows/manager.py +++ b/src/pipecat_flows/manager.py @@ -45,8 +45,9 @@ from pipecat.services.anthropic import AnthropicLLMService from pipecat.services.google import GoogleLLMService from pipecat.services.openai import OpenAILLMService + from pipecat.services.gemini_multimodal_live.gemini import GeminiMultimodalLiveLLMService - LLMService = Union[OpenAILLMService, AnthropicLLMService, GoogleLLMService] + LLMService = Union[OpenAILLMService, AnthropicLLMService, GoogleLLMService, GeminiMultimodalLiveLLMService] else: LLMService = Any