Add requirements list to competency extraction pipeline (#158)

ls1intum · Sep 17, 2024 · df45ef0 · df45ef0
1 parent 4dc9fa7
commit df45ef0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/app/pipeline/competency_extraction_pipeline.py b/app/pipeline/competency_extraction_pipeline.py
@@ -31,7 +31,12 @@ def __init__(self, callback: Optional[CompetencyExtractionCallback] = None):
             implementation_id="competency_extraction_pipeline_reference_impl"
         )
         self.callback = callback
-        self.request_handler = CapabilityRequestHandler(requirements=RequirementList())
+        self.request_handler = CapabilityRequestHandler(
+            requirements=RequirementList(
+                gpt_version_equivalent=4.5,
+                context_length=16385,
+            )
+        )
         self.output_parser = PydanticOutputParser(pydantic_object=Competency)
 
     def __call__(