()
| 142 | |
| 143 | |
| 144 | def _get_table_selection_messages(): |
| 145 | # default_messages = [{ |
| 146 | # "role": "system", |
| 147 | # "content": ( |
| 148 | # f""" |
| 149 | # You are a helpful assistant for identifying relevant SQL tables to use for answering a natural language query. |
| 150 | # You respond in JSON format with your answer in a field named \"tables\" which is a list of strings. |
| 151 | # Respond with an empty list if you cannot identify any relevant tables. |
| 152 | # Make sure to write your answer in markdown format. |
| 153 | # The following are descriptions of available tables and enums: |
| 154 | # --------------------- |
| 155 | # {get_table_schemas_str()} |
| 156 | # --------------------- |
| 157 | # """ |
| 158 | # ) |
| 159 | # }] |
| 160 | default_messages = [] |
| 161 | default_messages.extend(get_few_shot_messages(mode="table_selection")) |
| 162 | return default_messages |
| 163 | |
| 164 | |
| 165 | def _extract_text_from_markdown(text): |
no test coverage detected