removed text and added quotations to prompt

app.py

@@ -51,13 +51,13 @@ def gpt_respond(tab_name, message, chat_history, max_convo_length = 10):
     formatted_prompt = format_chat_prompt(message, chat_history, max_convo_length)
     print('Prompt + Context:')
     print(formatted_prompt)
-    bot_message = chat(system_prompt = f'''Generate the output only for the assistant. Output any <{tab_name}> in the following sentence one per line without any additional text.''',
+    bot_message = chat(system_prompt = f'''Generate the output only for the assistant. Output any <{tab_name}> in the following sentence one per line.''',
                        user_prompt = formatted_prompt)
     chat_history.append((message, bot_message))
     return "", chat_history
 
 def vicuna_respond(tab_name, message, chat_history):
-    formatted_prompt = f'''Generate the output only for the assistant. Output any {tab_name} in the following sentence one per line without any additional text: {message}'''
+    formatted_prompt = f'''Output any {tab_name} in the following sentence one per line: "{message}"'''
     print('Vicuna Ling Ents Fn - Prompt + Context:')
     print(formatted_prompt)
     input_ids = vicuna_tokenizer.encode(formatted_prompt, return_tensors="pt")
@@ -74,7 +74,7 @@ def vicuna_respond(tab_name, message, chat_history):
     return tab_name, "", chat_history
 
 def llama_respond(tab_name, message, chat_history):
-    formatted_prompt = f'''Generate the output only for the assistant. Output any {tab_name} in the following sentence one per line without any additional text: {message}'''
+    formatted_prompt = f'''Output any {tab_name} in the following sentence one per line: "{message}"'''
     # print('Llama - Prompt + Context:')
     # print(formatted_prompt)
     input_ids = llama_tokenizer.encode(formatted_prompt, return_tensors="pt")
@@ -96,18 +96,18 @@ def gpt_strategies_respond(have_key, strategy, task_name, task_ling_ent, message
     formatted_system_prompt = ""
     if (task_name == "POS Tagging"):
         if (strategy == "S1"):
-            formatted_system_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line without any additional text: {message}'''
+            formatted_system_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line: "{message}"'''
         elif (strategy == "S2"):
-            formatted_system_prompt = f'''POS tag the following sentence using Universal POS tag set without generating any additional text: {message}'''
+            formatted_system_prompt = f'''POS tag the following sentence using Universal POS tag set: "{message}"'''
         elif (strategy == "S3"):
-            formatted_system_prompt = f'''POS tag the following sentence using Universal POS tag set without generating any additional text: {message}'''
+            formatted_system_prompt = f'''POS tag the following sentence using Universal POS tag set: "{message}"'''
     elif (task_name == "Chunking"):
         if (strategy == "S1"):
-            formatted_system_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line without any additional text: {message}'''
+            formatted_system_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line: "{message}"'''
         elif (strategy == "S2"):
-            formatted_system_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags without outputing any additional text: {message}'''
+            formatted_system_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags: "{message}"'''
         elif (strategy == "S3"):
-            formatted_system_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags without outputing any additional text: {message}'''
+            formatted_system_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags: "{message}"'''
         
     formatted_prompt = format_chat_prompt(message, chat_history, max_convo_length)
     print('Prompt + Context:')
@@ -121,18 +121,18 @@ def vicuna_strategies_respond(strategy, task_name, task_ling_ent, message, chat_
     formatted_prompt = ""
     if (task_name == "POS Tagging"):
         if (strategy == "S1"):
-            formatted_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line without any additional text: {message}'''
+            formatted_prompt = f'''Output any {task_ling_ent} in the following sentence one per line: "{message}"'''
         elif (strategy == "S2"):
-            formatted_prompt = f'''POS tag the following sentence using Universal POS tag set without generating any additional text: {message}'''
+            formatted_prompt = f'''POS tag the following sentence using Universal POS tag set: "{message}"'''
         elif (strategy == "S3"):
-            formatted_prompt = f'''POS tag the following sentence using Universal POS tag set without generating any additional text: {message}'''
+            formatted_prompt = f'''POS tag the following sentence using Universal POS tag set: "{message}"'''
     elif (task_name == "Chunking"):
         if (strategy == "S1"):
-            formatted_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line without any additional text: {message}'''
+            formatted_prompt = f'''Output any {task_ling_ent} in the following sentence one per line: "{message}"'''
         elif (strategy == "S2"):
-            formatted_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags without outputing any additional text: {message}'''
+            formatted_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags: "{message}"'''
         elif (strategy == "S3"):
-            formatted_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags without outputing any additional text: {message}'''
+            formatted_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags: "{message}"'''
     
     print('Vicuna Strategy Fn - Prompt + Context:')
     print(formatted_prompt)
@@ -153,18 +153,18 @@ def llama_strategies_respond(strategy, task_name, task_ling_ent, message, chat_h
     formatted_prompt = ""
     if (task_name == "POS Tagging"):
         if (strategy == "S1"):
-            formatted_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line without any additional text: {message}'''
+            formatted_prompt = f'''Output any {task_ling_ent} in the following sentence one per line: "{message}"'''
         elif (strategy == "S2"):
-            formatted_prompt = f'''POS tag the following sentence using Universal POS tag set without generating any additional text: {message}'''
+            formatted_prompt = f'''POS tag the following sentence using Universal POS tag set: "{message}"'''
         elif (strategy == "S3"):
-            formatted_prompt = f'''POS tag the following sentence using Universal POS tag set without generating any additional text: {message}'''
+            formatted_prompt = f'''POS tag the following sentence using Universal POS tag set: "{message}"'''
     elif (task_name == "Chunking"):
         if (strategy == "S1"):
-            formatted_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line without any additional text: {message}'''
+            formatted_prompt = f'''Output any {task_ling_ent} in the following sentence one per line: "{message}"'''
         elif (strategy == "S2"):
-            formatted_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags without outputing any additional text: {message}'''
+            formatted_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags: "{message}"'''
         elif (strategy == "S3"):
-            formatted_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags without outputing any additional text: {message}'''
+            formatted_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags: "{message}"'''
     
     # print('Llama Strategies - Prompt + Context:')
     # print(formatted_prompt)