From b10e6700088e8d370efb70c05730c1371529ecd2 Mon Sep 17 00:00:00 2001 From: Maximilian Anzinger Date: Mon, 16 Sep 2024 19:12:48 +0200 Subject: [PATCH 1/2] Update competency_extraction_pipeline.py --- app/pipeline/competency_extraction_pipeline.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/app/pipeline/competency_extraction_pipeline.py b/app/pipeline/competency_extraction_pipeline.py index da224ffe..00856eed 100644 --- a/app/pipeline/competency_extraction_pipeline.py +++ b/app/pipeline/competency_extraction_pipeline.py @@ -31,7 +31,10 @@ 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__( From 2826eb91ff68572e5c8f14020c584deb310df15e Mon Sep 17 00:00:00 2001 From: Maximilian Anzinger Date: Mon, 16 Sep 2024 19:17:13 +0200 Subject: [PATCH 2/2] fix style --- app/pipeline/competency_extraction_pipeline.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/app/pipeline/competency_extraction_pipeline.py b/app/pipeline/competency_extraction_pipeline.py index 00856eed..a2288ab5 100644 --- a/app/pipeline/competency_extraction_pipeline.py +++ b/app/pipeline/competency_extraction_pipeline.py @@ -31,10 +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( - gpt_version_equivalent=4.5, - context_length=16385, - )) + self.request_handler = CapabilityRequestHandler( + requirements=RequirementList( + gpt_version_equivalent=4.5, + context_length=16385, + ) + ) self.output_parser = PydanticOutputParser(pydantic_object=Competency) def __call__(