Add hard limit on generated competencies to max_n

ls1intum · Aug 9, 2024 · 117a6e2 · 117a6e2
1 parent 41df050
commit 117a6e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/pipeline/competency_extraction_pipeline.py b/app/pipeline/competency_extraction_pipeline.py
@@ -66,8 +66,8 @@ def __call__(
 
         generated_competencies: list[Competency] = []
 
-        # Find all competencies in the response
-        competencies = response.split("\n\n")
+        # Find all competencies in the response up to the max_n
+        competencies = response.split("\n\n")[: dto.max_n]
         for i, competency in enumerate(competencies):
             logger.debug(f"Processing competency {i + 1}: {competency}")
             if "{" not in competency or "}" not in competency: