(assistant_message_content)
| 143 | |
| 144 | |
| 145 | def extract_sql_query_from_message(assistant_message_content): |
| 146 | try: |
| 147 | data = json.loads(assistant_message_content) |
| 148 | except Exception as e: |
| 149 | print('e: ', e) |
| 150 | raise e |
| 151 | |
| 152 | if data.get('MissingData'): |
| 153 | return data |
| 154 | |
| 155 | sql = data['SQL'] |
| 156 | |
| 157 | return {"SQL": sql} |
| 158 | |
| 159 | |
| 160 | def extract_sql_from_markdown(assistant_message_content): |
no outgoing calls
no test coverage detected