From 18a5fe61095f0aa2e2a6f80ba74d9fc26c8e3a61 Mon Sep 17 00:00:00 2001 From: Toddy <167494546+toddyoe@users.noreply.github.com> Date: Thu, 27 Feb 2025 07:35:12 +0000 Subject: [PATCH] fix: adapt gemini format --- app/services/gemini_chat_service.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/app/services/gemini_chat_service.py b/app/services/gemini_chat_service.py index c8a949a..48d4991 100644 --- a/app/services/gemini_chat_service.py +++ b/app/services/gemini_chat_service.py @@ -31,6 +31,12 @@ def _build_tools(model: str, payload: Dict[str, Any]) -> List[Dict[str, Any]]: tools.append({"code_execution": {}}) if model.endswith("-search"): tools.append({"googleSearch": {}}) + + if payload and isinstance(payload, dict) and "tools" in payload: + items = payload.get("tools", []) + if items and isinstance(items, list): + tools.extend(items) + return tools