From 81f15fb1f9631f0768d08d64b0286e24e454cf62 Mon Sep 17 00:00:00 2001
From: Chris Clark <chris@untrod.com>
Date: Thu, 29 Aug 2024 09:34:31 -0400
Subject: [PATCH] assistant history

---
 HISTORY.rst                                   |   1 +
 explorer/assistant/forms.py                   |   2 +-
 explorer/src/js/assistant.js                  |  14 +-
 explorer/src/js/explorer.js                   |   2 +-
 explorer/src/js/tableDescription.js           |   7 +
 explorer/src/scss/assistant.scss              |  12 +
 explorer/src/scss/choices.scss                | 323 ++++++++++++++++++
 explorer/src/scss/styles.scss                 |   3 +-
 explorer/src/scss/variables.scss              |   1 +
 .../assistant/table_description_form.html     |  71 ++--
 explorer/templates/explorer/assistant.html    |  15 +-
 explorer/tests/test_assistant.py              | 134 ++++++++
 12 files changed, 539 insertions(+), 46 deletions(-)
 create mode 100644 explorer/src/scss/choices.scss

diff --git a/HISTORY.rst b/HISTORY.rst
index 7f2b7f31..764b602d 100644
--- a/HISTORY.rst
+++ b/HISTORY.rst
@@ -21,6 +21,7 @@ vNext
   - Relevant tables are added client-side visually, in real time, based on what's in the SQL editor and/or any tables
     mentioned in the assistant request. The dependency on sql_metadata is therefore removed, as server-side SQL parsing
     is no longer necessary.
+  - Ability to view Assistant request/response history.
   - Improved system prompt that emphasizes the particular SQL dialect being used.
   - Addresses issue #657.
 
diff --git a/explorer/assistant/forms.py b/explorer/assistant/forms.py
index f4c69780..ddcaffea 100644
--- a/explorer/assistant/forms.py
+++ b/explorer/assistant/forms.py
@@ -25,7 +25,7 @@ def __init__(self, *args, **kwargs):
 
         f = forms.ChoiceField(
             choices=choices,
-            widget=forms.Select(attrs={"class": "form-select"})
+            widget=forms.Select(attrs={"class": "form-select", "data-placeholder": "Select table"})
         )
 
         # We don't actually care about validating the 'choices' that the ChoiceField does by default.
diff --git a/explorer/src/js/assistant.js b/explorer/src/js/assistant.js
index eb052c3c..8e7ce159 100644
--- a/explorer/src/js/assistant.js
+++ b/explorer/src/js/assistant.js
@@ -46,8 +46,6 @@ function setupTableList() {
             removeItemButton: true,
             searchEnabled: true,
             shouldSort: false,
-            placeholder: true,
-            placeholderValue: 'Relevant tables',
             position: 'bottom'
         });
 
@@ -132,8 +130,10 @@ export function setUpAssistant(expand = false) {
 
 function getAssistantHistory() {
 
+    const historyModalId = 'historyModal';
+
     // Remove any existing modal with the same ID
-    const existingModal = document.getElementById('historyModal');
+    const existingModal = document.getElementById(historyModalId);
     if (existingModal) {
         existingModal.remove()
     }
@@ -186,7 +186,7 @@ function getAssistantHistory() {
 
         // Insert the table into a new Bootstrap modal
         const modalHtml = `
-            <div class="modal fade" id="historyModal" tabindex="-1" aria-labelledby="historyModalLabel" aria-hidden="true">
+            <div class="modal fade" id="${historyModalId}" tabindex="-1" aria-labelledby="historyModalLabel" aria-hidden="true">
                 <div class="modal-dialog modal-lg">
                     <div class="modal-content">
                         <div class="modal-header">
@@ -204,12 +204,10 @@ function getAssistantHistory() {
             </div>
         `;
 
-        // Append the modal to the body
         document.body.insertAdjacentHTML('beforeend', modalHtml);
-
-        // Show the modal
-        const historyModal = new bootstrap.Modal(document.getElementById('historyModal'));
+        const historyModal = new bootstrap.Modal(document.getElementById(historyModalId));
         historyModal.show();
+
     })
     .catch(error => {
         console.error('There was a problem with the fetch operation:', error);
diff --git a/explorer/src/js/explorer.js b/explorer/src/js/explorer.js
index 02968d37..823d25fd 100644
--- a/explorer/src/js/explorer.js
+++ b/explorer/src/js/explorer.js
@@ -25,7 +25,7 @@ function updateSchema() {
         });
     });
 
-    $("#schema_frame").attr("src", `../schema/${getConnElement().value}`);
+    $("#schema_frame").attr("src", `${window.baseUrlPath}schema/${getConnElement().value}`);
 }
 
 
diff --git a/explorer/src/js/tableDescription.js b/explorer/src/js/tableDescription.js
index 0b59ffe9..7aadcbe8 100644
--- a/explorer/src/js/tableDescription.js
+++ b/explorer/src/js/tableDescription.js
@@ -35,7 +35,14 @@ function populateTableList() {
     });
 }
 
+function updateSchema() {
+    document.getElementById("schema_frame").src = `${window.baseUrlPath}schema/${getConnElement().value}`;
+}
+
 export function setupTableDescription() {
     getConnElement().addEventListener('change', populateTableList);
     populateTableList();
+
+    getConnElement().addEventListener('change', updateSchema);
+    updateSchema();
 }
diff --git a/explorer/src/scss/assistant.scss b/explorer/src/scss/assistant.scss
index 92fdedd2..4f9fa7ea 100644
--- a/explorer/src/scss/assistant.scss
+++ b/explorer/src/scss/assistant.scss
@@ -18,3 +18,15 @@
     max-height: 120px;
     overflow: hidden;
 }
+
+.assistant-icons {
+    width: 1rem;
+    position: absolute;
+    right: .75rem;
+}
+
+#table-list {
+    .choices__inner {
+        min-height: 7rem !important;
+    }
+}
diff --git a/explorer/src/scss/choices.scss b/explorer/src/scss/choices.scss
new file mode 100644
index 00000000..dc9ccaf6
--- /dev/null
+++ b/explorer/src/scss/choices.scss
@@ -0,0 +1,323 @@
+@import "variables";
+
+.choices {
+  position: relative;
+  overflow: hidden;
+  margin-bottom: 24px;
+}
+.choices:focus {
+  outline: none;
+}
+.choices:last-child {
+  margin-bottom: 0;
+}
+.choices.is-open {
+  overflow: visible;
+}
+.choices.is-disabled .choices__inner,
+.choices.is-disabled .choices__input {
+  background-color: #eaeaea;
+  cursor: not-allowed;
+  -webkit-user-select: none;
+          user-select: none;
+}
+.choices.is-disabled .choices__item {
+  cursor: not-allowed;
+}
+.choices [hidden] {
+  display: none !important;
+}
+
+.choices[data-type*=select-one] {
+  cursor: pointer;
+}
+.choices[data-type*=select-one] .choices__inner {
+  padding-bottom: 7.5px;
+}
+.choices[data-type*=select-one] .choices__input {
+  display: block;
+  width: 100%;
+  padding: 10px;
+  border-bottom: 1px solid #ddd;
+  background-color: #fff;
+  margin: 0;
+}
+.choices[data-type*=select-one] .choices__button {
+  background-image: url("data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMjEiIGhlaWdodD0iMjEiIHZpZXdCb3g9IjAgMCAyMSAyMSIgeG1sbnM9Imh0dHA6Ly93d3cudzMub3JnLzIwMDAvc3ZnIj48ZyBmaWxsPSIjMDAwIiBmaWxsLXJ1bGU9ImV2ZW5vZGQiPjxwYXRoIGQ9Ik0yLjU5Mi4wNDRsMTguMzY0IDE4LjM2NC0yLjU0OCAyLjU0OEwuMDQ0IDIuNTkyeiIvPjxwYXRoIGQ9Ik0wIDE4LjM2NEwxOC4zNjQgMGwyLjU0OCAyLjU0OEwyLjU0OCAyMC45MTJ6Ii8+PC9nPjwvc3ZnPg==");
+  padding: 0;
+  background-size: 8px;
+  position: absolute;
+  top: 50%;
+  right: 0;
+  margin-top: -10px;
+  margin-right: 25px;
+  height: 20px;
+  width: 20px;
+  border-radius: 10em;
+  opacity: 0.25;
+}
+.choices[data-type*=select-one] .choices__button:hover, .choices[data-type*=select-one] .choices__button:focus {
+  opacity: 1;
+}
+.choices[data-type*=select-one] .choices__button:focus {
+  box-shadow: 0 0 0 2px #005F75;
+}
+.choices[data-type*=select-one] .choices__item[data-placeholder] .choices__button {
+  display: none;
+}
+.choices[data-type*=select-one]::after {
+  content: "";
+  height: 0;
+  width: 0;
+  border-style: solid;
+  border-color: #333 transparent transparent transparent;
+  border-width: 5px;
+  position: absolute;
+  right: 11.5px;
+  top: 50%;
+  margin-top: -2.5px;
+  pointer-events: none;
+}
+.choices[data-type*=select-one].is-open::after {
+  border-color: transparent transparent #333;
+  margin-top: -7.5px;
+}
+.choices[data-type*=select-one][dir=rtl]::after {
+  left: 11.5px;
+  right: auto;
+}
+.choices[data-type*=select-one][dir=rtl] .choices__button {
+  right: auto;
+  left: 0;
+  margin-left: 25px;
+  margin-right: 0;
+}
+
+.choices[data-type*=select-multiple] .choices__inner,
+.choices[data-type*=text] .choices__inner {
+  cursor: text;
+}
+.choices[data-type*=select-multiple] .choices__button,
+.choices[data-type*=text] .choices__button {
+  position: relative;
+  display: inline-block;
+  margin-top: 0;
+  margin-right: -4px;
+  margin-bottom: 0;
+  margin-left: 8px;
+  padding-left: 16px;
+  border-left: 1px solid #003642;
+  background-image: url("data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMjEiIGhlaWdodD0iMjEiIHZpZXdCb3g9IjAgMCAyMSAyMSIgeG1sbnM9Imh0dHA6Ly93d3cudzMub3JnLzIwMDAvc3ZnIj48ZyBmaWxsPSIjRkZGIiBmaWxsLXJ1bGU9ImV2ZW5vZGQiPjxwYXRoIGQ9Ik0yLjU5Mi4wNDRsMTguMzY0IDE4LjM2NC0yLjU0OCAyLjU0OEwuMDQ0IDIuNTkyeiIvPjxwYXRoIGQ9Ik0wIDE4LjM2NEwxOC4zNjQgMGwyLjU0OCAyLjU0OEwyLjU0OCAyMC45MTJ6Ii8+PC9nPjwvc3ZnPg==");
+  background-size: 8px;
+  width: 8px;
+  line-height: 1;
+  opacity: 0.75;
+  border-radius: 0;
+}
+.choices[data-type*=select-multiple] .choices__button:hover, .choices[data-type*=select-multiple] .choices__button:focus,
+.choices[data-type*=text] .choices__button:hover,
+.choices[data-type*=text] .choices__button:focus {
+  opacity: 1;
+}
+
+.choices__inner {
+  display: inline-block;
+  vertical-align: top;
+  width: 100%;
+  background-color: #f9f9f9;
+  padding: 7.5px 7.5px 3.75px;
+  border: 1px solid #ddd;
+  border-radius: 6px;
+  overflow: hidden;
+}
+.is-focused .choices__inner, .is-open .choices__inner {
+  border-color: #b7b7b7;
+}
+.is-open .choices__inner {
+  border-radius: 2.5px 2.5px 0 0;
+}
+.is-flipped.is-open .choices__inner {
+  border-radius: 0 0 2.5px 2.5px;
+}
+
+.choices__list {
+  margin: 0;
+  padding-left: 0;
+  list-style: none;
+}
+.choices__list--single {
+  display: inline-block;
+  padding: 4px 16px 4px 4px;
+  width: 100%;
+}
+[dir=rtl] .choices__list--single {
+  padding-right: 4px;
+  padding-left: 16px;
+}
+.choices__list--single .choices__item {
+  width: 100%;
+}
+
+.choices__list--multiple {
+  display: inline;
+}
+.choices__list--multiple .choices__item {
+  display: inline-block;
+  vertical-align: middle;
+  border-radius: 20px;
+  padding: 4px 10px;
+  font-weight: 500;
+  margin-right: 3.75px;
+  margin-bottom: 3.75px;
+  background-color: $dark-lightened;
+  border: 1px solid $dark;
+  color: #fff;
+  word-break: break-all;
+  box-sizing: border-box;
+}
+.choices__list--multiple .choices__item[data-deletable] {
+  padding-right: 5px;
+}
+[dir=rtl] .choices__list--multiple .choices__item {
+  margin-right: 0;
+  margin-left: 3.75px;
+}
+.is-disabled .choices__list--multiple .choices__item {
+  background-color: #aaaaaa;
+  border: 1px solid #919191;
+}
+
+.choices__list--dropdown, .choices__list[aria-expanded] {
+  display: none;
+  z-index: 1;
+  position: absolute;
+  width: 100%;
+  background-color: #fff;
+  border: 1px solid #ddd;
+  top: 100%;
+  margin-top: -1px;
+  border-bottom-left-radius: 2.5px;
+  border-bottom-right-radius: 2.5px;
+  overflow: hidden;
+  word-break: break-all;
+}
+.is-active.choices__list--dropdown, .is-active.choices__list[aria-expanded] {
+  display: block;
+}
+.is-open .choices__list--dropdown, .is-open .choices__list[aria-expanded] {
+  border-color: #b7b7b7;
+}
+.is-flipped .choices__list--dropdown, .is-flipped .choices__list[aria-expanded] {
+  top: auto;
+  bottom: 100%;
+  margin-top: 0;
+  margin-bottom: -1px;
+  border-radius: 0.25rem 0.25rem 0 0;
+}
+.choices__list--dropdown .choices__list, .choices__list[aria-expanded] .choices__list {
+  position: relative;
+  max-height: 300px;
+  overflow: auto;
+  -webkit-overflow-scrolling: touch;
+  will-change: scroll-position;
+}
+.choices__list--dropdown .choices__item, .choices__list[aria-expanded] .choices__item {
+  position: relative;
+  padding: 10px;
+}
+[dir=rtl] .choices__list--dropdown .choices__item, [dir=rtl] .choices__list[aria-expanded] .choices__item {
+  text-align: right;
+}
+@media (min-width: 640px) {
+  .choices__list--dropdown .choices__item--selectable[data-select-text], .choices__list[aria-expanded] .choices__item--selectable[data-select-text] {
+    padding-right: 100px;
+  }
+  .choices__list--dropdown .choices__item--selectable[data-select-text]::after, .choices__list[aria-expanded] .choices__item--selectable[data-select-text]::after {
+    content: attr(data-select-text);
+    opacity: 0;
+    position: absolute;
+    right: 10px;
+    top: 50%;
+    transform: translateY(-50%);
+  }
+  [dir=rtl] .choices__list--dropdown .choices__item--selectable[data-select-text], [dir=rtl] .choices__list[aria-expanded] .choices__item--selectable[data-select-text] {
+    text-align: right;
+    padding-left: 100px;
+    padding-right: 10px;
+  }
+  [dir=rtl] .choices__list--dropdown .choices__item--selectable[data-select-text]::after, [dir=rtl] .choices__list[aria-expanded] .choices__item--selectable[data-select-text]::after {
+    right: auto;
+    left: 10px;
+  }
+}
+.choices__list--dropdown .choices__item--selectable.is-highlighted, .choices__list[aria-expanded] .choices__item--selectable.is-highlighted {
+  background-color: #f2f2f2;
+}
+.choices__list--dropdown .choices__item--selectable.is-highlighted::after, .choices__list[aria-expanded] .choices__item--selectable.is-highlighted::after {
+  opacity: 0.5;
+}
+
+.choices__item {
+  cursor: default;
+}
+
+.choices__item--selectable {
+  cursor: pointer;
+}
+
+.choices__item--disabled {
+  cursor: not-allowed;
+  -webkit-user-select: none;
+          user-select: none;
+  opacity: 0.5;
+}
+
+.choices__heading {
+  font-weight: 600;
+  padding: 10px;
+  border-bottom: 1px solid #f7f7f7;
+  color: gray;
+}
+
+.choices__button {
+  text-indent: -9999px;
+  appearance: none;
+  border: 0;
+  background-color: transparent;
+  background-repeat: no-repeat;
+  background-position: center;
+  cursor: pointer;
+}
+.choices__button:focus {
+  outline: none;
+}
+
+.choices__input {
+  display: inline-block;
+  vertical-align: baseline;
+  background-color: #f9f9f9;
+  margin-bottom: 5px;
+  border: 0;
+  border-radius: 0;
+  max-width: 100%;
+  padding: 4px 0 4px 2px;
+}
+.choices__input:focus {
+  outline: 0;
+}
+.choices__input::-webkit-search-decoration, .choices__input::-webkit-search-cancel-button, .choices__input::-webkit-search-results-button, .choices__input::-webkit-search-results-decoration {
+  display: none;
+}
+.choices__input::-ms-clear, .choices__input::-ms-reveal {
+  display: none;
+  width: 0;
+  height: 0;
+}
+[dir=rtl] .choices__input {
+  padding-right: 2px;
+  padding-left: 0;
+}
+
+.choices__placeholder {
+  opacity: 0.5;
+}
diff --git a/explorer/src/scss/styles.scss b/explorer/src/scss/styles.scss
index 9558b24f..39515c17 100644
--- a/explorer/src/scss/styles.scss
+++ b/explorer/src/scss/styles.scss
@@ -5,9 +5,8 @@
 $bootstrap-icons-font-dir: "../../../node_modules/bootstrap-icons/font/fonts";
 @import "~bootstrap-icons/font/bootstrap-icons.css";
 
