Log LLM usage

routes/sql_explorer.py

@@ -27,6 +27,19 @@ def record_sql_audit(query, success, error_message=None):
     except Exception as e:
         current_app.logger.error(f"Failed to record SQL audit: {e}")
 
+def record_llm_audit(prompt, response, model, success, error_message=None):
+    """Records an LLM interaction in the audit table."""
+    try:
+        person_id = getattr(current_user, 'id', None)
+        ip_address = get_client_ip()
+        sql = """
+            INSERT INTO llm_audit (person_id, prompt, response, model, ip_address, success, error_message)
+            VALUES (%s, %s, %s, %s, %s, %s, %s)
+        """
+        db.execute(sql, [person_id, prompt, response, model, ip_address, success, error_message], commit=True)
+    except Exception as e:
+        current_app.logger.error(f"Failed to record LLM audit: {e}")
+
 def _execute_sql(query):
     """Executes arbitrary SQL query, returning results, columns, and error."""
     results, columns, error = None, [], None
@@ -76,6 +89,7 @@ def _generate_sql_from_natural_language(natural_query):
     api_url = f"https://generativelanguage.googleapis.com/v1beta/models/{gemni_model}:generateContent?key={api_key}"
     headers = {'Content-Type': 'application/json'}
 
+    prompt = natural_query
     try:
         # Get and format schema
         schema_info = db.schema.get_schema_info()
@@ -128,14 +142,20 @@ Return ONLY the SQL query, without any explanation or surrounding text/markdown.
         filtered_lines = [line for line in sql_lines if not line.strip().startswith('--')]
         final_sql = "\n".join(filtered_lines).strip()
 
-        return final_sql, None
+        generated_sql, error = final_sql, None
+        record_llm_audit(prompt, generated_sql, gemni_model, True)
+        return generated_sql, error
 
     except requests.exceptions.RequestException as e:
         current_app.logger.error(f"Gemini API request error: {e}")
-        return None, f"Error communicating with API: {e}"
+        error_msg = f"Error communicating with API: {e}"
+        record_llm_audit(prompt, None, gemni_model, False, error_msg)
+        return None, error_msg
     except (KeyError, IndexError, Exception) as e:
         current_app.logger.error(f"Error processing Gemini API response: {e} - Response: {response_data if 'response_data' in locals() else 'N/A'}")
-        return None, f"Error processing API response: {e}"
+        error_msg = f"Error processing API response: {e}"
+        record_llm_audit(prompt, None, gemni_model, False, error_msg)
+        return None, error_msg
 
 
 # --- Routes ---