question classifier prompt optimization (#3479)

langgenius · Apr 15, 2024 · fd90d99 · fd90d99
1 parent 22994a6
commit fd90d99
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/api/core/workflow/nodes/question_classifier/question_classifier_node.py b/api/core/workflow/nodes/question_classifier/question_classifier_node.py
@@ -211,7 +211,7 @@ def _get_prompt_template(self, node_data: QuestionClassifierNodeData, query: str
         model_mode = ModelMode.value_of(node_data.model.mode)
         classes = node_data.classes
         class_names = [class_.name for class_ in classes]
-        class_names_str = ','.join(class_names)
+        class_names_str = ','.join(f'"{name}"' for name in class_names)
         instruction = node_data.instruction if node_data.instruction else ''
         input_text = query
         memory_str = ''

diff --git a/api/core/workflow/nodes/question_classifier/template_prompts.py b/api/core/workflow/nodes/question_classifier/template_prompts.py
@@ -35,7 +35,7 @@
 
 QUESTION_CLASSIFIER_ASSISTANT_PROMPT_2 = """
     {"keywords": ["bad service", "slow", "food", "tip", "terrible", "waitresses"],
-    "categories": ["Experience""]}```
+    "categories": ["Experience"]}```
 """
 
 QUESTION_CLASSIFIER_USER_PROMPT_3 = """