a few error fixes

multi_agent_debate.py

@@ -477,7 +477,7 @@ class MultiAgentDebate:
             if initialization:
                 print("ROUND %s: (This is the initialization round where agents are assigned initial stance as their beliefs.)\n"%str(round_counter+1))
                 for n in range(len(agents_responses)):
-                    print("Agent %s: "%str(n+1) + agents_responses[n] + "\n")
+                    writer("Agent %s: "%str(n+1) + agents_responses[n] + "\n", "This is my initial belief.")
                 print("----------------------------------------------------")
                 round_counter += 1
             print("ROUND %s:\n"%str(round_counter+1))
@@ -516,7 +516,8 @@ class MultiAgentDebate:
                 while label!="Unknown" and label_val != label_list[n][0] and rep_ctr != 1:
                     llm_response, _ = self.make_openai_api_call(prompt, model_name, 1)
                     argument, label = self.parse_output_w_chat_label(llm_response)
-                    strlabel = "Support" if label == 1 else "Refute"
+                    print(f">>>>>>>\n\t{label}\n")
+                    strlabel = "Support" if label == "1" else "Refute"
                     writer("Agent %s's Assessment:\n"%str(n+1) + '%s. \n'%strlabel, 'Explanation: %s'%argument + "\n")
                     print("***************")
                     rep_ctr += 1
@@ -609,7 +610,7 @@ class MultiAgentDebate:
                     # print(llm_response)
                     # print("***************")
                     argument, label = self.parse_output_w_chat_label(llm_response)
-                    strlabel = "Support" if label == 1 else "Refute"
+                    strlabel = "Support" if label == "1" else "Refute"
                     writer("Agent %s's Assessment: \n"%str(n+1) + '%s. \n'%strlabel, 'Explanation: %s'%argument + "\n")
                     print("***************")
                     if label != "Unknown":