diff --git a/rag_tutorials/wfgy_rag_failure_clinic/wfgy_rag_failure_clinic.py b/rag_tutorials/wfgy_rag_failure_clinic/wfgy_rag_failure_clinic.py index 4d6a061..7691aa9 100644 --- a/rag_tutorials/wfgy_rag_failure_clinic/wfgy_rag_failure_clinic.py +++ b/rag_tutorials/wfgy_rag_failure_clinic/wfgy_rag_failure_clinic.py @@ -147,11 +147,25 @@ def choose_bug_description() -> str: print() if choice == "1": - return EXAMPLE_1 + bug = EXAMPLE_1 + print("You selected Example 1. Full bug description:\n") + print(bug) + print() + return bug + if choice == "2": - return EXAMPLE_2 + bug = EXAMPLE_2 + print("You selected Example 2. Full bug description:\n") + print(bug) + print() + return bug + if choice == "3": - return EXAMPLE_3 + bug = EXAMPLE_3 + print("You selected Example 3. Full bug description:\n") + print(bug) + print() + return bug print("Paste your bug description. End with an empty line.") lines = [] @@ -163,9 +177,15 @@ def choose_bug_description() -> str: if not line.strip(): break lines.append(line) + user_bug = "\n".join(lines).strip() if not user_bug: print("No bug description detected, aborting this round.\n") + return "" + + print("\nYou pasted the following bug description:\n") + print(user_bug) + print() return user_bug