diff --git a/.vscode/settings.json b/.vscode/settings.json
index 5d3ef653..b086ceaf 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -1,7 +1,10 @@
 {
     "cSpell.words": [
+        "agno",
+        "genai",
         "langsrv",
         "multilspy",
-        "pixi"
+        "pixi",
+        "sensai"
     ]
 }
\ No newline at end of file
diff --git a/scripts/agno_agent.py b/scripts/agno_agent.py
index 98271347..25b61ce1 100644
--- a/scripts/agno_agent.py
+++ b/scripts/agno_agent.py
@@ -1,11 +1,12 @@
 import os
 from logging import Logger
 
-from agno.agent import Agent
-from agno.memory import AgentMemory
-from agno.models.anthropic import Claude
-from agno.models.google import Gemini
-from agno.playground import Playground, serve_playground_app
+from agno.agent.agent import Agent
+from agno.memory.agent import AgentMemory
+from agno.models.anthropic.claude import Claude
+from agno.models.google.gemini import Gemini
+from agno.playground.playground import Playground
+from agno.playground.serve import serve_playground_app
 from dotenv import load_dotenv
 from sensai.util import logging
 from sensai.util.helper import mark_used
diff --git a/src/serena/agno.py b/src/serena/agno.py
index 982cc29d..54ccd166 100644
--- a/src/serena/agno.py
+++ b/src/serena/agno.py
@@ -1,7 +1,8 @@
 import logging
 from typing import Any
 
-from agno.tools import Function, Toolkit
+from agno.tools.function import Function
+from agno.tools.toolkit import Toolkit
 
 from serena.agent import SerenaAgent, Tool