query-agent/examples/nestjs-example.ts

An AI-powered database query agent that integrates with existing Express apps using Socket.IO and HTTP routes

import { NestFactory } from '@nestjs/core';
import { createServer } from 'http';
import { Server } from 'socket.io';
import queryAgent from 'query-agent';
import { QueryAgentOptions } from 'query-agent';

// Example NestJS app module (you would have your own)
class AppModule {}

async function bootstrap() {
  const app = await NestFactory.create(AppModule);
  const server = createServer(app.getHttpAdapter().getInstance());
  
  // Create Socket.IO server
  const io = new Server(server, {
    cors: {
      origin: "*",
      methods: ["GET", "POST"]
    }
  });

  // Define your SQL execution function
  const executeSQLQuery = async (query: string): Promise<any> => {
    // Your database query execution logic here
    console.log('Executing SQL query:', query);
    // Example: return await yourDatabase.execute(query);
    return { result: 'query executed successfully' };
  };

  // Initialize Query Agent
  const queryAgentOptions: QueryAgentOptions = {
    executeSQLQuery,
    corsOptions: {
      origin: "*",
      methods: ["GET", "POST"]
    }
  };

  const { io: queryAgentIo, queryAgentNamespace } = queryAgent(io, queryAgentOptions);

  await app.listen(3000);
  console.log('NestJS app with Query Agent running on port 3000');
}

bootstrap();