adjust method signatures to take is_graded

ls1intum · Jan 15, 2025 · dbce6ae · dbce6ae
1 parent 63384d3
commit dbce6ae
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/modules/text/module_text_llm/module_text_llm/basic_approach/__init__.py b/modules/text/module_text_llm/module_text_llm/basic_approach/__init__.py
@@ -9,6 +9,6 @@ class BasicApproachConfig(ApproachConfig):
     type: Literal['basic'] = 'basic'
     generate_suggestions_prompt: GenerateSuggestionsPrompt = Field(default=GenerateSuggestionsPrompt())
 
-    async def generate_suggestions(self, exercise: Exercise, submission: Submission, config, debug: bool):
-        return await generate_suggestions(exercise,submission,config,debug)
+    async def generate_suggestions(self, exercise: Exercise, submission: Submission, config, debug: bool,is_graded: bool):
+        return await generate_suggestions(exercise,submission,config,debug,is_graded)
 
diff --git a/modules/text/module_text_llm/module_text_llm/chain_of_thought_approach/__init__.py b/modules/text/module_text_llm/module_text_llm/chain_of_thought_approach/__init__.py
@@ -12,6 +12,6 @@ class ChainOfThoughtConfig(ApproachConfig):
     thinking_prompt: ThinkingPrompt = Field(default=ThinkingPrompt())
     generate_suggestions_prompt: CoTGenerateSuggestionsPrompt = Field(default=CoTGenerateSuggestionsPrompt())
 
-    async def generate_suggestions(self, exercise: Exercise, submission: Submission, config, debug: bool):
-        return await generate_suggestions(exercise,submission,config,debug)
+    async def generate_suggestions(self, exercise: Exercise, submission: Submission, config, debug: bool,is_graded:bool):
+        return await generate_suggestions(exercise,submission,config,debug,is_graded)