Skip to content

Commit 594a23b

Browse files
committed
Fix attributeerror for gemini 2.0 flash thinking
1 parent 0159bf6 commit 594a23b

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

aimodels/gemini/config.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ async def _fetch_tool(self, db_conn) -> dict:
5555
# Check if tool is code execution
5656
if _Tool:
5757
if "gemini-2.0-flash-thinking" in self._model_name:
58-
await self._discord_ctx_send("> ⚠️ The Gemini 2.0 Flash Thinking doesn't support tools, please switch to another Gemini model to use it.")
58+
await self._discord_method_send("> ⚠️ The Gemini 2.0 Flash Thinking doesn't support tools, please switch to another Gemini model to use it.")
5959
_tool_schema = None
6060
else:
6161
if _tool_selection_name == "code_execution":

aimodels/gemini/infer.py

-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
import aiofiles
1111
import asyncio
1212
import discord
13-
import importlib
1413
import io
1514
import logging
1615
import typing

0 commit comments

Comments
 (0)