workflows(expectation_revision): Refactor to support new prompt

ui/course_learn_suggest_expectations.py

@@ -136,7 +136,10 @@ def main():
                         if st.button("Reload suggestions", key=f"reload_{response['expectation']}"):
                             rework = ExpectationRevision()
                             feedback_inputs = {
-                                "expectation": response["expectation"], "check_question": response["check_question"], "request": feedback}
+                                "expectation": response["expectation"], "check_question": response["check_question"], "request": feedback,
+                                "course": course, "module": module,
+                                "tasks": tasks.split('\n'),
+                            }
                             suggestions = rework.kickoff(
                                 inputs=feedback_inputs)
                             st.write(suggestions)

workflows/courses/expectation_revision.py

@@ -3,9 +3,13 @@ from langchain_core.output_parsers import JsonOutputParser
 from langchain_openai import ChatOpenAI
 from pydantic import BaseModel, UUID4
 from .suggest_expectations import Expectation
+from typing import List
 import os
 
 class Inputs(BaseModel):
+    course: str
+    module: str
+    tasks: List[str]
     expectation: str
     check_question: str
     request: str
@@ -20,6 +24,9 @@ class ExpectationRevision:
         self.learning_outcome = inputs["expectation"]
         self.check_question = inputs["check_question"]
         self.request = inputs["request"]
+        self.course = inputs["course"]
+        self.module = inputs["module"]
+        self.tasks =  inputs["tasks"]
         llm_response = self._get_suggestion()
         return {
             "run_id": self.run_id,
@@ -44,6 +51,7 @@ class ExpectationRevision:
         with callbacks.collect_runs() as cb:
             response = chain.invoke({
                 "learning_outcome": self.learning_outcome, "check_question": self.check_question, "request": self.request,
+                "course": self.course, "module": self.module, "tasks": "* " + ("\n* ".join(self.tasks)),
                 "format_instructions": parser.get_format_instructions()
               })
             self.run_id = cb.traced_runs[0].id