Update Devid and TechnicalProjectManager agents to use new tool imports and set model to gpt-4o; enhance instructions and demo integration

app/agents/Devid/Devid.py

@@ -18,6 +18,7 @@ class Devid(Agent):
             tools=[FileSearch],
             tools_folder="./tools",
             validation_attempts=1,
+            model="gpt-4o",
             temperature=0,
             max_prompt_tokens=25000,
         )

app/agents/TechnicalProjectManager/TechnicalProjectManager.py

@@ -1,10 +1,10 @@
 from agency_swarm.agents import Agent
 from .tools.SendWhatsAppText import SendWhatsAppText
-from .tools.CreateTask import CreateTask
-from .tools.UpdateTask import UpdateTask
-from .tools.DeleteTask import DeleteTask
-from .tools.GetTask import GetTask
-from .tools.GetTasks import GetTasks
+from .tools.CreateTask import CreateTaskTool
+from .tools.UpdateTask import UpdateTaskTool
+from .tools.DeleteTask import DeleteTaskTool
+from .tools.GetTask import GetTaskTool
+from .tools.GetTasks import GetTasksTool
 
 
 class TechnicalProjectManager(Agent):
@@ -17,11 +17,11 @@ class TechnicalProjectManager(Agent):
             schemas_folder="./schemas",
             tools=[
                 SendWhatsAppText,
-                CreateTask,
-                UpdateTask,
-                DeleteTask,
-                GetTask,
-                GetTasks,
+                CreateTaskTool,
+                UpdateTaskTool,
+                DeleteTaskTool,
+                GetTaskTool,
+                GetTasksTool,
             ],
             tools_folder="./tools",
             model="gpt-4o",

app/agents/TechnicalProjectManager/instructions.md

@@ -4,8 +4,18 @@ You are a Technical Project Manager working at VRSEN AI.
 
 Your Team consists of:
 
-- Devid (Code writing and execution)
-- BrowsingAgent (Online research and broswer actions)
+- Devid (Code execution): Useful for perfoming calculations, writing code, handling files and all other tasks that require or benefit from code execution.
+- BrowsingAgent (Online research and broswer actions): Useful for supplying up to date information, finding information, and providing support for tasks that require online research.
+
+## CAPABILITIES
+
+- Interacting with a pre-defined Notion database
+  - Create tasks
+  - Update tasks
+  - Delete tasks
+  - Get tasks
+  - Get task
+- Texting via Whatsapp (sending only)
 
 ## NOTION INSTRUCTIONS
 
@@ -32,6 +42,10 @@ Always use this database ID unless specified otherwise
 - Priority
 - Due Date
 
+## MANUAL INTERACTION WITH NOTION
+
+The user can manually
+
 ## WORKFLOWS
 
 ### Create a high level WBS

app/demo.py

