Skip to content

Commit 79f15cc

Browse files
committed
useful print
1 parent 6d85c83 commit 79f15cc

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/agents/utils/synthetic_conversation_generation.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ def generate_synthetic_conversations(raw_text: str, num_turns: int, student_agen
105105
if __name__ == "__main__":
106106
num_turns = 6
107107
# Can be "informational", "socratic", "google_learnlm"
108-
tutor_agent_types = ["google_learnlm"]
108+
tutor_agent_types = ["socratic"]
109109
# Can be "base", "curious", "contradicting", "reliant", "confused", "unrelated"
110110
student_agent_types = ["base", "curious", "contradicting", "reliant", "confused", "unrelated"]
111111

@@ -127,6 +127,7 @@ def generate_synthetic_conversations(raw_text: str, num_turns: int, student_agen
127127

128128
for student_agent_type in student_agent_types:
129129
for question in questions:
130+
print(f"Generating synthetic conversation for {question} with tutor: {tutor_agent_type} and student: {student_agent_type}")
130131
with open(question, "r") as file:
131132
raw_text = file.read()
132133

0 commit comments

Comments
 (0)