Restore createIndexFunction handler

Author: kris-szlapa

packages/createIndexFunction/app.py

@@ -1,7 +1,6 @@
 import json
 import logging
 import os
-import time
 
 import boto3
 from botocore.exceptions import NoCredentialsError
@@ -33,29 +32,6 @@ def get_opensearch_client(endpoint):
     )
 
 
-def wait_for_index(opensearch_client, index_name, timeout=120, poll_interval=4):
-    """
-    Waits for the OpenSearch index to exist and be at least 'yellow' health.
-    """
-    logger.info(f"Polling for index '{index_name}' to exist and be ready...")
-    start = time.time()
-    while True:
-        try:
-            if opensearch_client.indices.exists(index=index_name):
-                health = opensearch_client.cluster.health(index=index_name, wait_for_status="yellow", timeout="5s")
-                status = health.get("status")
-                logger.info(f"Index '{index_name}' exists, health: {status}")
-                if status in ("yellow", "green"):
-                    return
-            else:
-                logger.info(f"Index '{index_name}' does not exist yet...")
-        except Exception as exc:
-            logger.info(f"Error checking index status: {exc}")
-        if time.time() - start > timeout:
-            raise TimeoutError(f"Timed out waiting for index '{index_name}' to become ready.")
-        time.sleep(poll_interval)
-
-
 def handler(event, context):
     logger.info("Received event: %s", json.dumps(event, indent=2))
     print(event)
@@ -90,11 +66,11 @@ def handler(event, context):
                                     "space_type": "l2",
                                 },
                             },
-                            "AMAZON_BEDROCK_METADATA": {
+                            "AMAZON_BEDROCK_METADATA\t": {
                                 "type": "text",
                                 "index": "false",
                             },
-                            "AMAZON_BEDROCK_TEXT_CHUNK": {
+                            "AMAZON_BEDROCK_TEXT_CHUNK\t": {
                                 "type": "text",
                                 "index": "true",
                             },
@@ -104,23 +80,11 @@ def handler(event, context):
             }
 
             try:
-                # Check if index exists first
-                if not opensearch_client.indices.exists(index=params["index"]):
-                    logger.info(f"Creating index {params['index']}")
-                    opensearch_client.indices.create(
-                        index=params["index"], body=params["body"]
-                    )
-                    logger.info(f"Index {params['index']} creation initiated.")
-                else:
-                    logger.info(f"Index {params['index']} already exists")
-                # Wait for the index to be available and ready
-                wait_for_index(opensearch_client, params["index"], timeout=300)
-                logger.info(f"Index {params['index']} is ready.")
-                time.sleep(10)
-                logger.info(f"Index {params['index']} is ready.")
+                opensearch_client.indices.create(
+                    index=params["index"], body=params["body"]
+                )
             except Exception as e:
-                logger.error(f"Error creating or waiting for index: {e}")
-                raise e  # Re-raise to fail the custom resource
+                logger.error(e)
 
         elif event["RequestType"] == "Delete":
             try: