joaomorossini commited on
Commit
34aef60
·
1 Parent(s): 49f4f70

Add CreateTaskTool to NotionProjectAgent for enhanced task management

Browse files
agency_ai_demo/agents/NotionProjectAgent/NotionProjectAgent.py CHANGED
@@ -3,6 +3,7 @@ import os
3
  from agency_swarm.agents import Agent
4
  from .tools.GetTasks import GetTasksTool
5
  from .tools.GetTask import GetTaskTool
 
6
 
7
 
8
  class NotionProjectAgent(Agent):
@@ -22,7 +23,7 @@ class NotionProjectAgent(Agent):
22
  instructions="./instructions.md",
23
  files_folder="./files",
24
  schemas_folder="./schemas",
25
- tools=[GetTasksTool, GetTaskTool],
26
  tools_folder="./tools",
27
  model="gpt-4o",
28
  temperature=0.3,
 
3
  from agency_swarm.agents import Agent
4
  from .tools.GetTasks import GetTasksTool
5
  from .tools.GetTask import GetTaskTool
6
+ from .tools.CreateTask import CreateTaskTool
7
 
8
 
9
  class NotionProjectAgent(Agent):
 
23
  instructions="./instructions.md",
24
  files_folder="./files",
25
  schemas_folder="./schemas",
26
+ tools=[GetTasksTool, GetTaskTool, CreateTaskTool],
27
  tools_folder="./tools",
28
  model="gpt-4o",
29
  temperature=0.3,