diff --git a/gurt/tools.py b/gurt/tools.py index ffda06b..8e62d4d 100644 --- a/gurt/tools.py +++ b/gurt/tools.py @@ -4552,8 +4552,8 @@ async def send_tenor_gif( # Generate AI selection try: - model = genai.GenerativeModel(DEFAULT_MODEL) - response = await model.generate_content_async( + response = await cog.genai_client.aio.models.generate_content( + model=DEFAULT_MODEL, contents=[types.Content(role="user", parts=ai_content)], config=types.GenerateContentConfig( temperature=0.5, # Low temperature for consistent selection