seclab-taskflow-agent/examples/taskflows/example_triage_taskflow.yaml at main · GitHubSecurityLab/seclab-taskflow-agent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# SPDX-FileCopyrightText: GitHub, Inc.
# SPDX-License-Identifier: MIT

# a simple example of the triage Agent pattern
seclab-taskflow-agent:
  version: "1.0"
  filetype: taskflow

taskflow:
  - task:
      must_complete: true
      agents:
        - seclab_taskflow_agent.personalities.assistant
      user_prompt: |
        Store the json array ["apples", "oranges", "bananas"] in the `fruits` memory key.
      env:
        MEMCACHE_STATE_DIR: "example_taskflow/"
        MEMCACHE_BACKEND: "dictionary_file"
      toolboxes:
        - seclab_taskflow_agent.toolboxes.memcache
  - task:
      must_complete: true
      agents:
        - seclab_taskflow_agent.personalities.assistant
      user_prompt: |
        Retrieve the contents of the `fruits` memory key.
      env:
        MEMCACHE_STATE_DIR: "example_taskflow/"
        MEMCACHE_BACKEND: "dictionary_file"
      toolboxes:
        - seclab_taskflow_agent.toolboxes.memcache
  - task:
      repeat_prompt: true
      agents:
        # primary agent for this task
        - examples.personalities.example_triage_agent
        # handoff agents
        - examples.personalities.apple_expert
        - examples.personalities.orange_expert
        - examples.personalities.banana_expert
      user_prompt: |
        Tell me more about how {{ result }} are grown.