refactor ai and sync engine

apps/api/package.json

@@ -37,6 +37,7 @@
     "@nestjs/core": "^11.0.1",
     "@nestjs/platform-express": "^11.0.1",
     "@nestjs/schedule": "^6.1.1",
+    "@nexiom/ai-engine": "workspace:*",
     "@nexiom/auth": "workspace:*",
     "@nexiom/cache": "workspace:*",
     "@nexiom/credentials": "workspace:*",
@@ -46,6 +47,7 @@
     "@nexiom/identity": "workspace:*",
     "@nexiom/piece-framework": "workspace:*",
     "@nexiom/piece-quickbooks": "workspace:*",
+    "@nexiom/piece-registry": "workspace:*",
     "@nexiom/piece-salesforce": "workspace:*",
     "@nexiom/queue": "workspace:*",
     "ai": "^6.0.146",

apps/api/src/app/app.module.ts

@@ -19,7 +19,8 @@ import * as schema from '../db/schema.js';
 import { ConnectionsModule } from '../modules/connections/connections.module.js';
 import { TriggerModule } from '../modules/trigger/trigger.module.js';
 import { EmailModule } from '../modules/email/email.module.js';
-import { StorageResolverModule, PiecesModule } from '@nexiom/engine';
+import { StorageResolverModule } from '@nexiom/engine';
+import { PiecesModule } from '@nexiom/piece-registry';
 import { CacheModule } from '@nexiom/cache';
 import { QueueModule, createQueueModuleOptions } from '@nexiom/queue';
 import { DbManagerModule } from '../modules/dbmanager/dbmanager.module.js';
@@ -33,7 +34,7 @@ import { ObservabilityModule } from '../modules/observability/observability.modu
 import { TraceModule } from '../modules/trace/trace.module.js';
 import { ExceptionsModule } from '../modules/exceptions/exceptions.module.js';
 import { AiModule } from '../modules/ai/ai.module.js';
