From 1913a3c90955bffd06f009010fc27e7f0b976650 Mon Sep 17 00:00:00 2001
From: yinpeng <2291314224@qq.com>
Date: Mon, 16 Dec 2024 23:28:23 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E9=9D=9E=E6=B5=81=E5=BC=8F?=
 =?UTF-8?q?=E6=A8=A1=E5=BC=8F=E4=B8=8B=E7=A7=BB=E9=99=A4=E4=BB=A3=E7=A0=81?=
 =?UTF-8?q?=E6=89=A7=E8=A1=8C=E5=B7=A5=E5=85=B7=E7=9A=84=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/services/chat_service.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/app/services/chat_service.py b/app/services/chat_service.py
index fb44d0d..058f4bc 100644
--- a/app/services/chat_service.py
+++ b/app/services/chat_service.py
@@ -172,7 +172,8 @@ class ChatService:
 
         if not stream:
             # 非流式模式下，移除代码执行工具
-            tools.remove({"code_execution": {}})
+            if {"code_execution": {}} in tools:
+                tools.remove({"code_execution": {}})
         payload = {
             "contents": gemini_messages,
             "generationConfig": {"temperature": temperature},