report_pipeline.yaml

models:
  - label: "default"
    name: "OpenAI"
    args:
      model: "gpt-3.5-turbo"
      max_tokens: 4096
  - label: "smart"
    name: "OpenAI"
    args:
      model: "gpt-4"
      max_tokens: 4096

signatures:
  - name: "ProcessDataSignature"
    docstring: "Processes raw data to synthesize into a structured format suitable for report generation."
    inputs:
      - name: "raw_data"
        desc: "Raw data input that needs processing."
      - name: "data_format"
        desc: "The desired format of the output data."
    outputs:
      - name: "processed_data"
        desc: "Data processed into a structured format."

  - name: "GenerateReportSignature"
    docstring: "Generates a comprehensive report from structured data in the output_format. IT MUST BE IN THE output_format"
    inputs:
      - name: "processed_data"
        desc: "Structured data to be included in the report."
      - name: "report_template"
        desc: "Template specifying the report's format and structure."
      - name: "output_format"
        desc: "The desired format of the output report."
    outputs:
      - name: "report"
        desc: "The final report generated from the structured data in the output_format."

modules:
  - name: "DataProcessorModule"
    signature: "ProcessDataSignature"
    predictor: "Predict"
    model: "default"
    args:
      - name: "raw_data"
        value: "{{ user_input_data }}"
      - name: "data_format"
        value: "JSON"

  - name: "ReportGeneratorModule"
    signature: "GenerateReportSignature"
    predictor: "Predict"
    model: "default"
    args:
      - name: "report_template"
        value: "StandardReportTemplate"
      - name: "output_format"
        value: "HTML"

steps:
  - module: "DataProcessorModule"
    args:
      raw_data: "id, name, age\n1, John, 25\n2, Jane, 30"
      data_format: "YAML"

  - module: "ReportGeneratorModule"
    args:
      processed_data: "{{ processed_data }}"
      output_format: "CSV"
      report_template: "templates/standard_report.csv"