-
 @import "explorer";
 @import "assistant";
+@import "choices";
 
 @import "pivot.css";
-@import "choices.js/public/assets/styles/choices.css";
diff --git a/explorer/src/scss/variables.scss b/explorer/src/scss/variables.scss
index 6397f6dc..e2cae45e 100644
--- a/explorer/src/scss/variables.scss
+++ b/explorer/src/scss/variables.scss
@@ -5,6 +5,7 @@ $blue: rgb(3, 68, 220);
 $green: rgb(127, 176, 105);
 $primary: $blue;
 $dark: rgb(1, 32, 63);
+$dark-lightened: rgba(1, 32, 63, 0.75);
 $secondary: $orange;
 $warning: $orange;
 $danger: $orange;
diff --git a/explorer/templates/assistant/table_description_form.html b/explorer/templates/assistant/table_description_form.html
index e6d42118..c3bacaad 100644
--- a/explorer/templates/assistant/table_description_form.html
+++ b/explorer/templates/assistant/table_description_form.html
@@ -2,38 +2,49 @@
 
 {% block sql_explorer_content %}
 <div class="container mt-5">
-  <h1>{% if form.instance.pk %}Edit{% else %}Create{% endif %} Table Description</h1>
-  {% if form.errors %}
-    {% for field in form %}
-        {% for error in field.errors %}
-            <div class="alert alert-danger">
-                <strong>{{ error|escape }}</strong>
+    <div class="row">
+        <div class="col-9">
+          <h1>{% if form.instance.pk %}Edit{% else %}Create{% endif %} Table Description</h1>
+          {% if form.errors %}
+            {% for field in form %}
+                {% for error in field.errors %}
+                    <div class="alert alert-danger">
+                        <strong>{{ error|escape }}</strong>
+                    </div>
+                {% endfor %}
+            {% endfor %}
+            {% for error in form.non_field_errors %}
+                <div class="alert alert-danger">
+                    <strong>{{ error|escape }}</strong>
+                </div>
+            {% endfor %}
+          {% endif %}
+          <form method="post" id="table-description-form">
+            {% csrf_token %}
+            <div>
+              <div class="mb-3 form-floating">
+                {{ form.database_connection }}
+                <label for="{{ form.database_connection.id_for_label }}" class="form-label">Connection</label>
+              </div>
+              <div class="mb-3 form-floating z-3">
+                <small>Table Name</small>
+                {{ form.table_name }}
+              </div>
+              <div class="mb-3 form-floating">
+                {{ form.description }}
+                <label for="{{ form.description.id_for_label }}" class="form-label">Description</label>
+              </div>
+            </div>
+            <button type="submit" class="btn btn-primary">{% if form.instance.pk %}Update{% else %}Save{% endif %} Annotation</button>
+            <a href="{% url 'table_description_list' %}" class="btn btn-secondary">Cancel</a>
+          </form>
+        </div>
+        <div class="col-md-3">
+            <div id="schema">
+                <iframe class="no-autofocus" src="" height="828px" frameBorder="0" id="schema_frame"></iframe>
             </div>
