feat: add few-shot-lambda to configuration schema

Author: Daniel Lorch

lib/idp_common_pkg/idp_common/config/models.py

@@ -115,6 +115,9 @@ class ExtractionConfig(BaseModel):
     custom_prompt_lambda_arn: Optional[str] = Field(
         default=None, description="ARN of custom prompt Lambda"
     )
+    dynamic_few_shot_lambda_arn: Optional[str] = Field(
+        default=None, description="ARN of Lambda to provide dynamic few-shot document examples with ground truth data"
+    )
 
     @field_validator("temperature", "top_p", "top_k", mode="before")
     @classmethod

patterns/pattern-2/template.yaml

@@ -1008,6 +1008,10 @@ Resources:
                 type: string
                 description: "(Optional) ARN of a Lambda function to generate custom extraction prompts. Function name must start with 'GENAIIDP-'. If not provided, default prompts will be used. The Lambda function receives the complete config, prompt placeholders, default task prompt content, and serialized document, and returns custom system_prompt and task_prompt_content. Example: arn:${AWS::Partition}:lambda:us-east-1:123456789012:function:GENAIIDP-my-extractor"
                 order: 8
+              dynamic_few_shot_lambda_arn:
+                type: string
+                description: !Sub "(Optional) ARN of Lambda to provide dynamic few-shot document examples with ground truth data. Function name must start with 'GENAIIDP-'. If not provided, few-shot examples from config are used. The Lambda function receives the class label, document texts and base64 encoded images, and returns images with expected attributes. Example: arn:${AWS::Partition}:lambda:us-east-1:123456789012:function:GENAIIDP-dynamic-few-shot-examples"
+                order: 9
           assessment:
             order: 5
             type: object