refactor: extract streaming mode detection to _is_sse_streaming helper

Removes duplicated logic in _retry_same_model and _try_fallbacks (DRY).

Author: chillum-codeX

src/google/adk_community/plugins/llm_resilience_plugin.py

@@ -225,25 +225,33 @@ def _get_invocation_context(
       )
     return ic
 
-  async def _retry_same_model(
-      self,
-      *,
-      callback_context: CallbackContext | InvocationContext,
-      llm_request: LlmRequest,
-  ) -> Optional[LlmResponse]:
-    invocation_context = self._get_invocation_context(callback_context)
-    # Determine streaming mode
+  def _is_sse_streaming(self, invocation_context: InvocationContext) -> bool:
+    """Check if SSE streaming mode is enabled.
+
+    Args:
+      invocation_context: The invocation context to check.
+
+    Returns:
+      True if SSE streaming is enabled, False otherwise.
+    """
     streaming_mode = getattr(
         invocation_context.run_config, "streaming_mode", None
     )
-    stream = False
     try:
-      # Only SSE streaming is supported in generate_content_async
       from google.adk.agents.run_config import StreamingMode
 
-      stream = streaming_mode == StreamingMode.SSE
+      return streaming_mode == StreamingMode.SSE
     except (ImportError, AttributeError):
-      pass
+      return False
+
+  async def _retry_same_model(
+      self,
+      *,
+      callback_context: CallbackContext | InvocationContext,
+      llm_request: LlmRequest,
+  ) -> Optional[LlmResponse]:
+    invocation_context = self._get_invocation_context(callback_context)
+    stream = self._is_sse_streaming(invocation_context)
 
     agent = invocation_context.agent
     llm = agent.canonical_model
@@ -283,17 +291,7 @@ async def _try_fallbacks(
       llm_request: LlmRequest,
   ) -> Optional[LlmResponse]:
     invocation_context = self._get_invocation_context(callback_context)
-    # Determine streaming mode
-    streaming_mode = getattr(
-        invocation_context.run_config, "streaming_mode", None
-    )
-    stream = False
-    try:
-      from google.adk.agents.run_config import StreamingMode
-
-      stream = streaming_mode == StreamingMode.SSE
-    except (ImportError, AttributeError):
-      pass
+    stream = self._is_sse_streaming(invocation_context)
 
     for model_name in self.fallback_models:
       try: