feature: add parameters to chat with single product

devflowinc · Jan 10, 2025 · 5ab7556 · 5ab7556
1 parent bdb54e7
commit 5ab7556
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 4 deletions.
diff --git a/clients/search-component/src/utils/hooks/chat-context.tsx b/clients/search-component/src/utils/hooks/chat-context.tsx
@@ -88,6 +88,7 @@ function ChatProvider({ children }: { children: React.ReactNode }) {
 
   const clearConversation = () => {
     setCurrentTopic("");
+    setCurrentGroup(null);
     setMessages([]);
   };
 
@@ -201,7 +202,7 @@ function ChatProvider({ children }: { children: React.ReactNode }) {
             llm_options: {
               completion_first: false,
             },
-            page_size: props.searchOptions?.page_size ?? 5,
+            page_size: props.searchOptions?.page_size ?? 8,
             score_threshold: props.searchOptions?.score_threshold || null,
             use_group_search: props.useGroupSearch,
             filters:
@@ -258,6 +259,25 @@ function ChatProvider({ children }: { children: React.ReactNode }) {
   const askQuestion = async (question?: string, group?: ChunkGroup) => {
     setIsDoneReading(false);
 
+    if (props.groupTrackingId) {
+
+      const fetchedGroup = await trieveSDK.getGroupByTrackingId({
+        trackingId: props.groupTrackingId
+      });
+      if (fetchedGroup) {
+        group = {
+          created_at: fetchedGroup.created_at,
+          updated_at: fetchedGroup.updated_at,
+          dataset_id: fetchedGroup.dataset_id,
+          description: fetchedGroup.description,
+          id: fetchedGroup.id,
+          metadata: fetchedGroup.metadata,
+          name: props.cleanGroupName ? props.cleanGroupName : fetchedGroup.name,
+          tag_set: fetchedGroup.tag_set,
+        } as ChunkGroup;
+      }
+    }
+
     if (!currentGroup && group) {
       chatWithGroup(group);
       setCurrentGroup(group);

diff --git a/clients/search-component/src/utils/hooks/modal-context.tsx b/clients/search-component/src/utils/hooks/modal-context.tsx
@@ -110,6 +110,8 @@ export type ModalProps = {
   showFloatingInput?: boolean;
   inlineHeader?: string;
   suggestedQueriesContext: string;
+  groupTrackingId?: string;
+  cleanGroupName?: string;
 };
 
 const defaultProps = {
@@ -153,7 +155,9 @@ const defaultProps = {
   showFloatingInput: false,
   inline: false,
   inlineHeader: "AI Assistant by Trieve",
-  suggestedQueriesContext: "You are an ecommerce sales assistant to help people buy items."
+  suggestedQueriesContext: "You are an ecommerce sales assistant to help people buy items.",
+  groupTrackingId: undefined,
+  cleanGroupName: undefined,
 };
 
 const ModalContext = createContext<{

diff --git a/server/src/handlers/message_handler.rs b/server/src/handlers/message_handler.rs
@@ -952,7 +952,7 @@ pub async fn get_suggested_queries(
         context_sentence,
         number_of_suggestions_to_create,
         query_style,
-		number_of_suggestions_to_create,
+	number_of_suggestions_to_create,
         query_style,
         query_style,
         query_style,

diff --git a/server/src/public/page.html b/server/src/public/page.html
@@ -442,7 +442,8 @@
 
         if (params.inline) {
           const inlineRoot = document.querySelector("#inline-root");
-
+          params.groupTrackingId = params.singleProductOptions.groupTrackingId;
+          params.cleanGroupName = params.singleProductOptions.productName;
           renderToDiv(inlineRoot, {
             ...params,
           })