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')
  }
}

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

@Sse('stream')
async stream(@Query() messageDto: SseMessageDto): Promise<Observable<SseMessage>> {
  return await this.agentService.stream(messageDto)
}

@Get('history/:threadId')
async getHistory(@Param('threadId') threadId: string): Promise<MessageResponseDto[]> {
  return await this.agentService.getHistory(threadId)
}

@Injectable()
export class RedisService {
  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 https://github.com/IBJunior/agentailor-chat-ui.git
cd agentailor-chat-ui
pnpm install
pnpm dev