class for LOs updated with Typing

chains/learning_objectives_generator/learning_objectives_chain.py

@@ -7,6 +7,14 @@ from langchain_core.prompts.chat import ChatPromptTemplate
 class LearningObjectivesChain(BaseModel):
     """
     """
+    template_standardize: ChatPromptTemplate
+    llm_standardize: Any
+    template_gen_prompt_a: ChatPromptTemplate
+    template_gen_prompt_b: ChatPromptTemplate
+    default_llm_a: Any
+    default_llm_b: Any
+    template_sanitize: ChatPromptTemplate
+    llm_sanitize: Any
 
     class Config:
         arbitrary_types_allowed = True

chains/learning_objectives_generator/runner.py

@@ -34,7 +34,7 @@ async def run_learning_objectives_generator(
     llm_a = llms.get(model_choice_1, config["default_llm_a"])
     llm_b = llms.get(model_choice_2, config["default_llm_b"])
 
-    llm_sanitize=llms.get(config["llm_sanitize"])
+    llm_sanitize = llms.get(config["llm_sanitize"])
 
     # We will store the final sanitized results in an array of 4 strings
     # (2 prompts × 2 LLMs)