@twentyhq/people-data-labs 1.0.2 → 1.0.7

package/src/logic-functions/enrich-companies.function.ts

@@ -0,0 +1,77 @@
+import { defineLogicFunction, STANDARD_OBJECT_UNIVERSAL_IDENTIFIERS } from 'twenty-sdk/define';
+
+import { UPDATE_FIELDS_OPTION_VALUES } from 'src/constants/update-fields-option-values';
+import { PDL_LOGIC_FUNCTION_CONSTANTS } from 'src/constants/universal-identifiers';
+import { enrichCompaniesCore } from 'src/logic-functions/handlers/enrich-companies';
+import { type EnrichInput, toBulkEnrichInput } from 'src/logic-functions/utils/to-bulk-enrich-input';
+
+const handler = (input: EnrichInput) => {
+  return enrichCompaniesCore({ input: toBulkEnrichInput(input) });
+}
+
+export default defineLogicFunction({
+  universalIdentifier: PDL_LOGIC_FUNCTION_CONSTANTS.enrichCompanies.universalIdentifier,
+  name: 'enrich-companies',
+  description: 'Enrich one or more Company records with People Data Labs data',
+  timeoutSeconds: 300,
+  handler,
+  httpRouteTriggerSettings: {
+    path: PDL_LOGIC_FUNCTION_CONSTANTS.enrichCompanies.path,
+    httpMethod: 'POST',
+    isAuthRequired: true
+  },
+  workflowActionTriggerSettings: {
+    label: 'Enrich Companies',
+    icon: 'IconSparkles',
+    inputSchema: [
+      {
+        type: 'object',
+        properties: {
+          records: {
+            type: 'records',
+            objectUniversalIdentifier:
+              STANDARD_OBJECT_UNIVERSAL_IDENTIFIERS.company.universalIdentifier,
+            label: 'Records',
+          },
+          updateFields: {
+            type: 'string',
+            label: 'Update fields',
+            enum: [...UPDATE_FIELDS_OPTION_VALUES],
+          },
+        },
+      },
+    ],
+    outputSchema: [
+      {
+        type: 'object',
+        properties: {
+          success: { type: 'boolean', label: 'Success' },
+          total: { type: 'number', label: 'Total' },
+          matched: { type: 'number', label: 'Matched' },
+          notFound: { type: 'number', label: 'Not Found' },
+          skipped: { type: 'number', label: 'Skipped' },
+          errored: { type: 'number', label: 'Errored' },
+          results: {
+            type: 'array',
+            items: {
+              type: 'object',
+              properties: {
+                success: { type: 'boolean', label: 'Success' },
+                recordId: { type: 'string', label: 'Record Id' },
+                status: { type: 'string', label: 'Status' },
+                updatedFields: {
+                  type: 'array',
+                  items: { type: 'string' },
+                  label: 'Updated Fields',
+                },
+                data: { type: 'object', label: 'Data' },
+                message: { type: 'string', label: 'Message' },
+              },
+            },
+            label: 'Results',
+          },
+        },
+      },
+    ],
+  },
+});

package/src/logic-functions/enrich-company.function.mjs

@@ -211,6 +211,7 @@ var __anyHandler = {
 };
 var __anyStub = new Proxy(() => void 0, __anyHandler);
 var defineLogicFunction = __defineFactoryStub;
