chore: Replace isinstance call with hasattr call

Co-authored-by: Xiang (Sean) Zhou <seanzhougoogle@google.com>
PiperOrigin-RevId: 869919864
This commit is contained in:
Xiang (Sean) Zhou
2026-02-13 15:47:18 -08:00
committed by Copybara-Service
parent 811e50a0cb
commit 1d4b0f9ff5
+1 -2
View File
@@ -41,12 +41,11 @@ class _ContentLlmRequestProcessor(BaseLlmRequestProcessor):
async def run_async( async def run_async(
self, invocation_context: InvocationContext, llm_request: LlmRequest self, invocation_context: InvocationContext, llm_request: LlmRequest
) -> AsyncGenerator[Event, None]: ) -> AsyncGenerator[Event, None]:
from ...agents.llm_agent import LlmAgent
from ...models.google_llm import Gemini from ...models.google_llm import Gemini
agent = invocation_context.agent agent = invocation_context.agent
preserve_function_call_ids = False preserve_function_call_ids = False
if isinstance(agent, LlmAgent): if hasattr(agent, 'canonical_model'):
canonical_model = agent.canonical_model canonical_model = agent.canonical_model
preserve_function_call_ids = ( preserve_function_call_ids = (
isinstance(canonical_model, Gemini) isinstance(canonical_model, Gemini)