refactor: improve typescript type safety across codebase (#68)

* chore: improve developer experience

Signed-off-by: mateonunez <mateonunez95@gmail.com>

* fix(retove): slack query with consistent cursor comparison

Signed-off-by: mateonunez <mateonunez95@gmail.com>

* chore: fix and lint

Signed-off-by: mateonunez <mateonunez95@gmail.com>

* docs(readme): add seed step

Signed-off-by: mateonunez <mateonunez95@gmail.com>

---------

Signed-off-by: mateonunez <mateonunez95@gmail.com>

Author: mateonunez

README.md

@@ -29,11 +29,14 @@ cp .env.example .env
 # 4. Initialize database
 pnpm migrate
 
-# 5. Install Ollama models (required for AI features)
-docker exec -it ait_ollama ollama pull gemma3:latest # gpt-oss:20b-cloud
+# 5. Seed database (Initial providers configuration)
+pnpm run --filter @ait/postgres db:seed
+
+# 6. Install Ollama models (required for AI features)
+docker exec -it ait_ollama ollama pull gemma3:latest
 docker exec -it ait_ollama ollama pull mxbai-embed-large:latest
 
-# 6. Start development servers
+# 7. Start development servers
 pnpm dev
 ```
 
@@ -45,7 +48,7 @@ AIt follows a modular monorepo architecture with clear separation of concerns:
 ┌─────────────────────────────────────────────────────────────────────────────┐
 │                                   UIt                                       │
 │                          (Web Interface - React)                            │
-└─────────────────────────────────┬───────────────────────────────────────────┘
+└─────────────────────────────────────┬───────────────────────────────────────┘
                                   │
                                   ▼
 ┌─────────────────────────────────────────────────────────────────────────────┐
@@ -65,7 +68,7 @@ AIt follows a modular monorepo architecture with clear separation of concerns:
 │                         INFRASTRUCTURE LAYER                                │
 ├─────────────────┬─────────────────┬─────────────────┬───────────────────────┤
 │   PostgreSQL    │     Qdrant      │     Ollama      │        Redis          │
-│ (Structured DB) │ (Vector Store)  │  (Local LLM)    │    (Job Queue)        │
+│ (Structured DB) │ (Vector Store)  │   (Local LLM)   │      (Job Queue)      │
 └─────────────────┴─────────────────┴─────────────────┴───────────────────────┘
 ```
 

packages/apps/uit/src/components/chat/chat-header-actions.tsx

@@ -1,6 +1,7 @@
 "use client";
 
 import { cn } from "@/styles/utils";
+import type { TokenUsage } from "@ait/core";
 import { MessageSquarePlus, PanelRight } from "lucide-react";
 import { ModelSelector } from "../ai-elements";
 import { ContextWindowTracker } from "../context-window-tracker";
@@ -17,7 +18,7 @@ export const ChatHeaderActions = ({
 }: {
   selectedModel: string;
   setSelectedModel: (id: string) => void;
-  tokenUsage: any;
+  tokenUsage: TokenUsage;
   maxContextWindow: number;
   isLoading: boolean;
   messagesLength: number;

packages/apps/uit/src/components/chat/chat-sidebar-panel.tsx

@@ -2,7 +2,7 @@
 
 import { useIsMobile } from "@/hooks/use-mobile";
 
-import type { Conversation as ConversationType } from "@ait/core";
+import type { ChatMessageWithMetadata, Conversation as ConversationType, TokenUsage } from "@ait/core";
 import { AnimatePresence, motion } from "framer-motion";
 import { Brain, Database, ListChecks, MessageCircle, Wrench, Zap } from "lucide-react";
 import { useMemo } from "react";
@@ -20,11 +20,11 @@ interface ChatSidebarPanelProps {
   handleConversationSelect: (id: string) => void;
   handleDeleteConversation: (id: string) => Promise<void>;
   handleNewChat: () => void;
-  tokenUsage: any;
+  tokenUsage: TokenUsage;
   maxContextWindow: number;
   selectedModel: string;
   setSelectedModel: (id: string) => void;
-  latestAssistantMessage: any;
+  latestAssistantMessage: ChatMessageWithMetadata | null;
   isOpen?: boolean;
   onClose?: () => void;
 }

packages/apps/uit/src/components/context-window-tracker.tsx

@@ -1,6 +1,6 @@
-import type { TokenUsage } from "@/hooks/useAItChat";
 import { cn } from "@/styles/utils";
 import { calculateContextWindowUsage, formatTokenCount } from "@/utils/token-counter.utils";
+import type { TokenUsage } from "@ait/core";
 import { motion } from "framer-motion";
 import { AlertTriangle, Database, TrendingUp } from "lucide-react";
 import { useMemo } from "react";

packages/apps/uit/src/components/discovery/discovery-stats.tsx

@@ -8,6 +8,7 @@ import { Skeleton } from "@/components/ui/skeleton";
 import { ToggleGroup, ToggleGroupItem } from "@/components/ui/toggle-group";
 import { useInsights } from "@/contexts/insights.context";
 import { useConnectionStatus } from "@/hooks/useConnectionStatus";
+import type { DailyActivity, DiscoveryStatsData } from "@/services/discovery.service";
 import { fetchDiscoveryStats } from "@/services/observability.service";
 import { cn } from "@/styles/utils";
 import { getLogger } from "@ait/core";
@@ -33,17 +34,6 @@ import { AiInsightsPanel } from "./ai-insights-panel";
 
 const logger = getLogger();
 
-interface DailyActivity {
-  date: string;
-  [integrationKey: string]: number | string;
-}
-
-interface DiscoveryStatsData {
-  timeRange: "week" | "month" | "year";
-  data: DailyActivity[];
-  totals: Record<string, number>;
-}
-
 interface StatCardProps {
   item: {
     id: string;

packages/apps/uit/src/hooks/useAItChat.ts

@@ -1,7 +1,7 @@
 import { getConversation, sendMessage } from "@/services/chat.service";
 import { createEmptyMetadata } from "@/utils/stream-parser.utils";
 import { calculateConversationTokens } from "@/utils/token-counter.utils";
-import type { ChatMessageWithMetadata, MessageRole } from "@ait/core";
+import type { ChatMessageWithMetadata, MessageRole, TokenUsage } from "@ait/core";
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 
 export interface UseAItChatOptions {
@@ -13,13 +13,6 @@ export interface UseAItChatOptions {
   onError?: (error: string) => void;
 }
 
-export interface TokenUsage {
-  inputTokens: number;
-  outputTokens: number;
-  ragContextTokens: number;
-  totalTokens: number;
-}
-
 export interface UseAItChatReturn {
   messages: ChatMessageWithMetadata[];
   isLoading: boolean;
@@ -70,7 +63,7 @@ export function useAItChat(options: UseAItChatOptions = {}): UseAItChatReturn {
           id: message.id,
           role: message.role as MessageRole,
           content: message.content,
-          metadata: message.metadata as any,
+          metadata: message.metadata as ChatMessageWithMetadata["metadata"],
           traceId: message.traceId || undefined,
           createdAt: new Date(message.createdAt),
         }));

packages/apps/uit/src/services/discovery.service.ts

@@ -44,6 +44,17 @@ export interface DiscoveryServiceOptions {
   cacheTtlMs?: number;
 }
 
+export interface DailyActivity {
+  date: string;
+  [integrationKey: string]: number | string;
+}
+
+export interface DiscoveryStatsData {
+  timeRange: "week" | "month" | "year";
+  data: DailyActivity[];
+  totals: Record<string, number>;
+}
+
 /**
  * Fetch discovery feed with caching and retry logic
  */
@@ -109,19 +120,19 @@ export async function fetchDiscoveryFeed(
 /**
  * Fetch discovery stats with retries
  */
-export async function fetchDiscoveryStats(range: "week" | "month" | "year" = "week"): Promise<any> {
+export async function fetchDiscoveryStats(range: "week" | "month" | "year" = "week"): Promise<DiscoveryStatsData> {
   const url = `${apiConfig.apiBaseUrl}/discovery/stats?range=${range}`;
   let lastError: Error | null = null;
 
   for (let attempt = 1; attempt <= MAX_RETRIES; attempt++) {
     try {
-      const res = await apiGet<any>(url);
+      const res = await apiGet<unknown>(url);
 
       if (!res.ok) {
         throw new Error(res.error || "Failed to fetch discovery stats");
       }
 
-      return res.data;
+      return res.data as DiscoveryStatsData;
     } catch (error) {
       lastError = error instanceof Error ? error : new Error(String(error));
       logger.warn("[DiscoveryService] Stats fetch attempt failed", {

packages/connectors/src/domain/entities/vendors/google/connector.google-calendar-event.repository.ts

@@ -52,34 +52,34 @@ export class ConnectorGoogleCalendarEventRepository implements IConnectorGoogleC
           locked: eventDataTarget.locked,
           attendeesOmitted: eventDataTarget.attendeesOmitted,
           attendeesCount: eventDataTarget.attendeesCount,
-          creatorData: eventDataTarget.creatorData as any,
-          organizerData: eventDataTarget.organizerData as any,
-          attendeesData: eventDataTarget.attendeesData as any,
-          recurrenceData: eventDataTarget.recurrenceData as any,
-          remindersData: eventDataTarget.remindersData as any,
-          conferenceData: eventDataTarget.conferenceData as any,
-          attachmentsData: eventDataTarget.attachmentsData as any,
-          extendedPropertiesData: eventDataTarget.extendedPropertiesData as any,
-          metadata: eventDataTarget.metadata as any,
+          creatorData: eventDataTarget.creatorData as Record<string, unknown>,
+          organizerData: eventDataTarget.organizerData as Record<string, unknown>,
+          attendeesData: eventDataTarget.attendeesData as unknown[],
+          recurrenceData: eventDataTarget.recurrenceData as unknown[],
+          remindersData: eventDataTarget.remindersData as Record<string, unknown>,
+          conferenceData: eventDataTarget.conferenceData as Record<string, unknown>,
+          attachmentsData: eventDataTarget.attachmentsData as unknown[],
+          extendedPropertiesData: eventDataTarget.extendedPropertiesData as Record<string, unknown>,
+          metadata: eventDataTarget.metadata as Record<string, unknown>,
           eventCreatedAt: eventDataTarget.eventCreatedAt,
           eventUpdatedAt: eventDataTarget.eventUpdatedAt,
           updatedAt: new Date(),
         };
 
         await tx
           .insert(googleCalendarEvents)
-          .values(eventDataTarget as any)
+          .values(eventDataTarget as GoogleCalendarEventDataTarget)
           .onConflictDoUpdate({
             target: googleCalendarEvents.id,
             set: updateValues,
           })
           .execute();
       });
-    } catch (error: any) {
+    } catch (error: unknown) {
       logger.error("Failed to save Google Calendar event:", { eventId: event.id, error });
       throw new AItError(
         "GOOGLE_CALENDAR_SAVE_EVENT",
-        `Failed to save event ${event.id}: ${error.message}`,
+        `Failed to save event ${event.id}: ${(error as Error).message}`,
         { id: event.id },
         error,
       );

packages/connectors/src/domain/entities/vendors/slack/connector.slack-message.repository.ts

@@ -35,18 +35,18 @@ export class ConnectorSlackMessageRepository implements IConnectorSlackMessageRe
 
         await tx
           .insert(slackMessages)
-          .values(messageDataTarget as any)
+          .values(messageDataTarget as SlackMessageDataTarget)
           .onConflictDoUpdate({
             target: slackMessages.id,
             set: updateValues,
           })
           .execute();
       });
-    } catch (error: any) {
+    } catch (error: unknown) {
       logger.error("Failed to save message:", { messageId, error });
       throw new AItError(
         "SLACK_SAVE_MESSAGE",
-        `Failed to save message ${messageId}: ${error.message}`,
+        `Failed to save message ${messageId}: ${(error as Error).message}`,
         { id: messageId },
         error,
       );

packages/connectors/src/domain/entities/vendors/x/connector.x.repository.ts

@@ -48,16 +48,21 @@ export class ConnectorXTweetRepository implements IConnectorXTweetRepository {
 
         await tx
           .insert(xTweets)
-          .values(tweetData as any)
+          .values(tweetData as XTweetDataTarget)
           .onConflictDoUpdate({
             target: xTweets.id,
             set: updateValues,
           })
           .execute();
       });
-    } catch (error: any) {
+    } catch (error: unknown) {
       logger.error("Failed to save tweet:", { tweetId, error });
-      throw new AItError("X_SAVE_TWEET", `Failed to save tweet ${tweetId}: ${error.message}`, { id: tweetId }, error);
+      throw new AItError(
+        "X_SAVE_TWEET",
+        `Failed to save tweet ${tweetId}: ${(error as Error).message}`,
+        { id: tweetId },
+        error,
+      );
     }
   }