agent execution: default and custom stack
This commit is contained in:
parent
4b56fa94d7
commit
e3d2b7cc6b
2 changed files with 65 additions and 0 deletions
25
examples/agents-integrations/agno-agents/app.py
Normal file
25
examples/agents-integrations/agno-agents/app.py
Normal file
|
|
@ -0,0 +1,25 @@
|
|||
from agno.agent import Agent
|
||||
from agno.models.openai import OpenAIChat
|
||||
|
||||
from tools import CogneeTools
|
||||
from constants import INSTRUCTIONS, MY_PREFERENCE
|
||||
from dotenv import load_dotenv
|
||||
load_dotenv()
|
||||
|
||||
def main():
|
||||
cognee_tools = CogneeTools()
|
||||
llm = OpenAIChat(id="gpt-5-mini")
|
||||
|
||||
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()
|
||||
40
examples/agents-integrations/agno-agents/custom_stack.py
Normal file
40
examples/agents-integrations/agno-agents/custom_stack.py
Normal file
|
|
@ -0,0 +1,40 @@
|
|||
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"))
|
||||
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():
|
||||
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()
|
||||
Loading…
Add table
Reference in a new issue