push

src/controllers/chatController.ts

@@ -2,6 +2,7 @@ import { Request, Response } from 'express';
 import { AppDataSource } from '../config/database';
 import { Agent, AgentStatus } from '../entities/Agent';
 import { Subscription, SubscriptionStatus } from '../entities/Subscription';
+import { Transaction, TransactionType, TransactionStatus } from '../entities/Transaction';
 import { ChatProtocolService } from '../services/chatProtocol';
 import { WalletService } from '../services/walletService';
 
@@ -39,17 +40,27 @@ export class ChatController {
         });
       }
 
-      // Get agent
+      const user = (req as any).user;
+      const isAdmin = user?.isAdmin || false;
+
+      // Get agent - admins can test pending/rejected agents, regular users only approved
       const agent = await agentRepository.findOne({
-        where: { id: agentId, status: AgentStatus.APPROVED },
+        where: { 
+          id: agentId,
+          ...(isAdmin ? {} : { status: AgentStatus.APPROVED })
+        },
       });
 
       if (!agent) {
-        return res.status(404).json({ error: 'Agent not found or not approved' });
+        return res.status(404).json({ error: 'Agent not found' });
       }
 
+      // Admins don't pay for unapproved agents (pending/rejected) - they're testing quality
+      const isAdminTestingUnapproved = isAdmin && agent.status !== AgentStatus.APPROVED;
+      const shouldCharge = userId && !isAdminTestingUnapproved;
+
       // Check wallet balance and charge points (first 2 tasks free)
-      if (userId) {
+      if (shouldCharge) {
         const chargeResult = await walletService.chargeForTask(userId, agentId, agent);
         
         if (!chargeResult.success) {
@@ -69,6 +80,27 @@ export class ChatController {
             },
           });
         }
+      } else if (isAdminTestingUnapproved) {
+        // Create a free transaction for admin testing (for tracking purposes)
+        const wallet = await walletService.getOrCreateWallet(userId);
+        const transactionRepository = AppDataSource.getRepository(Transaction);
+        const transaction = transactionRepository.create({
+          walletId: wallet.id,
+          userId,
+          agentId,
+          type: TransactionType.ADMIN_TEST,
+          status: TransactionStatus.COMPLETED,
+          amount: 0,
+          balanceAfter: Number(wallet.balance),
+          description: `Admin test of ${agent.name} (${agent.status}) - no charge`,
+          metadata: { 
+            agentId, 
+            agentName: agent.name,
+            agentStatus: agent.status,
+            isAdminTest: true 
+          },
+        });
+        await transactionRepository.save(transaction);
       }
 
       // Convert multer files to chat protocol format
@@ -105,15 +137,20 @@ export class ChatController {
     try {
       const { id: agentId } = req.params;
       const { conversationId, limit } = req.query;
-      const userId = (req as any).user?.id;
+      const user = (req as any).user;
+      const userId = user?.id;
+      const isAdmin = user?.isAdmin || false;
 
       if (!userId) {
         return res.status(401).json({ error: 'Authentication required' });
       }
 
-      // Verify agent exists and is approved
+      // Admins can view history for any agent, regular users only approved agents
       const agent = await agentRepository.findOne({
-        where: { id: agentId, status: AgentStatus.APPROVED },
+        where: { 
+          id: agentId,
+          ...(isAdmin ? {} : { status: AgentStatus.APPROVED })
+        },
       });
 
       if (!agent) {

src/entities/Transaction.ts

@@ -15,6 +15,7 @@ export enum TransactionType {
   CHARGE = 'charge', // Charged for agent task
   REFUND = 'refund', // Refunded points
   FREE = 'free', // Free task
+  ADMIN_TEST = 'admin_test', // Admin testing unapproved agents (no charge)
 }
 
 export enum TransactionStatus {