42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
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()
|
|
|
|
def get_db_config():
|
|
system_path = pathlib.Path(__file__).parent
|
|
cognee.config.system_root_directory(os.path.join(system_path, ".cognee_system"))
|
|
cognee.config.data_root_directory(os.path.join(system_path, ".data_storage"))
|
|
cognee.config.set_relational_db_config({"db_provider": "sqlite"})
|
|
cognee.config.set_vector_db_config({
|
|
"vector_db_provider": "qdrant",
|
|
"vector_db_url": os.getenv("QDRANT_URL"),
|
|
"vector_db_key": os.getenv("QDRANT_API_KEY")
|
|
})
|
|
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")
|
|
|
|
agent = Agent(
|
|
model=llm,
|
|
tools=[cognee_tools],
|
|
description="You are my executive assistant who plans my itinerary based on my preference",
|
|
instructions=INSTRUCTIONS
|
|
)
|
|
|
|
agent.print_response(MY_PREFERENCE, stream=True)
|
|
print("\n")
|
|
agent.print_response("I am visiting Rome, give me restaurants list to stop by", stream=True)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|