profile-builder: allow to override profile_id

jitsucom · Feb 10, 2025 · 8f83ac8 · 8f83ac8
1 parent 0179928
commit 8f83ac8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/libs/core-functions/src/functions/lib/profiles-udf-wrapper.ts b/libs/core-functions/src/functions/lib/profiles-udf-wrapper.ts
@@ -493,7 +493,7 @@ export async function ProfileUDFTestRun({
     }
     const result = await wrapper?.userFunction(eventsProvider, userProvider, funcCtx);
     const profile = {
-      profile_id: user.id,
+      profile_id: result?.profile_id || user.id,
       traits: { ...user.traits, ...result?.traits },
       version: version,
       updated_at: new Date(),

diff --git a/services/profiles/src/lib/functions-chain.ts b/services/profiles/src/lib/functions-chain.ts
@@ -164,7 +164,7 @@ export async function runChain(
   try {
     result = await f.exec(eventsProvider, userProvider, f.context);
     return {
-      profile_id: profileId,
+      profile_id: result?.profile_id || profileId,
       traits: { ...(await userProvider()).traits, ...result?.traits },
       version: profileBuilder.version,
       updated_at: new Date(),

diff --git a/types/protocols/profile.d.ts b/types/protocols/profile.d.ts
@@ -2,6 +2,7 @@ import { FunctionContext } from "./functions";
 import { AnalyticsServerEvent } from "./analytics";
 
 export type ProfileResult = {
+  profile_id?: string;
   traits: Record<string, any>;
 };