Update app.py

app.py

@@ -171,7 +171,7 @@ unit_generator_prompt = """
             Character: Make a list of 5 opportunity that this unit can support the learners development towards the "portrait" of a graduate. Each opportunity should identify the trait and how it might be applied.
          """
 
-unit_generator = SingleInputAgent(prompt_template=unit_generator_prompt)
+unit_generator = Agent(prompt_template=unit_generator_prompt)
 
 lesson_idea_generator_prompt = """
             Below is a curriculum unit expressed using the understanding by design methodology ( delimited by the triple dollar signs).
@@ -182,7 +182,7 @@ lesson_idea_generator_prompt = """
             Each idea should include a title, a one sentence description of the activity, a one sentence description of how it addresses key concepts from the unit.
         """
 
-lesson_idea_generator = SingleInputAgent(prompt_template=lesson_idea_generator_prompt)
+lesson_idea_generator = Agent(prompt_template=lesson_idea_generator_prompt)
 
 lesson_builder_prompt = """ 
 The learning unit details are provided below (delimited by triple percent signs): 
@@ -201,7 +201,7 @@ Unit Map: Make a list of the referenced unit details. Reference relevant unit de
 
         """
 
-lesson_builder = SingleInputAgent(prompt_template=lesson_builder_prompt)
+lesson_builder = Agent(prompt_template=lesson_builder_prompt)
 
 def generate_unit(input):
     for next_token, content in unit_generator.stream(input):