Update cellemetry/agents/analyst.py

cellemetry/agents/analyst.py

@@ -3,6 +3,7 @@ Analyst Agent - Expert microscopy image analyst.
 Segments biological components and computes statistics.
 """
 from google.adk.agents import LlmAgent
+from google.adk.models.google_llm import Gemini
 from google.adk.tools.agent_tool import AgentTool
 
 from ..tools import ANALYST_TOOLS
@@ -59,10 +60,13 @@ Return your findings as structured data including:
 """
 
 analyst_agent = LlmAgent(
-    name="analyst",
-    model="gemini-3-pro-preview",
-    description="Expert microscopy analyst that segments and quantifies biological structures.",
-    instruction=ANALYST_INSTRUCTION,
-    tools=ANALYST_TOOLS,
-    output_key="analyst_result",
+    name = "analyst",
+    description = "Expert microscopy analyst for localizing and quantifying biological structures",
+    model = Gemini(
+        model="gemini-3-flash-preview", 
+        media_resolution="high"
+    ),
+    instruction = ANALYST_INSTRUCTION,
+    tools = ANALYST_TOOLS,
+    output_key = "analyst_result",
 )