prompt tuning

Author: GreenWizard2015

core/CAIAssistant.py

@@ -28,7 +28,7 @@ def __init__(self, promptsFolder=None):
         HumanMessagePromptTemplate(
           prompt=PromptTemplate.from_file(
             os.path.join(promptsFolder, 'translate_deep.txt'),
-            input_variables=['UserInput', 'FastTranslation', 'Language', 'Flags']
+            input_variables=['UserInput', 'FastTranslation', 'Language', 'Flags', 'InputLanguage']
           )
         ),
       ]),
@@ -70,6 +70,7 @@ def translate(self, text, fastTranslation, language):
       'UserInput': text,
       'FastTranslation': translation['Translation'], # use shallow translation as reference
       'Language': language,
+      'InputLanguage': translation.get('Input language', 'unknown'),
       'Flags': ', '.join([k for k, v in flags.items() if v])
     })
     # extract final translation

prompts/translate_deep.txt

@@ -1,7 +1,7 @@
 $UserInput$: "{UserInput}"
 $Reference$: "{FastTranslation}"
 $Extra flags$: {Flags}
-$Task$: translate $UserInput$ to {Language}. $Reference$ may not be accurate. Refine step by step translation of $UserInput$ to {Language} while keeping the meaning as close as possible to $UserInput$. Consider $Extra flags$.
+$Task$: translate $UserInput$ to {Language}. $Reference$ may not be accurate. Refine step by step translation of $UserInput$ to {Language} while keeping the meaning as close as possible to $UserInput$. Consider $Extra flags$. Input text is in {InputLanguage}, consider their mentality and cultural differences. Suggest clarifications, that would be useful for translation.
 Response in exactly following format:
 ```
 @$UserInput$ in English: {{here is $UserInput$ in English}}
@@ -13,7 +13,7 @@ Response in exactly following format:
   1. {{quote the parts needing clarity, jargon, etc.}}
 @List of ambiguous pronoun usage (at least 3):
   1. {{quote the parts with pronoun ambiguities}}
-@List of suggested translations to {Language} with corresponding fixes (per issue):
-  1. {{fixes. Resolve all issues. Resolve all ambiguities by most likely meaning.}}
+@List of ALL resolved issues described above:
+  1. {{Resolve issues and ambiguities by most likely meaning. Provide translation to {Language}}}
 @Translation: {{here is $UserInput$ in {Language} with all fixes}}
 ```

prompts/translate_shallow.txt

@@ -3,6 +3,7 @@ $FastTranslation$: "{FastTranslation}"
 $Task$: "translate $UserInput$ to {Language}. $FastTranslation$ is a fast translation. It may not be accurate. Take into account that user may use jokes, jargon, etc. in $UserInput$."
 Response in exactly following format:
 ```
+@Input language: {{main language of $UserInput$, single language}}
 @Translation:
 {{Translation of $UserInput$ to {Language}}}
 @Contains jokes: {{Yes/No}}