(input_text, sql_script, failure_message, attempt_number, app_name, session_id=None)
| 150 | |
| 151 | @failsoft |
| 152 | def log_sql_failure(input_text, sql_script, failure_message, attempt_number, app_name, session_id=None): |
| 153 | if not EVENTS_ENGINE: |
| 154 | return {"status": "no engine"} |
| 155 | |
| 156 | params = { |
| 157 | "input_text": input_text, |
| 158 | "sql_script": sql_script, |
| 159 | "failure_message": failure_message, |
| 160 | "attempt_number": attempt_number, |
| 161 | "app_name": app_name, |
| 162 | "session_id": session_id |
| 163 | } |
| 164 | |
| 165 | insert_query = text(""" |
| 166 | INSERT INTO sql_failures (input_text, sql_script, failure_message, attempt_number, app_name, session_id) |
| 167 | VALUES (:input_text, :sql_script, :failure_message, :attempt_number, :app_name, :session_id) |
| 168 | """) |
| 169 | |
| 170 | with EVENTS_ENGINE.connect() as conn: |
| 171 | conn.execute(insert_query, params) |
| 172 | conn.commit() |
| 173 | |
| 174 | return {"status": "success"} |
| 175 | |
| 176 | @failsoft |
| 177 | def log_suggested_query(input_text="", reason="", app_name="", parent_id=None, suggested_query="", prompt="", model="", session_id=None): |
no outgoing calls
no test coverage detected