| @@ -211,7 +211,7 @@ class QuestionClassifierNode(LLMNode): | |||
| 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 = '' | |||
| @@ -35,7 +35,7 @@ QUESTION_CLASSIFIER_USER_PROMPT_2 = """ | |||
| QUESTION_CLASSIFIER_ASSISTANT_PROMPT_2 = """ | |||
| {"keywords": ["bad service", "slow", "food", "tip", "terrible", "waitresses"], | |||
| "categories": ["Experience""]}``` | |||
| "categories": ["Experience"]}``` | |||
| """ | |||
| QUESTION_CLASSIFIER_USER_PROMPT_3 = """ | |||