Try install module with setuptool

distribution/lambda/Pipfile

@@ -4,6 +4,7 @@ verify_ssl = true
 name = "pypi"
 
 [packages]
+cdk = {file = "cdk", editable = true}
 aws-cdk-lib = "2.95.1"
 cargo-lambda = "1.1.0"
 constructs = "10.3.0"

distribution/lambda/cdk/app.py

@@ -50,6 +50,7 @@ def package_location_from_env(type: Literal["searcher"] | Literal["indexer"]) ->
     indexer_package_location=package_location_from_env("indexer"),
     searcher_package_location=package_location_from_env("searcher"),
     search_api_key=os.getenv("SEARCHER_API_KEY", None),
+    data_generation_interval_sec=os.getenv("DATA_GENERATION_INTERVAL_SEC", 300),
 )
 
 app.synth()

distribution/lambda/cdk/setup.py

@@ -0,0 +1,7 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="cdk",
+    version="0.1.0",
+    packages=find_packages(),
+)

distribution/lambda/cdk/stacks/examples/mock_data_stack.py

@@ -30,6 +30,7 @@ def __init__(
         construct_id: str,
         index_id: str,
         qw_svc: quickwit_service.QuickwitService,
+        data_generation_interval_sec: int,
         **kwargs,
     ):
         super().__init__(scope, construct_id, **kwargs)
@@ -59,7 +60,9 @@ def __init__(
         rule = aws_events.Rule(
             self,
             "ScheduledRule",
-            schedule=aws_events.Schedule.rate(aws_cdk.Duration.minutes(5)),
+            schedule=aws_events.Schedule.rate(
+                aws_cdk.Duration.seconds(data_generation_interval_sec)
+            ),
         )
         rule.add_target(aws_events_targets.LambdaFunction(generator_lambda))
 
@@ -139,6 +142,7 @@ def __init__(
         indexer_package_location: str,
         searcher_package_location: str,
         search_api_key: str | None = None,
+        data_generation_interval_sec: int = 300,
         **kwargs,
     ) -> None:
         """If `search_api_key` is not set, the search API is not deployed."""
@@ -167,7 +171,13 @@ def __init__(
             searcher_package_location=searcher_package_location,
         )
 
-        Source(self, "Source", index_id=index_id, qw_svc=qw_svc)
+        Source(
+            self,
+            "Source",
+            index_id=index_id,
+            qw_svc=qw_svc,
+            data_generation_interval_sec=data_generation_interval_sec,
+        )
 
         if search_api_key is not None:
             SearchAPI(