Upload ChatIPC.cpp

ChatIPC.cpp

@@ -1772,13 +1772,14 @@ static void print_kb_info(const std::string &fname) {
 
 static void print_commands(const char *p){
     std::cout << p << " command-line interface options:\n";
-    std::cout << "  --response-max-length N     Set maximum number of tokens in a response.\n";
-    std::cout << "  --save-kb FILE              Save the knowledge-base to a binary file.\n";
-    std::cout << "  --load-kb FILE              Load a previously saved knowledge-base from a binary file.\n";
-    std::cout << "  --info-kb FILE              Show the dictionary-depth and n-gram max size of a saved knowledge-base.\n";
+    std::cout << "  --response-max-length N     Set maximum number of token(s) in a response.\n";
+    std::cout << "  --save-kb FILE              Save the knowledge-base graph to a binary file.\n";
+    std::cout << "  --load-kb FILE              Load a previously saved knowledge-base graph from a binary file.\n";
+    std::cout << "  --info-kb FILE              Show the dictionary-depth and n-gram max size of a knowledge-base file.\n";
+    std::cout << "  --static-kb                 Disable learning prompt(s) and response(s) during interactive session.\n";
     std::cout << "  --dictionary-depth D        Set depth of dictionary-definition expansion used during learning.\n";
     std::cout << "  --n-gram-max-size N         Set maximum size of the n-gram where N is the size.\n";
-    std::cout << "  --learn f1 f2 ...           Learn from one or more text files to update the knowledge-base.\n";
+    std::cout << "  --learn f1 f2 ...           Learn from text file(s) to update the knowledge-base graph.\n";
     std::cout << "  --repeat-penalty P          Set penalty for repeated tokens when constructing response (higher values reduce repetition).\n";
     std::cout << "  --activate-agi              Activate the Artificial General Intelligence (AGI) features.\n";
     std::cout << "  --help                      Show " << p << " command-line interface options.\n";
@@ -1793,6 +1794,7 @@ int main(int argc, char **argv){
     int n_gram_size = 3;
     double repeat_penalty = 0.7; // default λ
     std::vector<std::string> learn_files;
+    bool learn_chat = true;
 
     for (int i=1;i<argc;++i){
         std::string a = argv[i];
@@ -1801,6 +1803,7 @@ int main(int argc, char **argv){
         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=="--info-kb" && i+1<argc){ print_kb_info(argv[++i]); return 0; }
+        if (a=="--static-kb"){ learn_chat = false; continue; }
         if (a=="--dictionary-depth" && i+1<argc){ def_depth = std::stoi(argv[++i]); continue; }
         if (a=="--n-gram-max-size" && 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; }
@@ -1812,7 +1815,10 @@ int main(int argc, char **argv){
             while(i+1<argc && argv[i+1][0] != '-') learn_files.push_back(argv[++i]);
             continue;
         }
-        learn_files.push_back(a);
+
+        std::cerr << "Error: Unknown or incomplete command-line option '" << a << "'\n";
+        print_commands(argv[0]);
+        return 1;
     }
 
     KnowledgeBase kb;
@@ -1837,10 +1843,10 @@ int main(int argc, char **argv){
     while (std::cout << "> " , std::getline(std::cin, line)){
         if (line.empty()){ std::cout << "\n"; continue; }
         auto prompt_toks = tokenize_whitespace(line);
-        learn_tokens_ngram(kb, prompt_toks, n_gram_size);
+        if (learn_chat) {learn_tokens_ngram(kb, prompt_toks, n_gram_size);}
         auto resp = construct_response(kb, prompt_toks, response_maxlen, repeat_penalty, n_gram_size);
         std::cout << "\n";
-        if (!resp.empty()){learn_tokens_ngram(kb, resp, n_gram_size);}
+        if (!resp.empty() && learn_chat){learn_tokens_ngram(kb, resp, n_gram_size);}
         if (!savefile.empty()){
             try { std::cerr << "Saving knowledge base: " << savefile << "\n";
             save_kb_binary(kb, savefile); std::cerr << "Saved knowledge base: " << savefile << "\n"; }