-
Notifications
You must be signed in to change notification settings - Fork 21
/
report_pipeline.yaml
69 lines (63 loc) · 1.98 KB
/
report_pipeline.yaml
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
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"