Agent Composer YAML Cheat Sheet

inputs:
  query: str

outputs:
  response: str

nodes:
  generate:
    type: ResponseGenerationStep
    input_mapping:
      query: __inputs__#query

  __outputs__:
    type: output
    input_mapping:
      response: generate#response

node_name#output_key

node_name:
  type: StepType
  config: {}
  config_overrides: {}
  input_mapping:
    input: upstream#output

__outputs__:
  type: output
  input_mapping:
    result: node#output

ui_stream_types:
  retrievals: true
  generation: true

my_subgraph:
  inputs:
    query: str
  outputs:
    docs: List[str]
  nodes: ...

retriever:
  type: subgraph:my_subgraph
  input_mapping:
    query: __inputs__#query

conditional:
  type: ConditionalStep
  config:
    variable: score
    branches:
      - condition: ">= 0.8"
        executable: HighQuality
      - condition: "< 0.8"
        executable: LowQuality
  input_mapping:
    score: eval#score

create_message_history:
  type: CreateMessageHistoryStep
  input_mapping:
    query: __inputs__#query

research:
  type: AgenticResearchStep
  ui_stream_types:
    retrievals: true

tools_config:
  - name: search_docs
    step_config:
      type: SearchUnstructuredDataStep

generate:
  type: GenerateFromResearchStep
  ui_stream_types:
    generation: true

graph = ExecutableGraph.from_yaml(yaml_string)
graph.execute_graph({"query": "Hello"})