Upload ChatIPC.cpp

ChatIPC.cpp

@@ -1138,7 +1138,7 @@ static void save_kb_binary(const KnowledgeBase &kb, const std::string &fname){
     }
 }
 
-static void load_kb_binary(KnowledgeBase &kb, const std::string &fname, int cli_dict_depth){
+static void load_kb_binary(KnowledgeBase &kb, const std::string &fname, int cli_def_depth){
     std::ifstream ifs(fname, std::ios::binary);
     if (!ifs) throw std::runtime_error("cannot open load file");
 
@@ -1242,8 +1242,8 @@ static void load_kb_binary(KnowledgeBase &kb, const std::string &fname, int cli_
         }
     }
 
-    if (cli_dict_depth != static_cast<int>(file_def_depth)){
-        kb.set_def_depth(cli_dict_depth);
+    if (cli_def_depth != static_cast<int>(file_def_depth)){
+        kb.set_def_depth(cli_def_depth);
 
         std::vector<StrPtr> targets;
         targets.reserve(ptrs.size() + kb.next.size() * 2);
@@ -1279,7 +1279,7 @@ static void load_kb_binary(KnowledgeBase &kb, const std::string &fname, int cli_
     }
 }
 
-static void print_usage(const char *p){
+static void print_commands(const char *p){
     std::cout << "CLI options: [--response-max-length N] [--save FILE] [--load-kb FILE] [--dictionary-depth D] [--learn f1 f2 ...] [--repeat-penalty P] [--help]\n";
     std::cout << "  --response-max-length N     Maximum number of tokens in a response.\n";
     std::cout << "  --save-kb FILE              Save the knowledge-base to a binary file.\n";
@@ -1296,18 +1296,18 @@ int main(int argc, char **argv){
     std::string savefile;
     std::string load_txt;
     std::string load_kb;
-    int dict_depth = 3;
+    int def_depth = 3;
     int n_gram_size = 3;
     double repeat_penalty = 0.7; // default λ
     std::vector<std::string> learn_files;
 
     for (int i=1;i<argc;++i){
         std::string a = argv[i];
-        if (a=="--help"){ print_usage(argv[0]); return 0; }
+        if (a=="--help"){ print_commands(argv[0]); return 0; }
         if (a=="--response-max-length" && i+1<argc){ response_maxlen = std::stoul(argv[++i]); continue; }
         if (a=="--save-kb" && i+1<argc){ savefile = argv[++i]; continue; }
         if (a=="--load-kb" && i+1<argc){ load_kb = argv[++i]; continue; }
-        if (a=="--dictionary-depth" && i+1<argc){ dict_depth = std::stoi(argv[++i]); continue; }
+        if (a=="--dictionary-depth" && i+1<argc){ def_depth = std::stoi(argv[++i]); continue; }
         if (a=="--n-gram" && i+1<argc){ n_gram_size = std::max(1, std::stoi(argv[++i])); continue; }
         if (a=="--repeat-penalty" && i+1<argc){ repeat_penalty = std::stod(argv[++i]); continue; }
         if (a=="--learn"){
@@ -1321,11 +1321,11 @@ int main(int argc, char **argv){
 
     global_dictionary_entries = parse_dictionary_json();
     build_def_tokens_cache();
-    kb.set_def_depth(dict_depth);
+    kb.set_def_depth(def_depth);
 
     if (!load_kb.empty()){
         try { std::cerr << "Loading knowledge base: " << load_kb << "\n";
-        load_kb_binary(kb, load_kb, dict_depth); std::cerr << "Loaded knowledge base: " << load_kb << "\n"; }
+        load_kb_binary(kb, load_kb, def_depth); std::cerr << "Loaded knowledge base: " << load_kb << "\n"; }
         catch (const std::exception &e){ std::cerr << "Error: " << e.what() << "\n"; }
     }