Update eb_agent_module.py

eb_agent_module.py

@@ -9,8 +9,10 @@ import textwrap
 
 # Attempt to import Google Generative AI and related types
 try:
-    from google import generativeai as genai # Renamed for clarity to avoid conflict
-    from google.generativeai import types as genai_types
+
+    from google import genai
+
+    from google.genai import types as genai_types
     # from google.generativeai import GenerationConfig # For direct use if needed
     # from google.generativeai.types import HarmCategory, HarmBlockThreshold, SafetySetting # For direct use