@@ -19,9 +19,9 @@ from openai.types.beta.threads.runs import (
 from agency_swarm.tools import FileSearch, CodeInterpreter
 
 # Import our agents - using the same imports as in demo.ipynb
-from agents.NotionProjectAgent import NotionProjectAgent
 from agents.TechnicalProjectManager import TechnicalProjectManager
-from agents.ResearchAndReportAgent import ResearchAndReportAgent
+from agents.BrowsingAgent import BrowsingAgent
+from agents.Devid import Devid
 
 
 # Helper functions for file handling (from agency.py)
@@ -146,6 +146,10 @@ class ProjectManagementAgency(Agency):
             # Create state for iframe visibility
             iframe_state = gr.State(value=True)
 
+            # Add Title and Image inline
+            with gr.Row():
+                gr.Markdown("# 🐝🤖🐝 Project Management Agency Swarm Demo 🐝🤖🐝")
+
             # Add toggle button and refresh button at the top
             with gr.Row():
                 toggle_button = gr.Button(
@@ -619,15 +623,15 @@ def main():
 
     # Instantiate agents
     technical_project_manager = TechnicalProjectManager()
-    research_and_report_agent = ResearchAndReportAgent()
-    notion_project_agent = NotionProjectAgent()
+    browsing_agent = BrowsingAgent()
+    developer_agent = Devid()
 
     # Create the agency with our agents - using NotionAgency instead of Agency
     agency = ProjectManagementAgency(
         agency_chart=[
             technical_project_manager,
-            [technical_project_manager, notion_project_agent],
-            [technical_project_manager, research_and_report_agent],
+            [technical_project_manager, browsing_agent],
+            [technical_project_manager, developer_agent],
         ],
         shared_instructions="agency_manifesto.md",
     )

poetry.lock

@@ -2452,6 +2452,21 @@ files = [
     {file = "orjson-3.10.15.tar.gz", hash = "sha256:05ca7fe452a2e9d8d9d706a2984c95b9c2ebc5db417ce0b7a49b91d50642a23e"},
 ]
 
+[[package]]
+name = "outcome"
+version = "1.3.0.post0"
+description = "Capture the outcome of Python function calls."
+optional = false
+python-versions = ">=3.7"
+groups = ["main"]
+files = [
+    {file = "outcome-1.3.0.post0-py2.py3-none-any.whl", hash = "sha256:e771c5ce06d1415e356078d3bdd68523f284b4ce5419828922b6871e65eda82b"},
+    {file = "outcome-1.3.0.post0.tar.gz", hash = "sha256:9dcf02e65f2971b80047b377468e72a268e15c0af3cf1238e6ff14f7f91143b8"},
+]
+
+[package.dependencies]
+attrs = ">=19.2.0"
+
 [[package]]
 name = "packaging"
 version = "24.2"
@@ -3161,6 +3176,19 @@ files = [
 requests = ">=2.26.0"
 websocket-client = "!=0.49"
 
+[[package]]
+name = "pysocks"
+version = "1.7.1"
+description = "A Python SOCKS client module. See https://github.com/Anorov/PySocks for more information."
+optional = false
+python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*"
+groups = ["main"]
+files = [
+    {file = "PySocks-1.7.1-py27-none-any.whl", hash = "sha256:08e69f092cc6dbe92a0fdd16eeb9b9ffbc13cadfe5ca4c7bd92ffb078b293299"},
+    {file = "PySocks-1.7.1-py3-none-any.whl", hash = "sha256:2725bd0a9925919b9b51739eea5f9e2bae91e83288108a9ad338b2e3a4435ee5"},
+    {file = "PySocks-1.7.1.tar.gz", hash = "sha256:3f8804571ebe159c380ac6de37643bb4685970655d3bba243530d6558b799aa0"},
+]
+
 [[package]]
 name = "python-dateutil"
 version = "2.9.0.post0"
@@ -3789,6 +3817,26 @@ httpx = "*"
 [package.extras]
 dev = ["pytest"]
 
+[[package]]
+name = "selenium"
+version = "4.29.0"
+description = "Official Python bindings for Selenium WebDriver"
+optional = false
+python-versions = ">=3.9"
+groups = ["main"]
+files = [
+    {file = "selenium-4.29.0-py3-none-any.whl", hash = "sha256:ce5d26f1ddc1111641113653af33694c13947dd36c2df09cdd33f554351d372e"},
+    {file = "selenium-4.29.0.tar.gz", hash = "sha256:3a62f7ec33e669364a6c0562a701deb69745b569c50d55f1a912bf8eb33358ba"},
+]
+
+[package.dependencies]
+certifi = ">=2021.10.8"
+trio = ">=0.17,<1.0"
+trio-websocket = ">=0.9,<1.0"
+typing_extensions = ">=4.9,<5.0"
+urllib3 = {version = ">=1.26,<3", extras = ["socks"]}
+websocket-client = ">=1.8,<2.0"
+
 [[package]]
 name = "semantic-version"
 version = "2.10.0"
@@ -3911,6 +3959,18 @@ files = [
     {file = "sniffio-1.3.1.tar.gz", hash = "sha256:f4324edc670a0f49750a81b895f35c3adb843cca46f0530f79fc1babb23789dc"},
 ]
 
+[[package]]
+name = "sortedcontainers"
+version = "2.4.0"
+description = "Sorted Containers -- Sorted List, Sorted Dict, Sorted Set"
+optional = false
+python-versions = "*"
+groups = ["main"]
+files = [
+    {file = "sortedcontainers-2.4.0-py2.py3-none-any.whl", hash = "sha256:a163dcaede0f1c021485e957a39245190e74249897e2ae4b2aa38595db237ee0"},
+    {file = "sortedcontainers-2.4.0.tar.gz", hash = "sha256:25caa5a06cc30b6b83d11423433f65d1f9d76c4c6a0c90e3379eaa43b9bfdb88"},
+]
+
 [[package]]
 name = "sqlalchemy"
 version = "2.0.38"
@@ -4221,6 +4281,43 @@ files = [
 docs = ["myst-parser", "pydata-sphinx-theme", "sphinx"]
 test = ["argcomplete (>=3.0.3)", "mypy (>=1.7.0)", "pre-commit", "pytest (>=7.0,<8.2)", "pytest-mock", "pytest-mypy-testing"]
 
+[[package]]
+name = "trio"
+version = "0.29.0"
+description = "A friendly Python library for async concurrency and I/O"
+optional = false
+python-versions = ">=3.9"
+groups = ["main"]
+files = [
+    {file = "trio-0.29.0-py3-none-any.whl", hash = "sha256:d8c463f1a9cc776ff63e331aba44c125f423a5a13c684307e828d930e625ba66"},
+    {file = "trio-0.29.0.tar.gz", hash = "sha256:ea0d3967159fc130acb6939a0be0e558e364fee26b5deeecc893a6b08c361bdf"},
+]
+
+[package.dependencies]
+attrs = ">=23.2.0"
+cffi = {version = ">=1.14", markers = "os_name == \"nt\" and implementation_name != \"pypy\""}
+idna = "*"
+outcome = "*"
+sniffio = ">=1.3.0"
+sortedcontainers = "*"
+
+[[package]]
+name = "trio-websocket"
+version = "0.12.2"
+description = "WebSocket library for Trio"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "trio_websocket-0.12.2-py3-none-any.whl", hash = "sha256:df605665f1db533f4a386c94525870851096a223adcb97f72a07e8b4beba45b6"},
+    {file = "trio_websocket-0.12.2.tar.gz", hash = "sha256:22c72c436f3d1e264d0910a3951934798dcc5b00ae56fc4ee079d46c7cf20fae"},
+]
+
+[package.dependencies]
+outcome = ">=1.2.0"
+trio = ">=0.11"
+wsproto = ">=0.14"
+
 [[package]]
 name = "typer"
 version = "0.15.1"
@@ -4291,6 +4388,9 @@ files = [
     {file = "urllib3-2.3.0.tar.gz", hash = "sha256:f8c5449b3cf0861679ce7e0503c7b44b5ec981bec0d1d3795a07f1ba96f0204d"},
 ]
 
+[package.dependencies]
+pysocks = {version = ">=1.5.6,<1.5.7 || >1.5.7,<2.0", optional = true, markers = "extra == \"socks\""}
+
 [package.extras]
 brotli = ["brotli (>=1.0.9) ; platform_python_implementation == \"CPython\"", "brotlicffi (>=0.8.0) ; platform_python_implementation != \"CPython\""]
 h2 = ["h2 (>=4,<5)"]
@@ -4445,6 +4545,21 @@ files = [
     {file = "websockets-15.0.tar.gz", hash = "sha256:ca36151289a15b39d8d683fd8b7abbe26fc50be311066c5f8dcf3cb8cee107ab"},
 ]
 
+[[package]]
+name = "wsproto"
+version = "1.2.0"
+description = "WebSockets state-machine based protocol implementation"
+optional = false
+python-versions = ">=3.7.0"
+groups = ["main"]
+files = [
+    {file = "wsproto-1.2.0-py3-none-any.whl", hash = "sha256:b9acddd652b585d75b20477888c56642fdade28bdfd3579aa24a4d2c037dd736"},
+    {file = "wsproto-1.2.0.tar.gz", hash = "sha256:ad565f26ecb92588a3e43bc3d96164de84cd9902482b130d0ddbaa9664a85065"},
+]
+
+[package.dependencies]
+h11 = ">=0.9.0,<1"
+
 [[package]]
 name = "yarl"
 version = "1.18.3"
@@ -4565,4 +4680,4 @@ type = ["pytest-mypy"]
 [metadata]
 lock-version = "2.1"
 python-versions = "3.11.*"
-content-hash = "1c1f938d1d32680b9be677966c122aa79347700fce898884166344755f9f5157"
+content-hash = "ab6e6a07a618fe81606a52a4f332ef5a9d7abdd4e8597696eb3018e5c864aa1e"

pyproject.toml

@@ -19,6 +19,7 @@ openai = "^1.64.0"
 gradio = "^5.19.0"
 rootpath = "^0.1.1"
 typing-extensions = "^4.12.2"
+selenium = "^4.29.0"
 
 [tool.poetry.group.dev.dependencies]
 ipykernel = "^6.29.5"