Fix competency extraction did not save Enum

ls1intum · Oct 12, 2024 · 6bd4b33 · 6bd4b33
1 parent 4c92900
commit 6bd4b33
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/app/pipeline/competency_extraction_pipeline.py b/app/pipeline/competency_extraction_pipeline.py
@@ -6,6 +6,7 @@
     ChatPromptTemplate,
 )
 
+from app.common.PipelineEnum import PipelineEnum
 from app.common.pyris_message import PyrisMessage, IrisMessageRole
 from app.domain import (
     CompetencyExtractionPipelineExecutionDTO,
@@ -76,7 +77,9 @@ def __call__(
         response = self.request_handler.chat(
             [prompt], CompletionArguments(temperature=0.4)
         )
-        self.tokens.append(response.token_usage)
+        self._append_tokens(
+            response.token_usage, PipelineEnum.IRIS_COMPETENCY_GENERATION
+        )
         response = response.contents[0].text_content
 
         generated_competencies: list[Competency] = []