Upload ChatIPC.cpp

ChatIPC.cpp

@@ -1358,7 +1358,8 @@ static std::vector<std::string> construct_response(KnowledgeBase &kb,
                                                    const std::vector<std::string> &prompt_toks,
                                                    size_t response_maxlen,
                                                    double repeat_penalty,
-                                                   int n_gram_size)
+                                                   int n_gram_size,
+                                                   bool allow_path_loops)
 {
     std::vector<std::string> resp;
     if (prompt_toks.empty() || response_maxlen == 0) return resp;
@@ -1414,7 +1415,7 @@ static std::vector<std::string> construct_response(KnowledgeBase &kb,
         if (candidates.size() == 1){
             std::string only = *candidates[0];
             std::string only_key = normalize_dictionary_key(only);
-            if (recent_counts[only_key.empty() ? only : only_key] > 0) break;
+            if (!allow_path_loops && recent_counts[only_key.empty() ? only : only_key] > 0) break;
 
             resp.push_back(only);
             StrPtr ptr = kb.interner.intern(only);
@@ -1772,15 +1773,16 @@ 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 token(s) in a response.\n";
+    std::cout << "  --response-max-length N     Set maximum number of token(s) in a response. Default number is 500.\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 << "  --dictionary-depth D        Set depth of dictionary-definition expansion used during learning. Default depth is 3.\n";
+    std::cout << "  --n-gram-max-size N         Set maximum size of the n-gram where N is the size. Default size is 3.\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 << "  --repeat-penalty P          Set penalty for repeated tokens in response. Default penalty is 0.7 (higher values reduce repetition).\n";
+    std::cout << "  --allow-path-loops          Allow infinite loops if the knowledge-base graph has a circular single-path reference.\n";
     std::cout << "  --activate-agi              Activate the Artificial General Intelligence (AGI) features.\n";
     std::cout << "  --help                      Show " << p << " command-line interface options.\n";
 }
@@ -1795,6 +1797,7 @@ int main(int argc, char **argv){
     double repeat_penalty = 0.7; // default λ
     std::vector<std::string> learn_files;
     bool learn_chat = true;
+    bool allow_path_loops = false;
 
     for (int i=1;i<argc;++i){
         std::string a = argv[i];
@@ -1806,15 +1809,13 @@ int main(int argc, char **argv){
         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=="--learn"){ while(i+1<argc && argv[i+1][0] != '-') learn_files.push_back(argv[++i]); continue; }
         if (a=="--repeat-penalty" && i+1<argc){ repeat_penalty = std::stod(argv[++i]); continue; }
+        if (a=="--allow-path-loops") { allow_path_loops = true; continue; }
         if (a=="--activate-agi"){
                 std::cerr << "AGI subscription fee is $1,000 CAD per month. To pay the fee, contact Caleb Nwokocha via email cnwokocha@proton.me\n";
                 return 0;
         }
-        if (a=="--learn"){
-            while(i+1<argc && argv[i+1][0] != '-') learn_files.push_back(argv[++i]);
-            continue;
-        }
 
         std::cerr << "Error: Unknown or incomplete command-line option '" << a << "'\n";
         print_commands(argv[0]);
@@ -1844,7 +1845,7 @@ int main(int argc, char **argv){
         if (line.empty()){ std::cout << "\n"; continue; }
         auto prompt_toks = tokenize_whitespace(line);
         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);
+        auto resp = construct_response(kb, prompt_toks, response_maxlen, repeat_penalty, n_gram_size, allow_path_loops);
         std::cout << "\n";
         if (!resp.empty() && learn_chat){learn_tokens_ngram(kb, resp, n_gram_size);}
         if (!savefile.empty()){