diff --git a/src/agent/base.py b/src/agent/base.py index f635ba8..dce4088 100644 --- a/src/agent/base.py +++ b/src/agent/base.py @@ -2,6 +2,7 @@ import os from deepagents import create_deep_agent from langchain_openai import ChatOpenAI from langgraph.checkpoint.memory import MemorySaver +from composio import Composio def create_agent(): @@ -11,9 +12,13 @@ def create_agent(): api_key=os.environ["PROVIDER_API_KEY"], ) + composio = Composio() + session = composio.create(user_id="default_user") + tools = session.tools() + return create_deep_agent( model=model, - system_prompt="You are a helpful assistant.", + system_prompt="You are a helpful assistant. Use Composio tools to take action when needed.", checkpointer=MemorySaver(), - ) - + tools=tools + ) \ No newline at end of file