feature: use different prompt for ecommerce suggested queries

Author: vid277

clients/search-component/src/utils/hooks/useSuggestedQuestions.ts

@@ -22,6 +22,7 @@ export const useSuggestedQuestions = () => {
         : currentGroup?.tracking_id,
       query,
       props,
+      isEcommerce: props.type === "ecommerce",
     });
     setSuggestedQuestions(queries.queries);
     setIsLoadingSuggestedQueries(false);

clients/search-component/src/utils/trieve.ts

@@ -431,6 +431,7 @@ export const getSuggestedQuestions = async ({
   prevUserMessages,
   is_followup,
   chunks,
+  isEcommerce,
 }: {
   trieve: TrieveSDK;
   abortController?: AbortController;
@@ -441,6 +442,7 @@ export const getSuggestedQuestions = async ({
   prevUserMessages?: string[];
   props?: ModalProps;
   chunks?: Chunk[] | null;
+  isEcommerce?: boolean;
 }) => {
   let context: string;
   if (groupTrackingId && modalProps?.cleanGroupName) {
@@ -485,6 +487,7 @@ export const getSuggestedQuestions = async ({
                 ],
               },
             }),
+          is_ecommerce: isEcommerce ?? false,
         },
         abortController?.signal,
       ),

clients/ts-sdk/openapi.json

@@ -22471,8 +22471,14 @@
             ],
             "nullable": true
           },
+          "is_ecommerce": {
+            "type": "boolean",
+            "description": "Whether of not the suggested queries are being generated for ecomm. If true, the suggested queries will be generated for ecomm. If false, the suggested queries will be generated for non-ecomm.",
+            "nullable": true
+          },
           "is_followup": {
             "type": "boolean",
+            "description": "Whether or not the suggested queries are being generated for a followup question. If true, the suggested queries will be generated for a followup question. If false, the suggested queries will be generated for a new question.",
             "nullable": true
           },
           "query": {

clients/ts-sdk/src/types.gen.ts

@@ -4524,6 +4524,13 @@ export type SuggestedQueriesReqPayload = {
      */
     context?: (string) | null;
     filters?: ((ChunkFilter) | null);
+    /**
+     * Whether of not the suggested queries are being generated for ecomm. If true, the suggested queries will be generated for ecomm. If false, the suggested queries will be generated for non-ecomm.
+     */
+    is_ecommerce?: (boolean) | null;
+    /**
+     * Whether or not the suggested queries are being generated for a followup question. If true, the suggested queries will be generated for a followup question. If false, the suggested queries will be generated for a new question.
+     */
     is_followup?: (boolean) | null;
     /**
      * The query to base the generated suggested queries off of using RAG. A hybrid search for 10 chunks from your dataset using this query will be performed and the context of the chunks will be used to generate the suggested queries.

server/src/handlers/message_handler.rs

@@ -829,8 +829,10 @@ pub struct SuggestedQueriesReqPayload {
     pub context: Option<String>,
     /// Filters is a JSON object which can be used to filter chunks. This is useful for when you want to filter chunks by arbitrary metadata. Unlike with tag filtering, there is a performance hit for filtering on metadata.
     pub filters: Option<ChunkFilter>,
-
+    /// Whether or not the suggested queries are being generated for a followup question. If true, the suggested queries will be generated for a followup question. If false, the suggested queries will be generated for a new question.
     pub is_followup: Option<bool>,
+    /// Whether of not the suggested queries are being generated for ecommerce.
+    pub is_ecommerce: Option<bool>,
 }
 
 #[derive(Deserialize, Serialize, Debug, ToSchema)]

server/src/operators/message_operator.rs

@@ -1921,10 +1921,17 @@ pub async fn suggested_new_queries(
     };
     let context_sentence = match payload.context.clone() {
         Some(context) => {
-            format!(
-                "Suggest varied {query_style} queries with the following context in mind: {}. Generate one question that requests for a comparison of two products, explicitly including these questions in the query through phrases such as 'compare, versus, different, etc...'. Another question should contain one product title, explicitly include the name of the product in the query and ask questions related to this product within the provided context. For another question, generate a question that usses the format of find me and then a broader category name that makes sense within the provided context. Ensure all queries are short and to the point.",
-                context
-            )
+            if payload.is_ecommerce.unwrap_or(false) {
+                format!(
+                    "Suggest varied {query_style} queries with the following context in mind: {}. Generate one question that requests for a comparison of two products, explicitly including these questions in the query through phrases such as 'compare, versus, different, etc...'. Another question should contain one product title, explicitly include the name of the product in the query and ask questions related to this product within the provided context. For another question, generate a question that usses the format of find me and then a broader category name that makes sense within the provided context. Ensure all queries are short and to the point.",
+                    context
+                )
+            } else {
+                format!(
+                    "Suggest varied {query_style} queries with the following context in mind: {}. Generate queries that are short, specific, and directly related to the provided context.",
+                    context
+                )
+            }
         }
         None => "".to_string(),
     };