-import { MetadataModule } from '../modules/metadata/metadata.module.js';
+import { MetadataModule } from '@nexiom/piece-registry';
 
 @Module({
   imports: [

apps/api/src/modules/ai/ai.module.ts

@@ -1,32 +1,13 @@
 import { Module } from '@nestjs/common';
 import { ObservabilityModule } from '../observability/observability.module.js';
-import { AiController } from './_controllers/ai.controller.js';
-import { OrchestratorService } from './_services/orchestrator.service.js';
-import { AiRateLimitGuard } from './_interceptors/ai-ratelimit.guard.js';
-import { AiTelemetryInterceptor } from './_interceptors/ai-telemetry.interceptor.js';
-import { ConnectionsModule } from '../connections/connections.module.js';
-import { StitchesModule } from '../stitches/stitches.module.js';
-import { MetadataModule } from '../metadata/metadata.module.js';
+import { AiController } from './controllers/ai.controller.js';
+import { AiRateLimitGuard } from './interceptors/ai-ratelimit.guard.js';
+import { AiTelemetryInterceptor } from './interceptors/ai-telemetry.interceptor.js';
+import { AiEngineModule } from '@nexiom/ai-engine';
 
-/**
- * AiModule
- *
- * - ObservabilityModule is imported to make @InjectPinoLogger available for
- *   OrchestratorService, AiRateLimitGuard, and AiTelemetryInterceptor.
- * - ConnectionsModule provides TokenManagerService (credential fetching + refresh).
- * - MetadataModule provides MetadataDiscoveryService (cached relation graphs).
- * - PiecesModule is intentionally NOT imported — already registered globally
- *   via AppModule.PiecesModule.forRoot(), so PieceRegistryService is injectable.
- * - CacheModule is likewise global via AppModule.
- */
 @Module({
-  imports: [
-    ObservabilityModule,
-    ConnectionsModule,
-    StitchesModule,
-    MetadataModule,
-  ],
+  imports: [ObservabilityModule, AiEngineModule],
   controllers: [AiController],
-  providers: [OrchestratorService, AiRateLimitGuard, AiTelemetryInterceptor],
+  providers: [AiRateLimitGuard, AiTelemetryInterceptor],
 })
 export class AiModule {}

apps/api/src/modules/ai/_controllers/ai.controller.spec.ts → apps/api/src/modules/ai/controllers/ai.controller.spec.ts

@@ -2,9 +2,9 @@
 import { describe, it, expect, vi, beforeEach } from 'vitest';
 import { Test, TestingModule } from '@nestjs/testing';
 import { AuthGuard } from '@nexiom/auth';
-import { AiRateLimitGuard } from '../_interceptors/ai-ratelimit.guard.js';
+import { AiRateLimitGuard } from '../interceptors/ai-ratelimit.guard.js';
 import { AiController } from './ai.controller.js';
-import { OrchestratorService } from '../_services/orchestrator.service.js';
+import { OrchestratorService } from '@nexiom/ai-engine';
 import { PinoLogger } from 'nestjs-pino';
 import { Response } from 'express';
 import { PassThrough } from 'stream';

apps/api/src/modules/ai/_controllers/ai.controller.ts → apps/api/src/modules/ai/controllers/ai.controller.ts

@@ -11,10 +11,9 @@ import {
   ValidationPipe,
 } from '@nestjs/common';
 import { AuthGuard } from '@nexiom/auth';
-import { AiRateLimitGuard } from '../_interceptors/ai-ratelimit.guard.js';
-import { AiTelemetryInterceptor } from '../_interceptors/ai-telemetry.interceptor.js';
-import { OrchestratorService } from '../_services/orchestrator.service.js';
-import { ChatRequest } from '../_types/chat-request.types.js';
+import { AiRateLimitGuard } from '../interceptors/ai-ratelimit.guard.js';
+import { AiTelemetryInterceptor } from '../interceptors/ai-telemetry.interceptor.js';
+import { OrchestratorService, ChatRequest } from '@nexiom/ai-engine';
 import type { UIMessage } from 'ai';
 import { PinoLogger } from 'nestjs-pino';
 

apps/api/src/modules/connections/connections.module.ts

@@ -13,7 +13,7 @@ import { ConnectorsController } from './connections/connectors.controller.js';
 import { DefaultOAuthRefreshClient } from './connections/token-refresh.service.js';
 import { ConnectorsService } from './connectors.service.js';
 import { OauthStateService } from './oauth-state.service.js';
-import { PiecesModule } from '@nexiom/engine';
+import { PiecesModule } from '@nexiom/piece-registry';
 
 import { DATABASE_CONNECTION } from '@nexiom/database';
 import { type DrizzleDb } from '@nexiom/database';

apps/api/src/modules/connections/connections/callback.controller.spec.ts

@@ -1,7 +1,7 @@
 /* eslint-disable @typescript-eslint/unbound-method */
 import { Test, TestingModule } from '@nestjs/testing';
 import { OAuthCallbackController } from './callback.controller.js';
-import { PieceRegistryService } from '@nexiom/engine';
+import { PieceRegistryService } from '@nexiom/piece-registry';
 import type { Piece } from '@nexiom/piece-framework';
 import type { Request, Response } from 'express';
 import { vi, describe, it, expect, beforeEach } from 'vitest';

apps/api/src/modules/connections/connections/callback.controller.ts

@@ -1,7 +1,7 @@
 import { Controller, Get, Req, Res, Logger } from '@nestjs/common';
 import type { Request, Response } from 'express';
 import { ConfigService } from '@nestjs/config';
-import { PieceRegistryService } from '@nexiom/engine';
+import { PieceRegistryService } from '@nexiom/piece-registry';
 
 import { OauthStateService } from '../oauth-state.service.js';
 

apps/api/src/modules/connections/connections/connectors.controller.spec.ts

@@ -14,7 +14,7 @@ import {
   UnauthorizedException,
   HttpException,
 } from '@nestjs/common';
-import { PieceRegistryService, PIECES } from '@nexiom/engine';
+import { PieceRegistryService, PIECES } from '@nexiom/piece-registry';
 import {
   describe,
   it,

apps/api/src/modules/connections/connections/connectors.controller.ts

@@ -37,7 +37,7 @@ import {
   member,
 } from '@nexiom/database';
 import { eq, and, count, desc } from 'drizzle-orm';
-import { PieceRegistryService } from '@nexiom/engine';
+import { PieceRegistryService } from '@nexiom/piece-registry';
 import type { ConnectionValueBlob } from '../connectors.service.js';
 import { REDIS_CLIENT, type Redis } from '@nexiom/cache';
 import { CreateOAuthSession } from '../validation/create-oauth-session.js';

apps/api/src/modules/connections/connections/token-refresh.service.spec.ts

@@ -1,6 +1,6 @@
 import { DefaultOAuthRefreshClient } from './token-refresh.service.js';
 import { EncryptionService, OAuthRefreshError } from '@nexiom/credentials';
-import { PieceRegistryService } from '@nexiom/engine';
+import { PieceRegistryService } from '@nexiom/piece-registry';
 import type { Piece } from '@nexiom/piece-framework';
 import {
   describe,

apps/api/src/modules/connections/connections/token-refresh.service.ts

@@ -14,7 +14,7 @@ import {
 } from '@nexiom/database';
 import { eq, and, desc } from 'drizzle-orm';
 import type { ConnectionValueBlob } from '../connectors.service.js';
-import { PieceRegistryService } from '@nexiom/engine';
+import { PieceRegistryService } from '@nexiom/piece-registry';
 
 @Injectable()
 export class DefaultOAuthRefreshClient implements OAuthRefreshClient {

apps/api/src/modules/connections/connectors.service.spec.ts

@@ -3,7 +3,7 @@ import { DATABASE_CONNECTION } from '@nexiom/database';
 import { ConfigService } from '@nestjs/config';
 import { ConnectorsService } from './connectors.service.js';
 import { EncryptionService, AppCredentialError } from '@nexiom/credentials';
-import { PieceRegistryService } from '@nexiom/engine';
+import { PieceRegistryService } from '@nexiom/piece-registry';
 import type { Piece } from '@nexiom/piece-framework';
 import { DB_MANAGER } from '../dbmanager/dbmanager.module.js';
 import { SchemaPlan } from '@nexiom/dbmanager';

apps/api/src/modules/connections/connectors.service.ts

@@ -26,7 +26,7 @@ import { eq, and, or, sql } from 'drizzle-orm';
 import { SchemaPlan } from '@nexiom/dbmanager';
 import type { DatabaseManager } from '@nexiom/dbmanager';
 import { DB_MANAGER } from '../dbmanager/dbmanager.module.js';
-import { PieceRegistryService } from '@nexiom/engine';
+import { PieceRegistryService } from '@nexiom/piece-registry';
 import * as crypto from 'node:crypto';
 import { extractPgError, PG_UNIQUE_VIOLATION } from '../../shared/db.utils.js';