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.agent import Agent
from agno.models.google import Gemini from agno.models.google import Gemini
from tools import CogneeTools from tools import CogneeTools
from constants import INSTRUCTIONS, MY_PREFERENCE from constants import INSTRUCTIONS, MY_PREFERENCE
from dotenv import load_dotenv from dotenv import load_dotenv
load_dotenv() load_dotenv()
import os, pathlib, cognee
from cognee_community_vector_adapter_qdrant import register
def get_db_config(): def get_db_config():
system_path = pathlib.Path(__file__).parent system_path = pathlib.Path(__file__).parent
cognee.config.system_root_directory(os.path.join(system_path, ".cognee_system")) 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"}) cognee.config.set_graph_db_config({"graph_database_provider": "kuzu"})
def main(): def main():
get_db_config()
cognee_tools = CogneeTools() cognee_tools = CogneeTools()
llm = Gemini(id="gemini-2.5-flash") 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) agent.print_response("I am visiting Rome, give me restaurants list to stop by", stream=True)
if __name__ == "__main__": if __name__ == "__main__":
main() main()