diff --git a/examples/agents-integrations/agno-agents/custom_stack.py b/examples/agents-integrations/agno-agents/custom_stack.py index 10c979f4f..7d84b7fd6 100644 --- a/examples/agents-integrations/agno-agents/custom_stack.py +++ b/examples/agents-integrations/agno-agents/custom_stack.py @@ -1,14 +1,15 @@ +import os, pathlib, cognee +from cognee_community_vector_adapter_qdrant import register + from agno.agent import Agent from agno.models.google import Gemini from tools import CogneeTools from constants import INSTRUCTIONS, MY_PREFERENCE + from dotenv import load_dotenv load_dotenv() -import os, pathlib, cognee -from cognee_community_vector_adapter_qdrant import register - def get_db_config(): system_path = pathlib.Path(__file__).parent cognee.config.system_root_directory(os.path.join(system_path, ".cognee_system")) @@ -22,6 +23,7 @@ def get_db_config(): cognee.config.set_graph_db_config({"graph_database_provider": "kuzu"}) def main(): + get_db_config() cognee_tools = CogneeTools() llm = Gemini(id="gemini-2.5-flash") @@ -37,4 +39,4 @@ def main(): agent.print_response("I am visiting Rome, give me restaurants list to stop by", stream=True) if __name__ == "__main__": - main() \ No newline at end of file + main()