feat: implement google genai provider

pyproject.toml

@@ -30,7 +30,11 @@ dependencies = [
     "a2a-types>=0.1.0",
     "opentelemetry-instrumentation-openai>=0.39.3",
     "opentelemetry-instrumentation-anthropic>=0.39.3",
+    "google-genai",
+    "opentelemetry-instrumentation-google-genai>=0.2b0",
     "tensorzero>=2025.4.7",
+    "google-genai",
+    "opentelemetry-instrumentation-google-genai>=0.2b0",
 ]
 
 [project.optional-dependencies]

src/mcp_agent/context.py

@@ -11,6 +11,7 @@
 from opentelemetry import trace
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 from opentelemetry.instrumentation.anthropic import AnthropicInstrumentor
+from opentelemetry.instrumentation.google_genai import GoogleGenAiSdkInstrumentor
 from opentelemetry.instrumentation.openai import OpenAIInstrumentor
 from opentelemetry.propagate import set_global_textmap
 from opentelemetry.sdk.resources import Resource
@@ -111,6 +112,7 @@ async def configure_otel(config: "Settings") -> None:
     trace.set_tracer_provider(tracer_provider)
     AnthropicInstrumentor().instrument()
     OpenAIInstrumentor().instrument()
+    GoogleGenAiSdkInstrumentor().instrument()
 
 
 async def configure_logger(config: "Settings") -> None:

src/mcp_agent/llm/model_factory.py

@@ -86,6 +86,9 @@ class ModelFactory:
         "claude-3-opus-20240229": Provider.ANTHROPIC,
         "claude-3-opus-latest": Provider.ANTHROPIC,
         "deepseek-chat": Provider.DEEPSEEK,
+        "gemini-2.0-flash": Provider.GOOGLE,
+        "gemini-2.5-flash-preview-04-17": Provider.GOOGLE,
+        "gemini-2.5-pro-preview-03-25": Provider.GOOGLE,
         #        "deepseek-reasoner": Provider.DEEPSEEK, reinstate on release
     }
 
@@ -101,6 +104,9 @@ class ModelFactory:
         "opus3": "claude-3-opus-latest",
         "deepseekv3": "deepseek-chat",
         "deepseek": "deepseek-chat",
+        "gemini2": "gemini-2.0-flash",
+        "gemini25": "gemini-2.5-flash-preview-04-17",
+        "gemini25pro": "gemini-2.5-pro-preview-03-25",
     }
 
     # Mapping of providers to their LLM classes