+var STANDARD_OBJECT_UNIVERSAL_IDENTIFIERS = __anyStub;
 
 // src/constants/update-fields-options.ts
 var UPDATE_FIELDS_OPTIONS = {
@@ -5728,32 +5729,36 @@ var buildSkippedResult = ({
 });
 
 // node_modules/twenty-sdk/dist/billing/index.mjs
-var e = "TWENTY_API_URL";
-var t = "TWENTY_APP_ACCESS_TOKEN";
-var n = 5e3;
-var r = async ({ creditsUsedMicro: r2, operationType: i, quantity: a = 1, resourceContext: o }) => {
-  let s = process.env[e], c = process.env[t];
+var e = /* @__PURE__ */ (function(e2) {
+  return e2.ACTOR = "ACTOR", e2.ADDRESS = "ADDRESS", e2.ARRAY = "ARRAY", e2.BOOLEAN = "BOOLEAN", e2.CURRENCY = "CURRENCY", e2.DATE = "DATE", e2.DATE_TIME = "DATE_TIME", e2.EMAILS = "EMAILS", e2.FILES = "FILES", e2.FULL_NAME = "FULL_NAME", e2.LINKS = "LINKS", e2.MORPH_RELATION = "MORPH_RELATION", e2.MULTI_SELECT = "MULTI_SELECT", e2.NUMBER = "NUMBER", e2.NUMERIC = "NUMERIC", e2.PHONES = "PHONES", e2.POSITION = "POSITION", e2.RATING = "RATING", e2.RAW_JSON = "RAW_JSON", e2.RELATION = "RELATION", e2.RICH_TEXT = "RICH_TEXT", e2.SELECT = "SELECT", e2.TEXT = "TEXT", e2.TS_VECTOR = "TS_VECTOR", e2.UUID = "UUID", e2;
+})({});
+e.TEXT, e.ARRAY, e.BOOLEAN, e.DATE, e.DATE_TIME, e.NUMBER, e.NUMERIC, e.RAW_JSON, e.RICH_TEXT, e.SELECT, e.MULTI_SELECT;
+var t = "TWENTY_API_URL";
+var n = "TWENTY_APP_ACCESS_TOKEN";
+var r = 5e3;
+var i = async ({ creditsUsedMicro: e2, operationType: i2, quantity: a = 1, resourceContext: o }) => {
+  let s = process.env[t], c = process.env[n];
   if (!(!s || !c)) try {
-    let e2 = await fetch(`${s.replace(/\/$/, "")}/app/billing/charge`, {
+    let t2 = await fetch(`${s.replace(/\/$/, "")}/app/billing/charge`, {
       method: "POST",
       headers: {
         Authorization: `Bearer ${c}`,
         "Content-Type": "application/json"
       },
       body: JSON.stringify({
-        creditsUsedMicro: r2,
+        creditsUsedMicro: e2,
         quantity: a,
-        operationType: i,
+        operationType: i2,
         resourceContext: o
       }),
-      signal: AbortSignal.timeout(n)
+      signal: AbortSignal.timeout(r)
     });
-    if (!e2.ok) {
-      let t2 = await e2.text().catch(() => "");
-      console.error(`chargeCredits: ${e2.status} ${e2.statusText}: ${t2}`);
+    if (!t2.ok) {
+      let e3 = await t2.text().catch(() => "");
+      console.error(`chargeCredits: ${t2.status} ${t2.statusText}: ${e3}`);
     }
-  } catch (e2) {
-    console.error(`chargeCredits: ${e2 instanceof Error ? e2.message : String(e2)}`);
+  } catch (e3) {
+    console.error(`chargeCredits: ${e3 instanceof Error ? e3.message : String(e3)}`);
   }
 };
 
@@ -5775,7 +5780,7 @@ var chargeMatchedEnrichments = async ({
   const creditsPerMatchMicro = Math.round(
     costPerMatchDollars * BILLING_MARGIN_MULTIPLIER * MICRO_CREDITS_PER_DOLLAR
   );
-  await r({
+  await i({
     creditsUsedMicro: matchedCount * creditsPerMatchMicro,
     operationType: "CODE_EXECUTION",
     quantity: matchedCount,
@@ -6165,7 +6170,8 @@ var enrich_company_function_default = defineLogicFunction({
         type: "object",
         properties: {
           recordId: {
-            type: "string",
+            type: "record",
+            objectUniversalIdentifier: STANDARD_OBJECT_UNIVERSAL_IDENTIFIERS.company.universalIdentifier,
             label: "Record"
           },
           updateFields: {