@nextsparkjs/plugin-ai 0.1.0-beta.74 → 0.1.0-beta.76

package/api/generate/route.ts

@@ -80,20 +80,25 @@ const postHandler = async (request: NextRequest) => {
 
     // 8. Save example if requested (opt-in)
     if (saveExample) {
+      const teamId = request.headers.get('x-team-id') || ''
       await saveExampleSafely(
         {
           prompt,
           response: result.text,
           model: selectedModel.modelName,
           status: 'completed',
+          operation: 'generate',
+          provider: selectedModel.provider,
           metadata: {
             tokens: tokens.total,
+            tokensInput: tokens.input,
+            tokensOutput: tokens.output,
             cost,
-            provider: selectedModel.provider,
             isLocal: selectedModel.isLocal
           }
         },
-        authResult.user!.id
+        authResult.user!.id,
+        teamId
       )
     }
 

package/entities/ai-history/ai-history.config.ts

@@ -17,7 +17,8 @@ export const aiHistoryEntityConfig: EntityConfig = {
   // ==========================================
   // 1. BASIC IDENTIFICATION
   // ==========================================
-  slug: 'ai-history',    // Single source of truth - derives tableName, apiPath, metaTableName, i18nNamespace
+  slug: 'ai-history',
+  tableName: 'ai_history',  // Explicit table name (underscores, not hyphens)
   enabled: true,
   names: {
     singular: 'AI History',

package/entities/ai-history/migrations/001_ai_history_table.sql

@@ -1,7 +1,7 @@
 -- =============================================
 -- AI HISTORY ENTITY MIGRATION - PLUGIN ENTITY
 -- Plugin: AI
--- Entity: ai-history
+-- Entity: ai_history
 -- Generic AI interaction history for all plugins
 -- Date: 2025-10-03
 -- =============================================
@@ -10,11 +10,12 @@
 -- MAIN AI_HISTORY TABLE
 -- =============================================
 
-DROP TABLE IF EXISTS "ai-history" CASCADE;
+DROP TABLE IF EXISTS "ai_history" CASCADE;
 
-CREATE TABLE "ai-history" (
+CREATE TABLE "ai_history" (
   id TEXT PRIMARY KEY DEFAULT gen_random_uuid()::text,
   "userId" TEXT NOT NULL REFERENCES "users"(id) ON DELETE CASCADE,
+  "teamId" TEXT NOT NULL REFERENCES "teams"(id) ON DELETE CASCADE,
 
   -- Generic polymorphic relationship (flexible for any future use case)
   "relatedEntityType" TEXT,
@@ -49,46 +50,48 @@ CREATE TABLE "ai-history" (
 -- INDEXES
 -- =============================================
 
-CREATE INDEX IF NOT EXISTS idx_ai-history_user_id ON "ai-history"("userId");
-CREATE INDEX IF NOT EXISTS idx_ai-history_entity ON "ai-history"("relatedEntityType", "relatedEntityId") WHERE "relatedEntityType" IS NOT NULL;
-CREATE INDEX IF NOT EXISTS idx_ai-history_status ON "ai-history"(status);
-CREATE INDEX IF NOT EXISTS idx_ai-history_operation ON "ai-history"(operation);
-CREATE INDEX IF NOT EXISTS idx_ai-history_model ON "ai-history"(model);
-CREATE INDEX IF NOT EXISTS idx_ai-history_provider ON "ai-history"(provider);
-CREATE INDEX IF NOT EXISTS idx_ai-history_created_at ON "ai-history"("createdAt" DESC);
-CREATE INDEX IF NOT EXISTS idx_ai-history_user_status ON "ai-history"("userId", status);
-CREATE INDEX IF NOT EXISTS idx_ai-history_user_operation ON "ai-history"("userId", operation);
-CREATE INDEX IF NOT EXISTS idx_ai-history_tokens_input ON "ai-history"("tokensInput") WHERE "tokensInput" IS NOT NULL;
-CREATE INDEX IF NOT EXISTS idx_ai-history_tokens_output ON "ai-history"("tokensOutput") WHERE "tokensOutput" IS NOT NULL;
+CREATE INDEX IF NOT EXISTS idx_ai_history_user_id ON "ai_history"("userId");
+CREATE INDEX IF NOT EXISTS idx_ai_history_team_id ON "ai_history"("teamId");
+CREATE INDEX IF NOT EXISTS idx_ai_history_team_user ON "ai_history"("teamId", "userId");
+CREATE INDEX IF NOT EXISTS idx_ai_history_entity ON "ai_history"("relatedEntityType", "relatedEntityId") WHERE "relatedEntityType" IS NOT NULL;
+CREATE INDEX IF NOT EXISTS idx_ai_history_status ON "ai_history"(status);
+CREATE INDEX IF NOT EXISTS idx_ai_history_operation ON "ai_history"(operation);
+CREATE INDEX IF NOT EXISTS idx_ai_history_model ON "ai_history"(model);
+CREATE INDEX IF NOT EXISTS idx_ai_history_provider ON "ai_history"(provider);
+CREATE INDEX IF NOT EXISTS idx_ai_history_created_at ON "ai_history"("createdAt" DESC);
+CREATE INDEX IF NOT EXISTS idx_ai_history_user_status ON "ai_history"("userId", status);
+CREATE INDEX IF NOT EXISTS idx_ai_history_user_operation ON "ai_history"("userId", operation);
+CREATE INDEX IF NOT EXISTS idx_ai_history_tokens_input ON "ai_history"("tokensInput") WHERE "tokensInput" IS NOT NULL;
+CREATE INDEX IF NOT EXISTS idx_ai_history_tokens_output ON "ai_history"("tokensOutput") WHERE "tokensOutput" IS NOT NULL;
 
 -- =============================================
 -- ROW LEVEL SECURITY (enableRLS: true)
 -- =============================================
 
 -- Enable RLS
-ALTER TABLE "ai-history" ENABLE ROW LEVEL SECURITY;
+ALTER TABLE "ai_history" ENABLE ROW LEVEL SECURITY;
 
--- Policies for ai-history table
-DROP POLICY IF EXISTS "Users can view own ai history" ON "ai-history";
-CREATE POLICY "Users can view own ai history" ON "ai-history"
+-- Policies for ai_history table
+DROP POLICY IF EXISTS "Users can view own ai history" ON "ai_history";
+CREATE POLICY "Users can view own ai history" ON "ai_history"
   FOR SELECT USING ("userId" = public.get_auth_user_id());
 
-DROP POLICY IF EXISTS "Users can insert own ai history" ON "ai-history";
-CREATE POLICY "Users can insert own ai history" ON "ai-history"
+DROP POLICY IF EXISTS "Users can insert own ai history" ON "ai_history";
+CREATE POLICY "Users can insert own ai history" ON "ai_history"
   FOR INSERT WITH CHECK ("userId" = public.get_auth_user_id());
 
-DROP POLICY IF EXISTS "Users can update own ai history" ON "ai-history";
-CREATE POLICY "Users can update own ai history" ON "ai-history"
+DROP POLICY IF EXISTS "Users can update own ai history" ON "ai_history";
+CREATE POLICY "Users can update own ai history" ON "ai_history"
   FOR UPDATE USING ("userId" = public.get_auth_user_id())
   WITH CHECK ("userId" = public.get_auth_user_id());
 
-DROP POLICY IF EXISTS "Users can delete own ai history" ON "ai-history";
-CREATE POLICY "Users can delete own ai history" ON "ai-history"
+DROP POLICY IF EXISTS "Users can delete own ai history" ON "ai_history";
+CREATE POLICY "Users can delete own ai history" ON "ai_history"
   FOR DELETE USING ("userId" = public.get_auth_user_id());
 
 -- Admins can manage all ai history
-DROP POLICY IF EXISTS "Admins can manage all ai history" ON "ai-history";
-CREATE POLICY "Admins can manage all ai history" ON "ai-history"
+DROP POLICY IF EXISTS "Admins can manage all ai history" ON "ai_history";
+CREATE POLICY "Admins can manage all ai history" ON "ai_history"
   FOR ALL USING (
     EXISTS (
       SELECT 1 FROM "users"
@@ -107,14 +110,14 @@ CREATE POLICY "Admins can manage all ai history" ON "ai-history"
 -- =============================================
 -- TRIGGER: Auto-update updatedAt
 -- =============================================
-CREATE TRIGGER ai-history_set_updated_at
-  BEFORE UPDATE ON "ai-history"
+CREATE TRIGGER ai_history_set_updated_at
+  BEFORE UPDATE ON "ai_history"
   FOR EACH ROW EXECUTE FUNCTION public.set_updated_at();
 
 -- =============================================
 -- TRIGGER: Auto-update completedAt
 -- =============================================
-CREATE OR REPLACE FUNCTION update_ai-history_completed_at()
+CREATE OR REPLACE FUNCTION update_ai_history_completed_at()
 RETURNS TRIGGER AS $$
 BEGIN
   IF NEW.status IN ('completed', 'failed') AND OLD.status NOT IN ('completed', 'failed') THEN
@@ -124,16 +127,16 @@ BEGIN
 END;
 $$ LANGUAGE plpgsql;
 
-CREATE TRIGGER ai-history_completed_at
-  BEFORE UPDATE ON "ai-history"
+CREATE TRIGGER ai_history_completed_at
+  BEFORE UPDATE ON "ai_history"
   FOR EACH ROW
-  EXECUTE FUNCTION update_ai-history_completed_at();
+  EXECUTE FUNCTION update_ai_history_completed_at();
 
 -- =============================================
 -- CONSTRAINT: Token Split Validation
 -- =============================================
 -- Ensures data integrity: tokensUsed = tokensInput + tokensOutput when both present
-ALTER TABLE "ai-history"
+ALTER TABLE "ai_history"
 ADD CONSTRAINT chk_tokens_sum
 CHECK (
   ("tokensInput" IS NULL AND "tokensOutput" IS NULL)
@@ -144,24 +147,24 @@ CHECK (
 -- =============================================
 -- COMMENTS
 -- =============================================
-COMMENT ON TABLE "ai-history" IS 'Generic audit trail for AI interactions with cost and performance metrics';
-COMMENT ON COLUMN "ai-history"."userId" IS 'User who initiated the AI operation';
-COMMENT ON COLUMN "ai-history"."relatedEntityType" IS 'Type of entity this operation relates to (e.g., ''contents'', ''products'', ''campaigns'')';
-COMMENT ON COLUMN "ai-history"."relatedEntityId" IS 'ID of the related entity (polymorphic relationship)';
-COMMENT ON COLUMN "ai-history".operation IS 'Type of AI operation: generate, refine, analyze, chat, completion, or other';
-COMMENT ON COLUMN "ai-history".model IS 'AI model used (e.g., claude-3-5-sonnet-20241022, gpt-4)';
-COMMENT ON COLUMN "ai-history".provider IS 'AI provider: anthropic, openai, etc.';
-COMMENT ON COLUMN "ai-history".status IS 'Processing status: pending, processing, completed, or failed';
-COMMENT ON COLUMN "ai-history"."tokensUsed" IS 'Total tokens consumed (input + output) - kept for backward compatibility';
-COMMENT ON COLUMN "ai-history"."tokensInput" IS 'Input tokens (prompt) - for precise cost calculation with asymmetric pricing';
-COMMENT ON COLUMN "ai-history"."tokensOutput" IS 'Output tokens (completion) - for precise cost calculation with asymmetric pricing';
-COMMENT ON COLUMN "ai-history"."creditsUsed" IS 'Credits deducted for this operation';
-COMMENT ON COLUMN "ai-history"."estimatedCost" IS 'Cost in USD based on model pricing';
-COMMENT ON COLUMN "ai-history"."balanceAfter" IS 'User credit balance after this operation';
-COMMENT ON COLUMN "ai-history"."errorMessage" IS 'Error message if operation failed';
-COMMENT ON COLUMN "ai-history"."createdAt" IS 'When the AI operation was initiated';
-COMMENT ON COLUMN "ai-history"."updatedAt" IS 'When the record was last updated';
-COMMENT ON COLUMN "ai-history"."completedAt" IS 'When the AI operation finished (success or failure)';
+COMMENT ON TABLE "ai_history" IS 'Generic audit trail for AI interactions with cost and performance metrics';
+COMMENT ON COLUMN "ai_history"."userId" IS 'User who initiated the AI operation';
+COMMENT ON COLUMN "ai_history"."relatedEntityType" IS 'Type of entity this operation relates to (e.g., ''contents'', ''products'', ''campaigns'')';
+COMMENT ON COLUMN "ai_history"."relatedEntityId" IS 'ID of the related entity (polymorphic relationship)';
+COMMENT ON COLUMN "ai_history".operation IS 'Type of AI operation: generate, refine, analyze, chat, completion, or other';
+COMMENT ON COLUMN "ai_history".model IS 'AI model used (e.g., claude-3-5-sonnet-20241022, gpt-4)';
+COMMENT ON COLUMN "ai_history".provider IS 'AI provider: anthropic, openai, etc.';
+COMMENT ON COLUMN "ai_history".status IS 'Processing status: pending, processing, completed, or failed';
+COMMENT ON COLUMN "ai_history"."tokensUsed" IS 'Total tokens consumed (input + output) - kept for backward compatibility';
+COMMENT ON COLUMN "ai_history"."tokensInput" IS 'Input tokens (prompt) - for precise cost calculation with asymmetric pricing';
+COMMENT ON COLUMN "ai_history"."tokensOutput" IS 'Output tokens (completion) - for precise cost calculation with asymmetric pricing';
+COMMENT ON COLUMN "ai_history"."creditsUsed" IS 'Credits deducted for this operation';
+COMMENT ON COLUMN "ai_history"."estimatedCost" IS 'Cost in USD based on model pricing';
+COMMENT ON COLUMN "ai_history"."balanceAfter" IS 'User credit balance after this operation';
+COMMENT ON COLUMN "ai_history"."errorMessage" IS 'Error message if operation failed';
+COMMENT ON COLUMN "ai_history"."createdAt" IS 'When the AI operation was initiated';
+COMMENT ON COLUMN "ai_history"."updatedAt" IS 'When the record was last updated';
+COMMENT ON COLUMN "ai_history"."completedAt" IS 'When the AI operation finished (success or failure)';
 
 -- =============================================
 -- MIGRATION COMPLETE
@@ -171,6 +174,6 @@ DO $$
 BEGIN
   RAISE NOTICE 'AI History entity migration completed successfully!';
   RAISE NOTICE 'Plugin: ai';
-  RAISE NOTICE 'Table created: ai-history';
+  RAISE NOTICE 'Table created: ai_history';
   RAISE NOTICE 'RLS policies enabled for user isolation';
 END $$;

package/entities/ai-history/migrations/002_ai_history_metas.sql

@@ -1,10 +1,10 @@
 -- =============================================
 -- AI HISTORY METADATA SYSTEM
 -- Plugin: AI
--- Entity: ai-history
+-- Entity: ai_history
 -- =============================================
 --
--- This migration adds comprehensive metadata capabilities through ai-history_metas table
+-- This migration adds comprehensive metadata capabilities through ai_history_metas table
 --
 -- Purpose: Flexible metadata storage for AI operations
 -- - Better query performance (indexed key-value lookups)
@@ -29,9 +29,9 @@
 -- AI_HISTORY_METAS TABLE
 -- =============================================
 
-CREATE TABLE IF NOT EXISTS public."ai-history_metas" (
+CREATE TABLE IF NOT EXISTS public."ai_history_metas" (
   id             TEXT PRIMARY KEY DEFAULT gen_random_uuid()::text,
-  "entityId"     TEXT NOT NULL REFERENCES public."ai-history"(id) ON DELETE CASCADE,
+  "entityId"     TEXT NOT NULL REFERENCES public."ai_history"(id) ON DELETE CASCADE,
   "metaKey"      TEXT NOT NULL,
   "metaValue"    JSONB NOT NULL DEFAULT '{}'::jsonb,
   "dataType"     TEXT DEFAULT 'json' CHECK ("dataType" IN ('string', 'number', 'boolean', 'json', 'array')),
@@ -41,54 +41,54 @@ CREATE TABLE IF NOT EXISTS public."ai-history_metas" (
   "updatedAt"    TIMESTAMPTZ NOT NULL DEFAULT now(),
 
   -- Prevent duplicate keys per history record
-  CONSTRAINT ai-history_metas_unique_key UNIQUE ("entityId", "metaKey")
+  CONSTRAINT ai_history_metas_unique_key UNIQUE ("entityId", "metaKey")
 );
 
 -- Trigger for auto-updating updatedAt (using Better Auth's function)
-CREATE TRIGGER ai-history_metas_set_updated_at
-BEFORE UPDATE ON public."ai-history_metas"
+CREATE TRIGGER ai_history_metas_set_updated_at
+BEFORE UPDATE ON public."ai_history_metas"
 FOR EACH ROW EXECUTE FUNCTION public.set_updated_at();
 
 -- Indexes for efficient queries
-CREATE INDEX idx_ai-history_metas_entity_id ON public."ai-history_metas"("entityId");
-CREATE INDEX idx_ai-history_metas_key ON public."ai-history_metas"("metaKey");
-CREATE INDEX idx_ai-history_metas_composite ON public."ai-history_metas"("entityId", "metaKey");
+CREATE INDEX idx_ai_history_metas_entity_id ON public."ai_history_metas"("entityId");
+CREATE INDEX idx_ai_history_metas_key ON public."ai_history_metas"("metaKey");
+CREATE INDEX idx_ai_history_metas_composite ON public."ai_history_metas"("entityId", "metaKey");
 
 -- GIN index for JSONB queries (enables @>, ?, ?&, ?| operators)
-CREATE INDEX idx_ai-history_metas_value_gin ON public."ai-history_metas" USING GIN ("metaValue");
+CREATE INDEX idx_ai_history_metas_value_gin ON public."ai_history_metas" USING GIN ("metaValue");
 
 -- RLS Policies
-ALTER TABLE public."ai-history_metas" ENABLE ROW LEVEL SECURITY;
+ALTER TABLE public."ai_history_metas" ENABLE ROW LEVEL SECURITY;
 
 -- Cleanup existing policies
-DROP POLICY IF EXISTS "AI history metas owner can do all" ON public."ai-history_metas";
-DROP POLICY IF EXISTS "AI history metas inherit parent access" ON public."ai-history_metas";
+DROP POLICY IF EXISTS "AI history metas owner can do all" ON public."ai_history_metas";
+DROP POLICY IF EXISTS "AI history metas inherit parent access" ON public."ai_history_metas";
 
--- Policy: Inherit access from parent ai-history record
+-- Policy: Inherit access from parent ai_history record
 CREATE POLICY "AI history metas inherit parent access"
-ON public."ai-history_metas"
+ON public."ai_history_metas"
 FOR ALL TO authenticated
 USING (
   EXISTS (
-    SELECT 1 FROM public."ai-history"
-    WHERE "ai-history".id = "ai-history_metas"."entityId"
-    AND "ai-history"."userId" = public.get_auth_user_id()
+    SELECT 1 FROM public."ai_history"
+    WHERE "ai_history".id = "ai_history_metas"."entityId"
+    AND "ai_history"."userId" = public.get_auth_user_id()
   )
 )
 WITH CHECK (
   EXISTS (
-    SELECT 1 FROM public."ai-history"
-    WHERE "ai-history".id = "ai-history_metas"."entityId"
-    AND "ai-history"."userId" = public.get_auth_user_id()
+    SELECT 1 FROM public."ai_history"
+    WHERE "ai_history".id = "ai_history_metas"."entityId"
+    AND "ai_history"."userId" = public.get_auth_user_id()
   )
 );
 
 -- Comments for documentation
-COMMENT ON TABLE public."ai-history_metas" IS 'Flexible metadata storage for AI history records. Supports any key-value pairs with JSONB values.';
-COMMENT ON COLUMN public."ai-history_metas"."metaValue" IS 'JSONB value allows storing any data structure (string, number, boolean, object, array)';
-COMMENT ON COLUMN public."ai-history_metas"."dataType" IS 'Hint for parsing metaValue on client side (actual type is always JSONB in DB)';
-COMMENT ON COLUMN public."ai-history_metas"."isPublic" IS 'Whether this meta is publicly accessible or private to owner';
-COMMENT ON COLUMN public."ai-history_metas"."isSearchable" IS 'Whether this meta should be included in search operations';
+COMMENT ON TABLE public."ai_history_metas" IS 'Flexible metadata storage for AI history records. Supports any key-value pairs with JSONB values.';
+COMMENT ON COLUMN public."ai_history_metas"."metaValue" IS 'JSONB value allows storing any data structure (string, number, boolean, object, array)';
+COMMENT ON COLUMN public."ai_history_metas"."dataType" IS 'Hint for parsing metaValue on client side (actual type is always JSONB in DB)';
+COMMENT ON COLUMN public."ai_history_metas"."isPublic" IS 'Whether this meta is publicly accessible or private to owner';
+COMMENT ON COLUMN public."ai_history_metas"."isSearchable" IS 'Whether this meta should be included in search operations';
 
 -- =============================================
 -- MIGRATION COMPLETE
@@ -97,7 +97,7 @@ COMMENT ON COLUMN public."ai-history_metas"."isSearchable" IS 'Whether this meta
 DO $$
 BEGIN
   RAISE NOTICE 'AI History metadata migration completed successfully!';
-  RAISE NOTICE 'Created: ai-history_metas table for flexible metadata';
+  RAISE NOTICE 'Created: ai_history_metas table for flexible metadata';
   RAISE NOTICE 'Performance: GIN indexes for fast JSONB queries';
-  RAISE NOTICE 'Security: RLS policies enabled on ai-history_metas';
+  RAISE NOTICE 'Security: RLS policies enabled on ai_history_metas';
 END $$;

package/lib/save-example.ts

@@ -1,28 +1,31 @@
 /**
- * AI Example Saving - PostgreSQL Implementation
+ * AI History Saving - PostgreSQL Implementation
  *
  * Uses project's PostgreSQL connection with RLS support.
- * Self-contained plugin with direct DB access.
+ * Saves AI interactions to the ai_history table.
  */
 
 import { mutateWithRLS, queryWithRLS } from '@nextsparkjs/core/lib/db'
 import { sanitizePrompt, sanitizeResponse } from './sanitize'
 
 export interface AIExampleData {
-  title?: string
   prompt: string
   response: string
   model: string
   status: 'pending' | 'processing' | 'completed' | 'failed'
-  userId?: string
+  operation?: 'generate' | 'refine' | 'analyze' | 'chat' | 'completion' | 'other'
+  provider?: string
   metadata?: {
     tokens?: number
+    tokensInput?: number
+    tokensOutput?: number
     cost?: number
     duration?: number
-    provider?: string
     isLocal?: boolean
     platforms?: number
     imagesProcessed?: number
+    relatedEntityType?: string
+    relatedEntityId?: string
   }
 }
 
@@ -34,11 +37,12 @@ export interface SaveExampleResult {
 }
 
 /**
- * Save AI example to PostgreSQL database
+ * Save AI interaction to ai_history table
  */
 export async function saveAIExample(
   data: AIExampleData,
-  userId: string
+  userId: string,
+  teamId: string
 ): Promise<SaveExampleResult> {
   try {
     // 1. Sanitize sensitive data
@@ -49,25 +53,35 @@ export async function saveAIExample(
       sanitizedPrompt !== data.prompt ||
       sanitizedResponse !== data.response
 
-    // 2. Generate title if not provided
-    const title = data.title || generateTitle(sanitizedPrompt)
+    // 2. Extract token info
+    const tokensInput = data.metadata?.tokensInput || 0
+    const tokensOutput = data.metadata?.tokensOutput || 0
+    const tokensUsed = data.metadata?.tokens || (tokensInput + tokensOutput)
 
-    // 3. Insert into database with RLS
+    // 3. Insert into ai_history table with RLS
     const query = `
-      INSERT INTO ai_example (
-        title, prompt, response, model, status, "userId", metadata, "createdAt", "updatedAt"
-      ) VALUES ($1, $2, $3, $4, $5, $6, $7, NOW(), NOW())
+      INSERT INTO ai_history (
+        "userId", "teamId", operation, model, provider, status,
+        "tokensUsed", "tokensInput", "tokensOutput", "estimatedCost",
+        "relatedEntityType", "relatedEntityId",
+        "createdAt", "updatedAt"
+      ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, NOW(), NOW())
       RETURNING id
     `
 
     const params = [
-      title,
-      sanitizedPrompt,
-      sanitizedResponse,
+      userId,
+      teamId,
+      data.operation || 'generate',
       data.model,
+      data.provider || 'ollama',
       data.status,
-      userId,
-      JSON.stringify(data.metadata || {})
+      tokensUsed || null,
+      tokensInput || null,
+      tokensOutput || null,
+      data.metadata?.cost || null,
+      data.metadata?.relatedEntityType || null,
+      data.metadata?.relatedEntityId || null
     ]
 
     const result = await mutateWithRLS<{ id: string }>(query, params, userId)
@@ -75,13 +89,14 @@ export async function saveAIExample(
     if (result.rowCount === 0 || !result.rows[0]) {
       return {
         success: false,
-        error: 'Failed to insert example'
+        error: 'Failed to insert ai_history record'
       }
     }
 
-    console.log('[AI Plugin] Example saved to database:', {
+    console.log('[AI Plugin] History saved to database:', {
       id: result.rows[0].id,
-      title,
+      operation: data.operation || 'generate',
+      model: data.model,
       sanitizationApplied
     })
 
@@ -92,7 +107,7 @@ export async function saveAIExample(
     }
 
   } catch (error) {
-    console.error('[AI Plugin] Save example failed:', error)
+    console.error('[AI Plugin] Save history failed:', error)
     return {
       success: false,
       error: error instanceof Error ? error.message : 'Unknown error'
@@ -101,29 +116,44 @@ export async function saveAIExample(
 }
 
 /**
- * Get AI examples from PostgreSQL
+ * Get AI history from PostgreSQL
  */
 export async function getAIExamples(params: {
   userId?: string
+  teamId?: string
   limit?: number
   offset?: number
   status?: string
+  operation?: string
 }) {
   try {
-    const { userId, limit = 50, offset = 0, status } = params
+    const { userId, teamId, limit = 50, offset = 0, status, operation } = params
 
     let query = `
-      SELECT id, title, prompt, response, model, status, "userId", metadata, "createdAt", "updatedAt"
-      FROM ai_example
+      SELECT id, "userId", "teamId", operation, model, provider, status,
+             "tokensUsed", "tokensInput", "tokensOutput", "estimatedCost",
+             "relatedEntityType", "relatedEntityId",
+             "createdAt", "updatedAt", "completedAt"
+      FROM ai_history
       WHERE 1=1
     `
     const queryParams: (string | number)[] = []
 
+    if (teamId) {
+      queryParams.push(teamId)
+      query += ` AND "teamId" = $${queryParams.length}`
+    }
+
     if (status) {
       queryParams.push(status)
       query += ` AND status = $${queryParams.length}`
     }
 
+    if (operation) {
+      queryParams.push(operation)
+      query += ` AND operation = $${queryParams.length}`
+    }
+
     query += ` ORDER BY "createdAt" DESC LIMIT $${queryParams.length + 1} OFFSET $${queryParams.length + 2}`
     queryParams.push(limit, offset)
 
@@ -134,7 +164,7 @@ export async function getAIExamples(params: {
       data: rows
     }
   } catch (error) {
-    console.error('[AI Plugin] Get examples failed:', error)
+    console.error('[AI Plugin] Get history failed:', error)
     return {
       success: false,
       error: error instanceof Error ? error.message : 'Unknown error',
@@ -144,13 +174,16 @@ export async function getAIExamples(params: {
 }
 
 /**
- * Get single AI example by ID from PostgreSQL
+ * Get single AI history record by ID
  */
 export async function getAIExampleById(id: string, userId?: string) {
   try {
     const query = `
-      SELECT id, title, prompt, response, model, status, "userId", metadata, "createdAt", "updatedAt"
-      FROM ai_example
+      SELECT id, "userId", "teamId", operation, model, provider, status,
+             "tokensUsed", "tokensInput", "tokensOutput", "estimatedCost",
+             "relatedEntityType", "relatedEntityId",
+             "createdAt", "updatedAt", "completedAt"
+      FROM ai_history
       WHERE id = $1
     `
 
@@ -159,7 +192,7 @@ export async function getAIExampleById(id: string, userId?: string) {
     if (rows.length === 0) {
       return {
         success: false,
-        error: 'Example not found'
+        error: 'History record not found'
       }
     }
 
@@ -168,7 +201,7 @@ export async function getAIExampleById(id: string, userId?: string) {
       data: rows[0]
     }
   } catch (error) {
-    console.error('[AI Plugin] Get example by ID failed:', error)
+    console.error('[AI Plugin] Get history by ID failed:', error)
     return {
       success: false,
       error: error instanceof Error ? error.message : 'Unknown error'
@@ -177,18 +210,18 @@ export async function getAIExampleById(id: string, userId?: string) {
 }
 
 /**
- * Delete AI example from PostgreSQL
+ * Delete AI history record
  */
 export async function deleteAIExample(id: string, userId: string) {
   try {
-    const query = `DELETE FROM ai_example WHERE id = $1 RETURNING id`
+    const query = `DELETE FROM ai_history WHERE id = $1 RETURNING id`
 
     const result = await mutateWithRLS<{ id: string }>(query, [id], userId)
 
     if (result.rowCount === 0) {
       return {
         success: false,
-        error: 'Example not found or unauthorized'
+        error: 'History record not found or unauthorized'
       }
     }
 
@@ -197,7 +230,7 @@ export async function deleteAIExample(id: string, userId: string) {
       id: result.rows[0].id
     }
   } catch (error) {
-    console.error('[AI Plugin] Delete example failed:', error)
+    console.error('[AI Plugin] Delete history failed:', error)
     return {
       success: false,
       error: error instanceof Error ? error.message : 'Unknown error'
@@ -206,32 +239,23 @@ export async function deleteAIExample(id: string, userId: string) {
 }
 
 /**
- * Generate title from prompt (helper)
- */
-function generateTitle(prompt: string): string {
-  const cleaned = prompt.replace(/\n/g, ' ').trim()
-  return cleaned.length > 50
-    ? cleaned.substring(0, 50) + '...'
-    : cleaned
-}
-
-/**
- * Save example with automatic error handling (fire-and-forget)
+ * Save history with automatic error handling (fire-and-forget)
  */
 export async function saveExampleSafely(
   data: AIExampleData,
-  userId: string
+  userId: string,
+  teamId: string
 ): Promise<void> {
   try {
-    const result = await saveAIExample(data, userId)
+    const result = await saveAIExample(data, userId, teamId)
 
     if (!result.success) {
-      console.error('[AI Plugin] Example save failed:', result.error)
+      console.error('[AI Plugin] History save failed:', result.error)
     } else if (result.sanitizationApplied) {
-      console.warn('[AI Plugin] Sanitization was applied to saved example')
+      console.warn('[AI Plugin] Sanitization was applied to saved history')
     }
   } catch (error) {
     // Silent fail - don't break the main request
-    console.error('[AI Plugin] Example save error:', error)
+    console.error('[AI Plugin] History save error:', error)
   }
-}
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nextsparkjs/plugin-ai",
-  "version": "0.1.0-beta.74",
+  "version": "0.1.0-beta.76",
   "private": false,
   "main": "./plugin.config.ts",
   "requiredPlugins": [],