-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathprompt_pool.py
143 lines (96 loc) · 2.72 KB
/
prompt_pool.py
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
QUERY_GENERATE = """
You are a great questioner of any text, and are adept at asking valuable and insightful questions.
Your goal is to generate 1 summary question for the text provided below.
The generated summary question should try to simulate the tone of human questions as much as possible,
and make sure that the generated question must be interrogative sentences and a summary question.
Important! Please make sure this text must be a complete and non-redundant answer to the generated summary question.
Please directly output the generated summary question, do not output irrelevant text.
DOCUMENT:
{document}
"""
QUERY_PROMPT_QMSUM = """
Refer to the following meeting transcripts and answer the question with brief but complete explanations.
SUPPORTING MATERIALS:
{materials}
QUESTION:
{question}
"""
QUERY_PROMPT_QMSUM_NORMAL = """
Refer to the following meeting transcripts and answer the question.
SUPPORTING MATERIALS:
{materials}
QUESTION:
{question}
"""
QUERY_PROMPT_SQuALITY = """
Refer to the following story and answer the question with brief but complete explanations.
SUPPORTING MATERIALS:
{materials}
QUESTION:
{question}
"""
QUERY_PROMPT_SQuALITY_NORMAL = """
Refer to the following story and answer the question.
SUPPORTING MATERIALS:
{materials}
QUESTION:
{question}
"""
QUERY_PROMPT_GOV = """
Refer to the following report and answer the question with brief but complete explanations.
SUPPORTING MATERIALS:
{materials}
QUESTION:
{question}
"""
QUERY_PROMPT_GOV_NORMAL = """
Refer to the following report and answer the question.
SUPPORTING MATERIALS:
{materials}
QUESTION:
{question}
"""
QUERY_PROMPT_WCEP = """
Refer to the following document and answer the question with brief but complete explanations.
SUPPORTING MATERIALS:
{materials}
QUESTION:
{question}
"""
QUERY_PROMPT_WCEP_NORMAL = """
Refer to the following document and answer the question.
SUPPORTING MATERIALS:
{materials}
QUESTION:
{question}
"""
QUERY_PROMPT_BOOK = """
Refer to the following narrative and answer the question with brief but complete explanations.
SUPPORTING MATERIALS:
{materials}
QUESTION:
{question}
"""
QUERY_PROMPT_BOOK_NORMAL = """
Refer to the following narrative and answer the question.
SUPPORTING MATERIALS:
{materials}
QUESTION:
{question}
"""
QUERY_PROMPT = {
"qmsum": QUERY_PROMPT_QMSUM,
"wcep": QUERY_PROMPT_WCEP,
"booksum": QUERY_PROMPT_BOOK,
"govreport": QUERY_PROMPT_GOV,
"squality": QUERY_PROMPT_SQuALITY
}
QUERY_PROMPT_NORMAL = {
"qmsum": QUERY_PROMPT_QMSUM_NORMAL,
"wcep": QUERY_PROMPT_WCEP_NORMAL,
"booksum": QUERY_PROMPT_BOOK_NORMAL,
"govreport": QUERY_PROMPT_GOV_NORMAL,
"squality": QUERY_PROMPT_SQuALITY_NORMAL
}
if __name__ == '__main__':
pass