Cleans message content to extract the SQL query
(assistant_message_content)
| 63 | |
| 64 | |
| 65 | def clean_message_content(assistant_message_content): |
| 66 | """ |
| 67 | Cleans message content to extract the SQL query |
| 68 | """ |
| 69 | # Ignore text after the SQL query terminator `;` |
| 70 | assistant_message_content = assistant_message_content.split(";")[0] |
| 71 | |
| 72 | # Remove prefix for corrected query assistant message |
| 73 | split_corrected_query_message = assistant_message_content.split(":") |
| 74 | if len(split_corrected_query_message) > 1: |
| 75 | sql_query = split_corrected_query_message[1].strip() |
| 76 | else: |
| 77 | sql_query = assistant_message_content |
| 78 | return sql_query |
| 79 | |
| 80 | |
| 81 | def extract_sql_query_from_message(assistant_message_content): |
no outgoing calls
no test coverage detected