Cleans message content to extract the SQL query
(assistant_message_content)
| 125 | return assistant_message |
| 126 | |
| 127 | def clean_sql_message_content(assistant_message_content): |
| 128 | """ |
| 129 | Cleans message content to extract the SQL query |
| 130 | """ |
| 131 | # Ignore text after the last SQL query terminator `;` |
| 132 | parts = assistant_message_content.split(";") |
| 133 | assistant_message_content = ";".join(parts[:-1]) |
| 134 | |
| 135 | # Remove prefix for corrected query assistant message |
| 136 | split_corrected_query_message = assistant_message_content.split(":") |
| 137 | if len(split_corrected_query_message) > 1: |
| 138 | sql_query = split_corrected_query_message[1].strip() |
| 139 | else: |
| 140 | sql_query = assistant_message_content |
| 141 | |
| 142 | return sql_query |
| 143 | |
| 144 | |
| 145 | def extract_sql_query_from_message(assistant_message_content): |
nothing calls this directly
no outgoing calls
no test coverage detected