From f4eaa9fd0dc08c50d1c18a9c7538c9e710e9cdc9 Mon Sep 17 00:00:00 2001
From: Francis Cao <franciscao@gmail.com>
Date: Thu, 22 Jan 2026 16:32:06 -0800
Subject: [PATCH] update segmentParamSchema, add typing to getEntity

---
 src/app/api/websites/[websiteId]/segments/route.ts | 4 ++--
 src/lib/entity.ts                                  | 3 ++-
 src/lib/schema.ts                                  | 2 +-
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/app/api/websites/[websiteId]/segments/route.ts b/src/app/api/websites/[websiteId]/segments/route.ts
index 10d47a81..db34193e 100644
--- a/src/app/api/websites/[websiteId]/segments/route.ts
+++ b/src/app/api/websites/[websiteId]/segments/route.ts
@@ -2,7 +2,7 @@ import { z } from 'zod';
 import { uuid } from '@/lib/crypto';
 import { getQueryFilters, parseRequest } from '@/lib/request';
 import { json, unauthorized } from '@/lib/response';
-import { searchParams, segmentParametersSchema, segmentTypeParam } from '@/lib/schema';
+import { searchParams, segmentParamSchema, segmentTypeParam } from '@/lib/schema';
 import { canUpdateWebsite, canViewWebsite } from '@/permissions';
 import { createSegment, getWebsiteSegments } from '@/queries/prisma';
 
@@ -42,7 +42,7 @@ export async function POST(
   const schema = z.object({
     type: segmentTypeParam,
     name: z.string().max(200),
-    parameters: segmentParametersSchema,
+    parameters: segmentParamSchema,
   });
 
   const { auth, body, error } = await parseRequest(request, schema);
diff --git a/src/lib/entity.ts b/src/lib/entity.ts
index 1b64e5dd..fd26252d 100644
--- a/src/lib/entity.ts
+++ b/src/lib/entity.ts
@@ -1,6 +1,7 @@
+import type { Link, Pixel, Website } from '@/generated/prisma/client';
 import { getLink, getPixel, getWebsite } from '@/queries/prisma';
 
-export async function getEntity(entityId: string) {
+export async function getEntity(entityId: string): Promise<Website | Link | Pixel | null> {
   const website = await getWebsite(entityId);
   const link = await getLink(entityId);
   const pixel = await getPixel(entityId);
diff --git a/src/lib/schema.ts b/src/lib/schema.ts
index addce3a2..ac360a8e 100644
--- a/src/lib/schema.ts
+++ b/src/lib/schema.ts
@@ -249,7 +249,7 @@ export const reportResultSchema = z.intersection(
 
 export const segmentTypeParam = z.enum(['segment', 'cohort']);
 
-export const segmentParametersSchema = z.object({
+export const segmentParamSchema = z.object({
   filters: z
     .array(
       z.object({