(input_text="", reason="", app_name="", parent_id=None, suggested_query="", prompt="", model="", session_id=None)
| 175 | |
| 176 | @failsoft |
| 177 | def log_suggested_query(input_text="", reason="", app_name="", parent_id=None, suggested_query="", prompt="", model="", session_id=None): |
| 178 | if not EVENTS_ENGINE: |
| 179 | return None |
| 180 | |
| 181 | params = { |
| 182 | "input_text": input_text, |
| 183 | "reason": reason, |
| 184 | "app_name": app_name, |
| 185 | "parent_id": parent_id, |
| 186 | "suggested_query": suggested_query, |
| 187 | "prompt": prompt, |
| 188 | "model": model, |
| 189 | "session_id": session_id, |
| 190 | } |
| 191 | |
| 192 | insert_query = text(""" |
| 193 | INSERT INTO suggested_queries (input_text, reason, app_name, parent_id, suggested_query, prompt, model, session_id) |
| 194 | VALUES (:input_text, :reason, :app_name, :parent_id, :suggested_query, :prompt, :model, :session_id) |
| 195 | returning id |
| 196 | """) |
| 197 | with EVENTS_ENGINE.connect() as conn: |
| 198 | # get the ID back |
| 199 | result = conn.execute(insert_query, params) |
| 200 | conn.commit() |
| 201 | row = result.fetchone() |
| 202 | generation_id = row[0] |
| 203 | |
| 204 | return str(generation_id) |
| 205 | |
| 206 | @failsoft |
| 207 | def update_suggestion_as_used(suggestion_id): |
no outgoing calls
no test coverage detected