chore: use custom_prompt_lambda_arn parameter

Author: Daniel Lorch

plugins/dynamic-few-shot-lambda/notebooks/step3_extraction_with_dynamic_few_shot.ipynb

@@ -14,13 +14,13 @@
     "- Handle errors and monitor performance\n",
     "\n",
     "**Prerequisites:**\n",
-    "- Completed Step 2 (Classification)\n",
+    "- Completed Step 2 (Classification) (`notebooks/fewshot_dataset_import.ipynb`)\n",
     "- AWS Lambda permissions to create/invoke functions\n",
     "- Dynamic few-shot Lambda function deployed\n",
-    "- S3 Vectors index populated with examples (`notebooks/misc/fewshot_dataset_import.ipynb`)\n",
+    "- S3 Vectors index populated with examples (`notebooks/fewshot_dataset_import.ipynb`)\n",
     "\n",
     "**Key Feature:**\n",
-    "The `dynamic_few_shot_lambda_arn` configuration field allows you to dynamically retrieve similar examples using S3 Vectors similarity search to improve extraction accuracy through few-shot prompting."
+    "The `custom_prompt_lambda_arn` configuration field allows you to dynamically retrieve similar examples using S3 Vectors similarity search to improve extraction accuracy through few-shot prompting."
    ]
   },
   {
@@ -71,7 +71,7 @@
    "outputs": [],
    "source": [
     "# Load document from previous step\n",
-    "classification_data_dir = Path(\".data/step2_classification\")\n",
+    "classification_data_dir = Path(\"../../../notebooks/examples/.data/step2_classification\")\n",
     "\n",
     "# Load document object from JSON\n",
     "document_path = classification_data_dir / \"document.json\"\n",
@@ -174,12 +174,12 @@
    "source": [
     "# Create configuration WITHOUT dynamic few-shot Lambda\n",
     "config_default = CONFIG.copy()\n",
-    "if 'dynamic_few_shot_lambda_arn' in config_default.get('extraction', {}):\n",
-    "    del config_default['extraction']['dynamic_few_shot_lambda_arn']\n",
+    "if 'custom_prompt_lambda_arn' in config_default.get('extraction', {}):\n",
+    "    del config_default['extraction']['custom_prompt_lambda_arn']\n",
     "\n",
     "print(\"=== DEFAULT EXTRACTION CONFIGURATION ===\")\n",
     "print(f\"Model: {config_default.get('extraction', {}).get('model')}\")\n",
-    "print(f\"Dynamic Few-Shot Lambda: {config_default.get('extraction', {}).get('dynamic_few_shot_lambda_arn', 'None')}\")\n",
+    "print(f\"Dynamic Few-Shot Lambda: {config_default.get('extraction', {}).get('custom_prompt_lambda_arn', 'None')}\")\n",
     "\n",
     "# Create extraction service with default config\n",
     "extraction_service_default = extraction.ExtractionService(config=config_default)\n",
@@ -259,7 +259,7 @@
     "if DYNAMIC_FEW_SHOT_LAMBDA_ARN:\n",
     "    # Create configuration WITH dynamic few-shot Lambda\n",
     "    config_few_shot = CONFIG.copy()\n",
-    "    config_few_shot['extraction']['dynamic_few_shot_lambda_arn'] = DYNAMIC_FEW_SHOT_LAMBDA_ARN\n",
+    "    config_few_shot['extraction']['custom_prompt_lambda_arn'] = DYNAMIC_FEW_SHOT_LAMBDA_ARN\n",
     "    \n",
     "    print(\"=== DYNAMIC FEW-SHOT EXTRACTION CONFIGURATION ===\")\n",
     "    print(f\"Model: {config_few_shot.get('extraction', {}).get('model')}\")\n",