File size: 13,184 Bytes
7bf4b88 |
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 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 |
def prompts(dataset_name):
"""
input:
dataset_name: the name of the dataset
output:
prompt: the prompt for the dataset
"""
if dataset_name == "amazon":
sys_prompt = """
You are a reasoning graph finder agent. Your role is to:
1. Identify the underlying **meta-path** from a given question, which consists of the **entity types** at each reasoning step.
2. Extract the **content restriction** for each **entity type** based on the question. If there is no restriction for an entity type, leave its value empty.
You will be provided with a predefined **Entity Type List**. Only use the entity types from this list when constructing the meta-path and restrictions. Your response must be concise and strictly adhere to the specified **Output Format**.
"""
# Define the entity type list
entity_type_list = """
Entity Type List:
- **brand**
- **category**
- **product**
"""
demonstrations = """
Here are some examples:
**Question 1:**
What are some 5x4 inch team sports decals by Football Fanatics that are easy to apply on exterior surfaces?
**Answer 1:**
Metapath: brand -> product
Restriction: {"brand": ["Football Fanatics"], "product": ["5x4 inch team sports decals by Football Fanatics that are easy to apply on exterior surfaces"]}
**Question 2:**
Looking for men's insulated pants suitable for heavy rain, up to 10k water resistance, and with reinforced cuffs specifically for added protection in parking lots. Any suggestions?
**Answer 2:**
Metapath: category -> product
Restriction: {"category": ["pants"], "product": ["men\'s insulated pants suitable for heavy rain", "reinforced cuffs specifically for added protection in parking lots"]}
**Question 3:**
Can you recommend a dive mask that would work well with the Dive Mask Freediving Mask Spearfishing Mask Low Volume Mini Mask? I need the two masks to be compatible for my diving activities.
**Answer 3:**
Metapath: product -> product
Restriction: {"product": ["work well with the Dive Mask Freediving Mask Spearfishing Mask Low Volume Mini Mask", "compatible for diving activities", "a dive mask"]}
**Question 4:**
What are some UV protective women's golf jackets from PUMA? I'm cautious about skin protection, but I'm a big fan of PUMA.
**Answer 4:**
Metapath: brand -> product <- category
Restriction: {"brand": ["PUMA"], "category": ["golf jackets"], "product": ["UV protective women\'s golf jackets from PUMA", "skin protection"]}
"""
output_format = """
**Output Format**
Metapath: "",
Restriction: {}
"""
sys_content = sys_prompt + entity_type_list + output_format + demonstrations
elif dataset_name == "mag":
sys_prompt = """
You are a reasoning finder agent. Your role is to:
1. Identify the underlying **meta-path** from a given question, which consists of the **entity types** at each reasoning step.
2. Extract the **content restriction** for each **entity type** based on the question. If there is no restriction for an entity type, leave its value empty.
You will be provided with a predefined **Entity Type List**. Only use the entity types from this list when constructing the meta-path and restrictions. Your response must be concise and strictly adhere to the specified **Output Format**.
"""
entity_type_list = """
Entity Type List:
- **paper**
- **author**
- **institution**
- **field_of_study**
"""
output_format = """
**Output Format**
Metapath: "",
Restriction: {}
"""
demonstrations = """
Here are some examples:
**Question 1:**
Show me research articles on the association of quasi-periodic oscillations (QPOs) with noise in celestial bodies within the context of Bicoherence.
**Answer 1:**
Metapath: field_of_study -> paper
Restriction: {"field_of_study": ["oscillations", "physics"], "paper": ["association of quasi-periodic oscillations (QPOs) with noise in celestial bodies within the context of Bicoherence"]}
**Question 2:**
What research on water absorption in different frequency ranges have been referenced or deemed significant in the paper entitled 'High-resolution terahertz atmospheric water vapor continuum measurements?
**Answer 2:**
Metapath: paper -> paper
Restriction: {"paper": ["water absorption in different frequency ranges", "High-resolution terahertz atmospheric water vapor continuum measurements"]}
**Question 3:**
Show me publications by A.J. Turvey on the topic of supersymmetry particle searches.
**Answer 3:**
Metapath: author -> paper
Restriction: {"author": ["A.J. Turvey"], "paper": ["supersymmetry particle searches"]}
**Question 4:**
Looking for papers co-authored by someone involved in "Strain transferring mechanism analysis of the substrate-bonded FBG sensor". The papers should be in the similar field which is fiber optic strain sensors, and further discuss their development and application.
**Answer 4:**
Metapath: paper -> author -> paper
Restriction: {"paper": ["Strain transferring mechanism analysis of the substrate-bonded FBG sensor", "in the similar field which is fiber optic strain sensors, and further discuss their development and application"]}
**Question 5:**
Can you find any publications by the authors of "Tradeoffs in the Realization of Electrically Pumped Vertical External Cavity Surface Emitting Lasers," that delve into the topic of hybrid quantum well/quantum dot structures in the context of lasers?
**Answer 5:**
Metapath: paper -> author -> paper <- field_of_study
Restriction: {"paper": ["Tradeoffs in the Realization of Electrically Pumped Vertical External Cavity Surface Emitting Lasers", "hybrid quantum well/quantum dot structures in the context of lasers"], "field_of_study": ["lasers", "hybrid quantum well/quantum dot", "physics", "Emission spectrum"]}
**Question 6:**
Which publications from Altair Engineering authors focus on improving directional sensitivity across a wide range of frequencies?
**Answer 6:**
Metapath: institution -> author -> paper
Restriction: {"institution": ["Altair Engineering"], "paper": ["improving directional sensitivity across a wide range of frequencies"]}
**Question 7:**
Publications by Carlisle Companies authors on satellite instrumentation and space performance
**Answer 7:**
Metapath: institution -> author -> paper <- field_of_study
Restriction: {"institution": ["Carlisle Companies"], "paper": ["satellite instrumentation and space performance"], "field_of_study": ["satellite", 'space performance']}
"""
sys_content = sys_prompt + entity_type_list + output_format + demonstrations
elif dataset_name == "prime":
sys_prompt = """
You are a triplets extractor. Given a list of triplets and a query, please
1. extract the triplets contained in the query and give a list to me.
2. make a restriction list which contains the description of the entity in the query.
3. tell me which entity the query is asking for.
Your response must be concise and strictly adhere to the specified **Output Format**.
"""
triplets = """
Triplets list:
[('anatomy', 'expression absent', 'gene/protein'),
('anatomy', 'expression present', 'gene/protein'),
('anatomy', 'parent-child', 'anatomy'),
('biological_process', 'interacts with', 'exposure'),
('biological_process', 'interacts with', 'gene/protein'),
('biological_process', 'parent-child', 'biological_process'),
('cellular_component', 'interacts with', 'exposure'),
('cellular_component', 'interacts with', 'gene/protein'),
('cellular_component', 'parent-child', 'cellular_component'),
('disease', 'associated with', 'gene/protein'),
('disease', 'contraindication', 'drug'),
('disease', 'indication', 'drug'),
('disease', 'linked to', 'exposure'),
('disease', 'off-label use', 'drug'),
('disease', 'parent-child', 'disease'),
('disease', 'phenotype absent', 'effect/phenotype'),
('disease', 'phenotype present', 'effect/phenotype'),
('drug', 'carrier', 'gene/protein'),
('drug', 'contraindication', 'disease'),
('drug', 'enzyme', 'gene/protein'),
('drug', 'indication', 'disease'),
('drug', 'off-label use', 'disease'),
('drug', 'side effect', 'effect/phenotype'),
('drug', 'synergistic interaction', 'drug'),
('drug', 'target', 'gene/protein'),
('drug', 'transporter', 'gene/protein'),
('effect/phenotype', 'associated with', 'gene/protein'),
('effect/phenotype', 'parent-child', 'effect/phenotype'),
('effect/phenotype', 'phenotype absent', 'disease'),
('effect/phenotype', 'phenotype present', 'disease'),
('effect/phenotype', 'side effect', 'drug'),
('exposure', 'interacts with', 'biological_process'),
('exposure', 'interacts with', 'cellular_component'),
('exposure', 'interacts with', 'gene/protein'),
('exposure', 'interacts with', 'molecular_function'),
('exposure', 'linked to', 'disease'),
('exposure', 'parent-child', 'exposure'),
('gene/protein', 'associated with', 'disease'),
('gene/protein', 'associated with', 'effect/phenotype'),
('gene/protein', 'carrier', 'drug'),
('gene/protein', 'enzyme', 'drug'),
('gene/protein', 'expression absent', 'anatomy'),
('gene/protein', 'expression present', 'anatomy'),
('gene/protein', 'interacts with', 'biological_process'),
('gene/protein', 'interacts with', 'cellular_component'),
('gene/protein', 'interacts with', 'exposure'),
('gene/protein', 'interacts with', 'molecular_function'),
('gene/protein', 'interacts with', 'pathway'),
('gene/protein', 'ppi', 'gene/protein'),
('gene/protein', 'target', 'drug'),
('gene/protein', 'transporter', 'drug'),
('molecular_function', 'interacts with', 'exposure'),
('molecular_function', 'interacts with', 'gene/protein'),
('molecular_function', 'parent-child', 'molecular_function'),
('pathway', 'interacts with', 'gene/protein'),
('pathway', 'parent-child', 'pathway')]
"""
output_format = """
**Output Format**
Triplets: []
Restriction: {}
Target: ""
"""
demonstrations = """
Here are some examples:
**Question 1:**
Search for conditions that lack any associated treatment medications and have a connection to the formation of Onion bulb structures.
**Answer 1:**
Triplets: [["effect/phenotype", "phenotype present", "disease"], ["drug", "off-label use", "disease"]]
Restriction: {'effect/phenotype': ['formation of Onion bulb structures']}
Target: disease
**Question 2:**
What drug should be avoided for focal hand dystonia and also targets the ORM2 gene/protein?
**Answer 2:**
Triplets: [["disease", "contraindication", "drug"], ["drug", "target", "gene/protein"]]
Restriction: {"disease": ["focal hand dystonia"], "gene/protein": ["ORM2"]}
Target: drug
**Question 3:**
Could my hip muscle weakness be a sign of the mitochondrial disease my mother has, or another related condition?
Triplets: [["disease", "parent-child", "disease"], ["effect/phenotype", "phenotype present", "disease"]]
**Answer 3:**
Restriction: {"anatomy": ["hip muscle"], "disease": ["mitochondrial disease my mother has", "another related condition"], "effect/phenotype": ["weakness"]}
Target: disease
**Question 4:**
Can you find the genes and proteins involved with the activity of dolichyl-phosphate-mannose-dependent alpha-1,6-mannosyltransferase?
**Answer 4:**
Triplets: [["molecular_function", "interacts with", "gene/protein"]]
Restriction: {"molecular_function": ["dolichyl-phosphate-mannose-dependent alpha-1,6-mannosyltransferase"]}
Target: gene/protein
"""
sys_content = sys_prompt + triplets + output_format + demonstrations
return sys_content |