Enable ddtrace in image and update dependency

airbytehq · Nov 21, 2024 · 814696c · 814696c
1 parent 660d683
commit 814696c
Show file tree

Hide file tree

Showing 3 changed files with 192 additions and 5 deletions.
diff --git a/...ectors/pipelines/pipelines/airbyte_ci/connectors/build_image/steps/build_customization.py b/...ectors/pipelines/pipelines/airbyte_ci/connectors/build_image/steps/build_customization.py
@@ -57,13 +57,12 @@ def get_main_file_name(connector: Connector) -> str:
 
 def get_entrypoint(connector: Connector) -> List[str]:
     main_file_name = get_main_file_name(connector)
-    return ["python", f"/airbyte/integration_code/{main_file_name}"]
+    return ["ddtrace-run", "python", f"/airbyte/integration_code/{main_file_name}"]
 
 
 def apply_airbyte_entrypoint(connector_container: Container, connector: Connector) -> Container:
     entrypoint = get_entrypoint(connector)
-
-    return connector_container.with_env_variable("AIRBYTE_ENTRYPOINT", " ".join(entrypoint)).with_entrypoint(entrypoint)
+    return connector_container.with_env_variable("DD_PROFILING_ENABLED", "true").with_env_variable("AIRBYTE_ENTRYPOINT", " ".join(entrypoint)).with_entrypoint(entrypoint)
 
 
 async def pre_install_hooks(connector: Connector, base_container: Container, logger: Logger) -> Container:

diff --git a/airbyte-integrations/connectors/source-klaviyo/poetry.lock b/airbyte-integrations/connectors/source-klaviyo/poetry.lock
diff --git a/airbyte-integrations/connectors/source-klaviyo/pyproject.toml b/airbyte-integrations/connectors/source-klaviyo/pyproject.toml
@@ -18,6 +18,7 @@ include = "source_klaviyo"
 [tool.poetry.dependencies]
 python = "^3.10,<3.12"
 airbyte_cdk = "^6"
+ddtrace = "2.16.4"
 
 [tool.poetry.scripts]
 source-klaviyo = "source_klaviyo.run:run"