gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

call get_db_config

Browse source
This commit is contained in:
Tarun Jain 2025-12-18 14:21:36 +01:00 • committed by GitHub
parent e23d9c3610
commit f9929158c4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
examples/agents-integrations/agno-agents/custom_stack.py
View file

@ -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()
main()