Identify relevant tables for answering a natural language query via LM
(natural_language_query)
| 170 | |
| 171 | |
| 172 | def get_relevant_tables_from_lm(natural_language_query): |
| 173 | """ |
| 174 | Identify relevant tables for answering a natural language query via LM |
| 175 | """ |
| 176 | content = _get_table_selection_message_with_descriptions(natural_language_query) |
| 177 | messages = _get_table_selection_messages().copy() |
| 178 | messages.append({ |
| 179 | "role": "user", |
| 180 | "content": content |
| 181 | }) |
| 182 | |
| 183 | tables_json_str = _extract_text_from_markdown( |
| 184 | get_assistant_message( |
| 185 | messages=messages, |
| 186 | model="gpt-3.5-turbo-0301", |
| 187 | )["message"]["content"] |
| 188 | ) |
| 189 | tables = json.loads(tables_json_str).get("tables") |
| 190 | return tables |
no test coverage detected