Use paths for referencing other yaml files.

Author: kevinbackhouse

main.py

@@ -10,6 +10,7 @@
 import re
 import json
 import uuid
+import pathlib
 
 from agent import DEFAULT_MODEL, TaskRunHooks, TaskAgentHooks
 #from agents.run import DEFAULT_MAX_TURNS # XXX: this is 10, we need more than that
@@ -628,11 +629,12 @@ async def _deploy_task_agents(resolved_agents, prompt):
                     break
 
 if __name__ == '__main__':
+    cwd = pathlib.Path.cwd()
     available_tools = AvailableTools(
-        YamlParser('personalities').get_yaml_dict() |
-        YamlParser('taskflows').get_yaml_dict() |
-        YamlParser('prompts').get_yaml_dict(dir_namespace=True) |
-        YamlParser('toolboxes').get_yaml_dict(recurse=True))
+        YamlParser(cwd).get_yaml_dict((cwd/'personalities').rglob('*')) |
+        YamlParser(cwd).get_yaml_dict((cwd/'taskflows').rglob('*')) |
+        YamlParser(cwd).get_yaml_dict((cwd/'prompts').rglob('*')) |
+        YamlParser(cwd).get_yaml_dict((cwd/'toolboxes').rglob('*')))
 
     p, t, l, user_prompt, help_msg = parse_prompt_args(available_tools)
 

mcp_utils.py

@@ -221,7 +221,7 @@ class MCPNamespaceWrap:
     def __init__(self, confirms, obj):
         self.confirms = confirms
         self._obj = obj
-        self.namespace = f"{obj.name.upper().replace(' ', '_')}_"
+        self.namespace = f"{obj.name.upper().replace(' ', '_').replace('/','-')}_"
 
     def __getattr__(self, name):
         attr = getattr(self._obj, name)

personalities/c_auditer.yaml

@@ -14,5 +14,5 @@ task: |
   your findings where possible.
 
 toolboxes:
-  - memcache
-  - codeql
+  - toolboxes/memcache.yaml
+  - toolboxes/codeql.yaml

personalities/examples/echo.yaml

@@ -9,5 +9,5 @@ task: |
   Echo user inputs using the echo tools.
   
 toolboxes:
-  - echo
+  - toolboxes/echo.yaml
 

taskflows/CVE-2023-2283/CVE-2023-2283.yaml

@@ -7,16 +7,16 @@ taskflow:
       must_complete: true
       headless: true
       agents:
-        - assistant
+        - personalities/assistant.yaml
       user_prompt: |
         Clear the memory cache.
       toolboxes:
-        - memcache
+        - toolboxes/memcache.yaml
   - task:
       model: gpt-4.1
       must_complete: false
       agents:
-        - c_auditer
+        - personalities/c_auditer.yaml
       user_prompt: |
         You are auditing code using the `libssh-mirror/libssh-codeql`
         CodeQL database.
@@ -76,15 +76,15 @@ taskflow:
         6. Make small and concise single line notes while you work. Update the
         existing value for `notes` in memory as you work.
       toolboxes:
-        - codeql
-        - memcache        
+        - toolboxes/codeql.yaml
+        - toolboxes/memcache.yaml
   - task:
       must_complete: true
       agents:
-        - c_auditer
+        - personalities/c_auditer.yaml
       user_prompt: |
         Fetch your audit notes from memory using the `notes`
         key. Do not perform any additional security review, only show me your
         notes.
       toolboxes:
-        - memcache
+        - toolboxes/memcache.yaml

taskflows/examples/echo.yaml

@@ -8,13 +8,13 @@ taskflow:
       max_steps: 5
       must_complete: true
       agents:
-        - echo
+        - personalities/examples/echo.yaml
       user_prompt: |
         Hello
   - task:
       must_complete: true
       agents:
-        - echo
+        - personalities/examples/echo.yaml
       user_prompt: |
         Goodbye
       env:

taskflows/examples/example.yaml

@@ -12,8 +12,8 @@ taskflow:
       must_complete: true
       # taskflows can set a primary (first entry) and handoff (additional entries) agent
       agents:
-        - c_auditer
-        - fruit_expert
+        - personalities/c_auditer.yaml
+        - personalities/examples/fruit_expert.yaml
       user_prompt: |
         Store an example vulnerable C program that uses `strcpy` in the
         `vulnerable_c_example` memory key and explain why `strcpy`
@@ -34,13 +34,13 @@ taskflow:
       # this normally only has the memcache toolbox, but we extend it here with
       # the GHSA toolbox
       toolboxes:
-        - memcache
-        - codeql
+        - toolboxes/memcache.yaml
+        - toolboxes/codeql.yaml
   - task:
       must_complete: true
       model: gpt-4.1
       agents:
-        - c_auditer
+        - personalities/c_auditer.yaml
       user_prompt: |
         Retrieve C code for security review from the `vulnerable_c_example`
         memory key and perform a review.
@@ -50,7 +50,7 @@ taskflow:
         MEMCACHE_STATE_DIR: "example_taskflow/"
         MEMCACHE_BACKEND: "dictionary_file"
       toolboxes:
-        - memcache
+        - toolboxes/memcache.yaml
       # headless mode does not prompt for tool call confirms configured for a server
       # note: this will auto-allow, if you want control over potentially dangerous
       # tool calls, then you should NOT run a task in headless mode (default: false)
@@ -63,6 +63,6 @@ taskflow:
   - task:
       repeat_prompt: true
       agents:
-        - assistant
+        - personalities/assistant.yaml
       user_prompt: |
         What kind of fruit is {{ RESULT }}?

taskflows/examples/example_globals.yaml

@@ -7,7 +7,7 @@ globals:
 taskflow:
   - task:
       agents:
-        - fruit_expert
+        - personalities/examples/fruit_expert.yaml
       user_prompt: |
         Tell me more about {{ GLOBALS_fruit }}.
         

taskflows/examples/example_inputs.yaml

@@ -5,7 +5,7 @@ seclab-taskflow-agent:
 taskflow:
   - task:
       agents:
-        - fruit_expert
+        - personalities/examples/fruit_expert.yaml
       inputs:
         fruit: apples
       user_prompt: |

taskflows/examples/example_large_list_result_iter.yaml

@@ -7,16 +7,16 @@ taskflow:
       exclude_from_context: true
       must_complete: true
       agents:
-        - assistant
+        - personalities/assistant.yaml
       user_prompt: |
         Fetch all the open pull requests from `github/codeql` github repository. 
         You do not need to provide a summary of the results.
       toolboxes:
-        - github_official
+        - toolboxes/github_official.yaml
   - task:
       must_complete: true
       repeat_prompt: true
       agents:
-        - echo
+        - toolboxes/echo.yaml
       user_prompt: |
         Echo this:  The title is {{ RESULT_title }} and the url is {{ RESULT_url }}.