refactor(types): remove forced tokenlens and cache casts

Remove forced tokenlens SourceModel casts and replace the links null cache cast with an honest symbol sentinel.

Author: izadoesdev

apps/dashboard/components/ai-elements/context.tsx

@@ -13,12 +13,11 @@ import { computeTokenCostsForModel } from "tokenlens/helpers";
 import { vercelModels } from "tokenlens/providers/vercel";
 import { Button, Progress } from "@databuddy/ui";
 
-type VercelModelId = keyof typeof vercelModels.models;
-
 const lookupModel = (modelId: string): SourceModel | undefined => {
-	const model = vercelModels.models[modelId as VercelModelId];
+	const model =
+		vercelModels.models[modelId as keyof typeof vercelModels.models];
 	return model
-		? ({ canonical_id: model.id, ...model } as unknown as SourceModel)
+		? ({ canonical_id: model.id, ...model } satisfies SourceModel)
 		: undefined;
 };
 

apps/links/src/routes/redirect.ts

@@ -23,8 +23,11 @@ const EXPIRED_URL = `${config.urls.dashboard}/dby/expired`;
 const NOT_FOUND_URL = `${config.urls.dashboard}/dby/not-found`;
 const OG_PROXY_URL = `${config.urls.dashboard}/dby/l`;
 
-const NULL_SENTINEL = Object.freeze({ __null: true }) as unknown as CachedLink;
-const linkCache = new LRUCache<string, CachedLink>({ max: 1000, ttl: 5000 });
+const NULL_SENTINEL = Symbol("null");
+const linkCache = new LRUCache<string, CachedLink | typeof NULL_SENTINEL>({
+	max: 1000,
+	ttl: 5000,
+});
 const etagCache = new LRUCache<string, string>({ max: 1000, ttl: 60_000 });
 const dedupCache = new LRUCache<string, true>({ max: 10_000, ttl: 300_000 });
 const botCache = new LRUCache<string, { isBot: boolean; isSocial: boolean }>({

packages/ai/src/lib/usage-telemetry.ts

@@ -9,48 +9,44 @@ import type { SourceModel } from "tokenlens";
 import { computeTokenCostsForModel } from "tokenlens/helpers";
 import { vercelModels } from "tokenlens/providers/vercel";
 
-type VercelModelId = keyof typeof vercelModels.models;
-
 interface CostModel {
 	fallback: boolean;
 	id: string;
 	model: SourceModel;
 }
 
-function createSourceModel(
-	id: string,
-	cost: AgentModelCostUsdPerMillion
-): SourceModel {
-	return { canonical_id: id, cost, id } as unknown as SourceModel;
-}
-
 function createCostModel(
 	id: string,
 	cost: AgentModelCostUsdPerMillion,
 	fallback = false
 ): CostModel {
-	return { fallback, id, model: createSourceModel(id, cost) };
-}
-
-function lookupConfiguredModel(modelId: string): CostModel | null {
-	const resolved = lookupAgentModelCost(modelId);
-	return resolved ? createCostModel(resolved.id, resolved.cost) : null;
+	return {
+		fallback,
+		id,
+		model: { canonical_id: id, cost, id, name: id } satisfies SourceModel,
+	};
 }
 
 function lookupTokenlensModel(modelId: string): CostModel | null {
-	const model = vercelModels.models[modelId as VercelModelId];
+	const model =
+		vercelModels.models[modelId as keyof typeof vercelModels.models];
 	if (!model?.cost) {
 		return null;
 	}
 	return {
 		fallback: false,
 		id: model.id,
-		model: { canonical_id: model.id, ...model } as unknown as SourceModel,
+		model: { canonical_id: model.id, ...model } satisfies SourceModel,
 	};
 }
 
 function resolveCostModel(modelId: string): CostModel {
-	const model = lookupConfiguredModel(modelId) ?? lookupTokenlensModel(modelId);
+	const configured = lookupAgentModelCost(modelId);
+	if (configured) {
+		return createCostModel(configured.id, configured.cost);
+	}
+
+	const model = lookupTokenlensModel(modelId);
 	if (model) {
 		return model;
 	}