-        {% endfor %}
-    {% endfor %}
-    {% for error in form.non_field_errors %}
-        <div class="alert alert-danger">
-            <strong>{{ error|escape }}</strong>
         </div>
-    {% endfor %}
-{% endif %}
-  <form method="post" id="table-description-form">
-    {% csrf_token %}
-    <div>
-      <div class="mb-3 form-floating">
-        {{ form.database_connection }}
-        <label for="{{ form.database_connection.id_for_label }}" class="form-label">Connection</label>
-      </div>
-      <div class="mb-3 form-floating z-3">
-        {{ form.table_name }}
-      </div>
-      <div class="mb-3 form-floating">
-        {{ form.description }}
-        <label for="{{ form.description.id_for_label }}" class="form-label">Description</label>
-      </div>
     </div>
-    <button type="submit" class="btn btn-primary">{% if form.instance.pk %}Update{% else %}Save{% endif %} Annotation</button>
-    <a href="{% url 'table_description_list' %}" class="btn btn-secondary">Cancel</a>
-  </form>
 </div>
+
 {% endblock %}
diff --git a/explorer/templates/explorer/assistant.html b/explorer/templates/explorer/assistant.html
index 9088059d..afec2aa0 100644
--- a/explorer/templates/explorer/assistant.html
+++ b/explorer/templates/explorer/assistant.html
@@ -27,15 +27,21 @@
                         "Ask Assistant" to try and automatically fix the issue. The assistant is already aware of error messages & context.
                     </div>
                 </div>
