export class ReactAgentBuilder {
  private readonly stateGraph: StateGraph<typeof MessagesAnnotation>;
  
  constructor(tools: any[], llm: BaseChatModel) {
    this.stateGraph = new StateGraph(MessagesAnnotation);
    this.initializeGraph();
  }
  
  private initializeGraph(): void {
    this.stateGraph
      .addNode('agent', this.callModel.bind(this))
      .addNode('tools', this.toolNode)
      .addEdge(START, 'agent')
      .addConditionalEdges('agent', this.shouldContinue.bind(this), ['tools', END])
      .addEdge('tools', 'agent');
  }
}

export function createPostgresMemory(): PostgresSaver {
  const connectionString = `postgresql://${username}:${password}@${host}:${port}/${dbName}`;
  return PostgresSaver.fromConnString(connectionString);
}

export class AgentFactory {
  public static createAgent(
    modelProvider: ModelProvider,
    tools: any[],
    checkpointer?: PostgresSaver
  ): CompiledStateGraph<typeof MessagesAnnotation, any> {
    switch (modelProvider) {
      case ModelProvider.OPENAI:
        return new ReactAgentBuilder(
          tools,
          new ChatOpenAI({ model: process.env.OPENAI_MODEL })
        ).build(checkpointer);
      // Có thể bổ sung provider khác ở đây
    }
  }
}

@Post('chat')
async chat(@Body() messageDto: MessageDto): Promise<MessageResponseDto> {
  return await this.agentService.chat(messageDto);
}

@Injectable()
export class RedisService {
  private client: RedisClient;

  constructor() {
    this.client = createClient({
      url: `redis://${process.env.REDIS_HOST}:${process.env.REDIS_PORT}`,
    });
  }

  async publish(channel: string, message: any): Promise<void> {
    await this.client.publish(channel, JSON.stringify(message));
  }
}

const response = await fetch('http://localhost:3001/api/agent/chat', {
  method: 'POST',
  headers: { 'Content-Type': 'application/json' },
  body: JSON.stringify({
    threadId: 'unique-thread-id',
    content: [{ type: 'text', text: 'Hello, AI!' }],
    type: 'human',
  }),
});

const sse = new EventSource('http://localhost:3001/api/agent/stream?threadId=unique-thread-id&content=Hello');
sse.onmessage = (event) => {
  const data = JSON.parse(event.data);
  console.log(data.content);
};

git clone <your-repo>
cd <your-repo>
pnpm install
docker compose up -d
pnpm run start:dev

pnpm test
pnpm test:e2e

git clone https://github.com/IBJunior/agentailor-chat-ui.git
cd agentailor-chat-ui
pnpm install
pnpm dev