From cdd132519356bd605d2f2dff0e7fc91de8b82238 Mon Sep 17 00:00:00 2001 From: rahul-aot Date: Tue, 19 May 2026 22:03:02 -0700 Subject: [PATCH] update GEMINI_MODEL to version 2.5 and enhance schema conversion for items --- sample.env | 2 +- src/agents/providers/gemini_provider.py | 8 ++++++++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/sample.env b/sample.env index 6a82c10..00383f6 100644 --- a/sample.env +++ b/sample.env @@ -83,7 +83,7 @@ OPENAI_MODEL=gpt-4o-mini # Google Gemini (optional) GEMINI_API_KEY=your-gemini-api-key -GEMINI_MODEL=gemini-2.0-flash +GEMINI_MODEL=gemini-2.5-flash # Anthropic / Claude (optional) ANTHROPIC_API_KEY=your-anthropic-api-key diff --git a/src/agents/providers/gemini_provider.py b/src/agents/providers/gemini_provider.py index 379dd8d..eed4399 100644 --- a/src/agents/providers/gemini_provider.py +++ b/src/agents/providers/gemini_provider.py @@ -33,6 +33,14 @@ def _mcp_schema_to_gemini(schema: Dict[str, Any]) -> Dict[str, Any]: cleaned["properties"] = { k: _mcp_schema_to_gemini(v) for k, v in cleaned["properties"].items() } + if "items" in cleaned: + if isinstance(cleaned["items"], dict): + cleaned["items"] = _mcp_schema_to_gemini(cleaned["items"]) + elif isinstance(cleaned["items"], list): + cleaned["items"] = [ + _mcp_schema_to_gemini(item) if isinstance(item, dict) else item + for item in cleaned["items"] + ] return cleaned