-                <div id="additional_table_container" class="col-4">
-                    <div id="table-list" class="list"></div>
+                <div id="additional_table_container" class="col-3" style="width: 31% !important">
+                    <div id="table-list"></div>
                 </div>
             </div>
-            <div class="row">
-                <div class="col-12 text-end">
+            <div class="assistant-icons" style="">
+                <div>
                     <i class="bi-plus-circle" id="select_all_button" style="cursor: pointer;" data-bs-toggle="tooltip" data-bs-placement="top" data-bs-title="Select all"></i>
+                </div>
+                <div>
                     <i class="bi-trash" id="deselect_all_button" style="cursor: pointer;" data-bs-toggle="tooltip" data-bs-placement="top" data-bs-title="Remove all"></i>
+                </div>
+                <div>
                     <i class="bi-card-list" id="assistant_history" style="cursor: pointer;" data-bs-toggle="tooltip" data-bs-placement="top" data-bs-title="History"></i>
+                </div>
+                <div>
                     <i class="bi-question-circle" style="cursor: pointer;"
                     data-bs-toggle="tooltip" data-bs-placement="top"
                     data-bs-title="SQL Assistant builds a prompt with your query, your request, and the tables (schema, sample data, and annotations) referenced here."></i>
@@ -48,6 +54,7 @@
                     </div>
                 </div>
             </div>
