remove args

PostHog · Jan 26, 2024 · 05bf04f · 05bf04f
1 parent 1776a54
commit 05bf04f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/posthog/temporal/data_imports/external_data_job.py b/posthog/temporal/data_imports/external_data_job.py
@@ -168,7 +168,8 @@ async def run_external_data_job(inputs: ExternalDataJobInputs) -> None:
         if not stripe_secret_key:
             raise ValueError(f"Stripe secret key not found for job {model.id}")
         source = stripe_source(
-            api_key=stripe_secret_key, endpoints=tuple(inputs.schemas), job_id=str(model.id), team_id=inputs.team_id
+            api_key=stripe_secret_key,
+            endpoints=tuple(inputs.schemas),
         )
     elif model.pipeline.source_type == ExternalDataSource.Type.HUBSPOT:
         from posthog.temporal.data_imports.pipelines.hubspot.auth import refresh_access_token
@@ -185,8 +186,6 @@ async def run_external_data_job(inputs: ExternalDataJobInputs) -> None:
         source = hubspot(
             api_key=hubspot_access_code,
             refresh_token=refresh_token,
-            job_id=str(model.id),
-            team_id=inputs.team_id,
             endpoints=tuple(inputs.schemas),
         )
     elif model.pipeline.source_type == ExternalDataSource.Type.POSTGRES:

diff --git a/posthog/temporal/data_imports/pipelines/hubspot/__init__.py b/posthog/temporal/data_imports/pipelines/hubspot/__init__.py
@@ -50,8 +50,6 @@
 def hubspot(
     api_key: str,
     refresh_token: str,
-    job_id: str,
-    team_id: int,
     endpoints: Sequence[str] = ("companies", "contacts", "deals", "tickets", "quotes"),
     include_history: bool = False,
 ) -> Iterable[DltResource]:

diff --git a/posthog/temporal/data_imports/pipelines/stripe/helpers.py b/posthog/temporal/data_imports/pipelines/stripe/helpers.py
@@ -82,7 +82,7 @@ def stripe_pagination(
 
 @dlt.source(max_table_nesting=0)
 def stripe_source(
-    api_key: str, endpoints: Tuple[str, ...], job_id: str, team_id: int, starting_after: Optional[str] = None
+    api_key: str, endpoints: Tuple[str, ...], starting_after: Optional[str] = None
 ) -> Iterable[DltResource]:
     for endpoint in endpoints:
         yield dlt.resource(