调整问题生成的提示词

backend/unit_test/test_question.py

@@ -79,7 +79,7 @@ class FileLoader:
         parser = {
         ".md" : ChunkMarkdownReader(),
         }
-        return parser
+        return None
 
 # 定义中文提示词和Python代码中提示词名称的映射
 prompt_mapping = {
@@ -168,13 +168,13 @@ def main(documents:List[Document], table_names:List[str],num_questions_per_promp
 
 # 获取命令行参数
 if __name__ == "__main__":
-    if len(sys.argv) != 2:
-        raise ValueError("Usage: python script.py <table_names_input> <prompt_categories_input> <num_questions_per_prompt>")
-    table_names_input = sys.argv[0]
-    num_questions_per_prompt = int(sys.argv[1])
+    # if len(sys.argv) != 2:
+    #     raise ValueError("Usage: python script.py <table_names_input> <prompt_categories_input> <num_questions_per_prompt>")
+    # table_names_input = sys.argv[0]
+    # num_questions_per_prompt = int(sys.argv[1])
 
-    # table_names_input = '[总算表]'
-    # num_questions_per_prompt = 2
+    table_names_input = '[总算表]'
+    num_questions_per_prompt = 10
 
     que_Dir = os.path.join(os.getcwd(),f'unit_test\\Quetions')     
     loader = FileLoader()