+
         </div>
     </div>
 </div>
diff --git a/explorer/tests/test_assistant.py b/explorer/tests/test_assistant.py
index fae7df30..d1bcde27 100644
--- a/explorer/tests/test_assistant.py
+++ b/explorer/tests/test_assistant.py
@@ -5,10 +5,12 @@
 
 import json
 from django.test import TestCase
+from django.utils import timezone
 from django.urls import reverse
 from django.contrib.auth.models import User
 from django.db import OperationalError
 from explorer.ee.db_connections.utils import default_db_connection
+from explorer.ee.db_connections.models import DatabaseConnection
 from explorer.assistant.utils import (
     sample_rows_from_table,
     ROW_SAMPLE_SIZE,
@@ -18,6 +20,7 @@
 )
 
 from explorer.assistant.models import TableDescription
+from explorer.models import PromptLog
 
 
 def conn():
@@ -282,3 +285,134 @@ def test_get_relevant_annotations(self):
         self.assertEqual(relevant1.id, res1.id)
         res2 = get_relevant_annotation(default_db_connection(), "vegetables")
         self.assertEqual(relevant2.id, res2.id)
+
+
+class TestAssistantHistoryApiView(TestCase):
+
+    def setUp(self):
+        self.user = User.objects.create_superuser(
+            "admin", "admin@admin.com", "pwd"
+        )
+        self.client.login(username="admin", password="pwd")
+
+    def test_assistant_history_api_view(self):
+        # Create some PromptLogs
+        connection = default_db_connection()
+        PromptLog.objects.create(
+            run_by_user=self.user,
+            database_connection=connection,
+            user_request="Test request 1",
+            response="Test response 1",
+            run_at=timezone.now()
+        )
+        PromptLog.objects.create(
+            run_by_user=self.user,
+            database_connection=connection,
+            user_request="Test request 2",
+            response="Test response 2",
+            run_at=timezone.now()
+        )
+
+        # Make a POST request to the API
+        url = reverse("assistant_history")
+        data = {
+            "connection_id": connection.id
+        }
+        response = self.client.post(url, data=json.dumps(data), content_type="application/json")
+
+        # Check the response
+        self.assertEqual(response.status_code, 200)
+        response_data = json.loads(response.content)
+        self.assertIn("logs", response_data)
+        self.assertEqual(len(response_data["logs"]), 2)
+        self.assertEqual(response_data["logs"][0]["user_request"], "Test request 2")
+        self.assertEqual(response_data["logs"][0]["response"], "Test response 2")
+        self.assertEqual(response_data["logs"][1]["user_request"], "Test request 1")
+        self.assertEqual(response_data["logs"][1]["response"], "Test response 1")
+
+    def test_assistant_history_api_view_invalid_json(self):
+        url = reverse("assistant_history")
+        response = self.client.post(url, data="invalid json", content_type="application/json")
+        self.assertEqual(response.status_code, 400)
+        response_data = json.loads(response.content)
+        self.assertEqual(response_data["status"], "error")
+        self.assertEqual(response_data["message"], "Invalid JSON")
+
+    def test_assistant_history_api_view_no_logs(self):
+        connection = default_db_connection()
+        url = reverse("assistant_history")
+        data = {
+            "connection_id": connection.id
+        }
+        response = self.client.post(url, data=json.dumps(data), content_type="application/json")
+        self.assertEqual(response.status_code, 200)
+        response_data = json.loads(response.content)
+        self.assertIn("logs", response_data)
+        self.assertEqual(len(response_data["logs"]), 0)
+
+    def test_assistant_history_api_view_filtered_results(self):
+        # Create two users
+        user1 = self.user
+        user2 = User.objects.create_superuser(
+            "admin2", "admin2@admin.com", "pwd"
+        )
+
+        # Create two database connections
+        connection1 = default_db_connection()
+        connection2 = DatabaseConnection.objects.create(
+            alias="test_connection",
+            engine="django.db.backends.sqlite3",
+            name=":memory:"
+        )
+
+        # Create prompt logs for both users and connections
+        PromptLog.objects.create(
+            run_by_user=user1,
+            database_connection=connection1,
+            user_request="User1 Connection1 request",
+            response="User1 Connection1 response",
+            run_at=timezone.now()
+        )
+        PromptLog.objects.create(
+            run_by_user=user1,
+            database_connection=connection2,
+            user_request="User1 Connection2 request",
+            response="User1 Connection2 response",
+            run_at=timezone.now()
+        )
+        PromptLog.objects.create(
+            run_by_user=user2,
+            database_connection=connection1,
+            user_request="User2 Connection1 request",
+            response="User2 Connection1 response",
+            run_at=timezone.now()
+        )
+
+        # Make a POST request to the API as user1
+        url = reverse("assistant_history")
+        data = {
+            "connection_id": connection1.id
+        }
+        response = self.client.post(url, data=json.dumps(data), content_type="application/json")
+
+        # Check the response
+        self.assertEqual(response.status_code, 200)
+        response_data = json.loads(response.content)
+        self.assertIn("logs", response_data)
+        self.assertEqual(len(response_data["logs"]), 1)
+        self.assertEqual(response_data["logs"][0]["user_request"], "User1 Connection1 request")
+        self.assertEqual(response_data["logs"][0]["response"], "User1 Connection1 response")
+
+        # Now test with user2
+        self.client.logout()
+        self.client.login(username="admin2", password="pwd")
+
+        response = self.client.post(url, data=json.dumps(data), content_type="application/json")
+
+        # Check the response
+        self.assertEqual(response.status_code, 200)
+        response_data = json.loads(response.content)
+        self.assertIn("logs", response_data)
+        self.assertEqual(len(response_data["logs"]), 1)
+        self.assertEqual(response_data["logs"][0]["user_request"], "User2 Connection1 request")
+        self.assertEqual(response_data["logs"][0]["response"], "User2 Connection1 response")