Dan rag update (#1549)

chillenberger · web-flow · commit e5e70237114a · 2024-07-09T12:48:25.000-06:00
diff --git a/pgml-dashboard/src/api/code_editor.rs b/pgml-dashboard/src/api/code_editor.rs
@@ -22,7 +22,7 @@ fn get_readonly_pool() -> PgPool {
                 .max_connections(1)
                 .idle_timeout(std::time::Duration::from_millis(60_000))
                 .max_lifetime(std::time::Duration::from_millis(60_000))
-                .connect_lazy(&std::env::var("CHATBOT_DATABASE_URL").expect("CHATBOT_DATABASE_URL not set"))
+                .connect_lazy(&std::env::var("EDITOR_DATABASE_URL").expect("EDITOR_DATABASE_URL not set"))
                 .expect("could not build lazy database connection")
         })
         .clone()
diff --git a/pgml-dashboard/src/components/layouts/marketing/sections/three_column/card/template.html b/pgml-dashboard/src/components/layouts/marketing/sections/three_column/card/template.html
@@ -1,6 +1,6 @@
 <div data-controller="layouts-marketing-section-three-column-card" class="flex-1">
   <div class="d-flex flex-column align-items-center p-3 p-lg-4">
-    <span class="material-symbols-outlined icon-<%- color %> image-top mb-4 text-white"><%- icon %></span>
+    <span class="material-symbols-outlined icon-<%- color %> image-top mb-4 text-white d-flex justify-content-center align-items-center"><%- icon %></span>
     <h6 class="text-center"><%+ title %></h6>
     <p class="text-center"><%+ paragraph %></p>
   </div>
diff --git a/pgml-dashboard/src/components/star/star.scss b/pgml-dashboard/src/components/star/star.scss
@@ -34,6 +34,7 @@ div[data-controller="star"] {
         flex-wrap: wrap;
         text-align: center;
         font-size: 0.8rem;
+        line-height: 1rem;
     }
 
     @-moz-keyframes spin { 100% { -moz-transform: rotate(360deg); } }
diff --git a/pgml-dashboard/static/css/modules.scss b/pgml-dashboard/static/css/modules.scss
@@ -81,3 +81,4 @@
 @import "../../src/components/tables/serverless_models/serverless_models.scss";
 @import "../../src/components/tables/serverless_pricing/serverless_pricing.scss";
 @import "../../src/components/tables/small/table/table.scss";
+@import "../../src/components/turbo/turbo_frame/turbo_frame.scss";
diff --git a/pgml-dashboard/static/js/utilities/demo.js b/pgml-dashboard/static/js/utilities/demo.js
@@ -28,7 +28,7 @@ export const generateSql = (task, model, userInput) => {
 );`;
   } else if (task === "embedded-query") {
     return `WITH embedded_query AS (
-  SELECT pgml.embed ('Alibaba-NLP/gte-base-en-v1.5', '${userInput}')::vector embedding
+  SELECT pgml.embed('mixedbread-ai/mxbai-embed-large-v1', 'What is Postgres?', '{"prompt": "Represent this sentence for searching relevant passages: "}'::JSONB)::vector embedding
 ),
 context_query AS (
   SELECT chunks.chunk FROM chunks
@@ -42,7 +42,7 @@ SELECT
       "task": "conversational",
       "model": "meta-llama/Meta-Llama-3-8B-Instruct"
     }'::jsonb,
-    inputs => ARRAY['{"role": "system", "content": "You are a friendly and helpful chatbot."}'::jsonb, replace('{"role": "user", "content": "Given the context answer the following question. ${userInput}? Context:\n\n{CONTEXT}"}', '{CONTEXT}', chunk)::jsonb],
+    inputs => ARRAY['{"role": "system", "content": "You are a friendly and helpful chatbot."}'::jsonb, jsonb_build_object('role', 'user', 'content', replace('Given the context answer the following question. ${userInput}? Context:\n{CONTEXT}', '{CONTEXT}', chunk))],
     args => '{
       "max_new_tokens": 100
     }'::jsonb

Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,7 @@ div[data-controller="star"] {`
`34`	`34`	`flex-wrap: wrap;`
`35`	`35`	`text-align: center;`
`36`	`36`	`font-size: 0.8rem;`
	`37`	`+ line-height: 1rem;`
`37`	`38`	`}`
`38`	`39`
`39`	`40`	`@-moz-keyframes spin { 100% { -moz-transform: rotate(360deg); } }`