From 335cd8d947cc2cf4608885629dc9e63eaa061150 Mon Sep 17 00:00:00 2001
From: Mug <2797716+SagsMug@users.noreply.github.com>
Date: Sat, 6 May 2023 13:18:25 +0200
Subject: [PATCH] Rename postfix to suffix to match upstream

---
 examples/Chat.py                   | 1 +
 examples/common.py                 | 4 ++--
 examples/low_level_api_chat_cpp.py | 4 ++--
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/examples/Chat.py b/examples/Chat.py
index 9283fcb8a..c78347168 100644
--- a/examples/Chat.py
+++ b/examples/Chat.py
@@ -63,6 +63,7 @@ params = GptParams(
     interactive=True,
     antiprompt=[f"{USER_NAME}:"],
     input_prefix=" ",
+    input_suffix=f"{AI_NAME}:",
     prompt=prompt,
 )
 
diff --git a/examples/common.py b/examples/common.py
index 9a465db6e..75a952583 100644
--- a/examples/common.py
+++ b/examples/common.py
@@ -37,6 +37,7 @@ class GptParams:
     prompt: str = ""
     path_session: str = ""
     input_prefix: str = " "
+    input_suffix: str = ""
     antiprompt: List[str] = field(default_factory=list)
 
     lora_adapter: str = ""
@@ -64,7 +65,6 @@ class GptParams:
     # Set to "\nUser:" etc.
     # This is an alternative to input_prefix which always adds it, so it potentially duplicates "User:""
     fix_prefix: str = ""
-    output_postfix: str = ""
     input_echo: bool = True,
 
     # Default instructions for Alpaca
@@ -110,6 +110,7 @@ def gpt_params_parse(argv = None):
     parser.add_argument("-f", "--file", type=str, default=None, help="file containing initial prompt to load",dest="file")
     parser.add_argument("--session", type=str, default=None, help="file to cache model state in (may be large!)",dest="path_session")
     parser.add_argument("--in-prefix", type=str, default="", help="string to prefix user inputs with", dest="input_prefix")
+    parser.add_argument("--in-suffix", type=str, default="", help="append to input", dest="input_suffix")
     parser.add_argument(
         "-r",
         "--reverse-prompt",
@@ -158,7 +159,6 @@ def gpt_params_parse(argv = None):
 
     #Custom args
     parser.add_argument("--fix-prefix", type=str, default="", help="append to input when generated n_predict tokens", dest="fix_prefix")
-    parser.add_argument("--out-postfix", type=str, default="", help="append to input", dest="output_postfix")
     parser.add_argument("--input-noecho", action="store_false", help="dont output the input", dest="input_echo")
 
     parser.add_argument(
diff --git a/examples/low_level_api_chat_cpp.py b/examples/low_level_api_chat_cpp.py
index d024f0860..d3a7d466f 100644
--- a/examples/low_level_api_chat_cpp.py
+++ b/examples/low_level_api_chat_cpp.py
@@ -527,8 +527,8 @@ n_keep = {self.params.n_keep}
                 self.input(self.read_input())
             else:
                 print(self.params.input_prefix, end="")
-                self.input(f"{self.params.input_prefix}{self.read_input()}{self.params.output_postfix}")
-                print(self.params.output_postfix,end="")
+                self.input(f"{self.params.input_prefix}{self.read_input()}{self.params.input_suffix}")
+                print(self.params.input_suffix,end="")
             self.set_color(CONSOLE_COLOR_DEFAULT)
 
             try: