Upload ChatIPC.cpp

ChatIPC.cpp

@@ -828,11 +828,11 @@ static std::string best_candidate_by_similarity(
 
 static std::vector<std::string> construct_response(KnowledgeBase &kb,
                                                    const std::vector<std::string> &prompt_toks,
-                                                   size_t maxlen,
+                                                   size_t response_maxlen,
                                                    double repeat_penalty)
 {
     std::vector<std::string> resp;
-    if (prompt_toks.empty() || maxlen == 0) return resp;
+    if (prompt_toks.empty() || response_maxlen == 0) return resp;
 
     auto prompt_ptrs = intern_tokens(kb, prompt_toks);
     std::vector<StrPtr> resp_ptrs;
@@ -846,7 +846,7 @@ static std::vector<std::string> construct_response(KnowledgeBase &kb,
 
     std::string last_printed;
 
-    for (size_t step = 0; step < maxlen; ++step){
+    for (size_t step = 0; step < response_maxlen; ++step){
         NextSet candidates;
         bool found = false;
         std::string context_tok;
@@ -1156,18 +1156,18 @@ static void load_kb_binary(KnowledgeBase &kb, const std::string &fname, int cli_
 }
 
 static void print_usage(const char *p){
-    std::cout << "Usage: " << p << " [--maxlen N] [--save FILE] [--load-kb FILE] [--dictionary-depth D] [--learn f1 f2 ...] [--repeat-penalty P] [--help]\n";
-    std::cout << "  --maxlen N                Maximum number of tokens constructed in a response.\n";
-    std::cout << "  --save FILE               Save the knowledge-base and dictionary expansions to a binary file.\n";
-    std::cout << "  --load-kb FILE            Load a previously saved knowledge-base (and dictionary expansions) from a binary file.\n";
-    std::cout << "  --dictionary-depth D      Depth of dictionary-definition expansion used during learning.\n";
-    std::cout << "  --learn f1 f2 ...         Learn from one or more text files to update the knowledge base.\n";
-    std::cout << "  --repeat-penalty P        Penalize repeated tokens during response generation (higher values discourage repetition).\n";
-    std::cout << "  --help                    Show command-line interface options for ChatIPC usage.\n";
+    std::cout << "Usage: " << p << " [--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 constructed in a response.\n";
+    std::cout << "  --save FILE                 Save the knowledge-base and dictionary expansions to a binary file.\n";
+    std::cout << "  --load-kb FILE              Load a previously saved knowledge-base (and dictionary expansions) from a binary file.\n";
+    std::cout << "  --dictionary-depth D        Depth of dictionary-definition expansion used during learning.\n";
+    std::cout << "  --learn f1 f2 ...           Learn from one or more text files to update the knowledge base.\n";
+    std::cout << "  --repeat-penalty P          Penalize repeated tokens during response generation (higher values discourage repetition).\n";
+    std::cout << "  --help                      Show command-line interface options for ChatIPC usage.\n";
 }
 
 int main(int argc, char **argv){
-    size_t maxlen = 100;
+    size_t response_maxlen = 100;
     std::string savefile;
     std::string load_txt;
     std::string load_kb;
@@ -1178,7 +1178,7 @@ int main(int argc, char **argv){
     for (int i=1;i<argc;++i){
         std::string a = argv[i];
         if (a=="--help"){ print_usage(argv[0]); return 0; }
-        if (a=="--maxlen" && i+1<argc){ maxlen = std::stoul(argv[++i]); continue; }
+        if (a=="--response-max-length" && i+1<argc){ response_maxlen = std::stoul(argv[++i]); continue; }
         if (a=="--save" && 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; }
@@ -1219,7 +1219,7 @@ int main(int argc, char **argv){
         if (line.empty()){ std::cout << "\n"; continue; }
         auto prompt_toks = tokenize_whitespace(line);
         for (size_t i=1;i<prompt_toks.size();++i) kb.add_pair(prompt_toks[i-1], prompt_toks[i]);
-        auto resp = construct_response(kb, prompt_toks, maxlen, repeat_penalty);
+        auto resp = construct_response(kb, prompt_toks, response_maxlen, repeat_penalty);
         std::cout << "\n";
         if (!resp.empty()){for (size_t i=1;i<resp.size();++i) kb.add_pair(resp[i-1], resp[i]);}
         if (!savefile.empty()){