io.on('connection', (socket) => {
  console.log('✅ Client connected');
  socket.on('chat-message', async (message: Message) => {
  
  // 1. Formalize input message, create run ids

  // 2. Emit RUN_STARTED event

  // 3. Call openai chat stream mode
  
  // 4. For every chunk reponses send the right event
  
  // 4-a. If chunk has content text, emit TEXT_MESSAGE_CHUNK event
  
  // 4-b. If chunk has toolCalls, emit TOOL_CALL_CHUNK event
  
  // 5. Emit RUN_FINISHED event
  
  // 6. On error emit RUN_ERROR event
  });

  socket.on('disconnect', () => {
    console.log('❌ Client disconnected');
  });
});

io.on('connection', (socket) => {
  console.log('✅ Client connected');
  socket.on('chat-message', async (message: Message) => {
    console.log('📩 User:', message);
    
    // 1. Formalize input message, create run ids
    const input: RunAgentInput = {
      threadId: generateUUID(),
      runId: generateUUID(),
      messages: [
        {
          content: message.content,
          role: message.role,
          id: message.messageId,
        },
      ],
      tools: [...],
      context: [],
    };

    // 2. Emit RUN_STARTED event
    socket.emit('event', {
      type: EventType.RUN_STARTED,
      threadId: input.threadId,
      runId: input.runId,
    });

    try {
      // 3. Call openai chat stream mode
      const completions = await openai.chat.completions.create({
        model: 'gpt-4o',
        tools: input.tools.map((tool) => ({
          type: 'function',
          function: {
            name: tool.name,
            description: tool.description,
            parameters: tool.parameters,
          },
        })),
        messages: input.messages.map((msg) => ({
          role: msg.role as 'user' | 'system' | 'assistant' | 'developer',
          content: (msg.content as string) ?? '',
        })),
        stream: true,
      });

      const messageId = generateUUID();
      let toolCallId: string | undefined = undefined;

      for await (const chunk of completions) {
        // 4. For every chunk reponses send the right event
        const content = chunk.choices[0]?.delta?.content;
        const toolCalls = chunk.choices[0]?.delta?.tool_calls;

        if (content) {
          // 4-a. If chunk has content text, emit TEXT_MESSAGE_CHUNK event
          socket.emit('event', {
            type: EventType.TEXT_MESSAGE_CHUNK,
            messageId,
            delta: chunk.choices[0]?.delta?.content || '',
          });
        } else if (toolCalls) {
          // 4-b. If chunk has toolCalls, emit TOOL_CALL_CHUNK event
          const toolCall = toolCalls[0];
          if (toolCall?.id) toolCallId = toolCall.id;
          socket.emit('event', {
            type: EventType.TOOL_CALL_CHUNK,
            parentMessageId: messageId,
            toolCallId,
            toolCallName: toolCall?.function?.name,
            delta: toolCall.function?.arguments,
          } as ToolCallChunkEvent);
        }
      }

      // 5. ✅ Emit RUN_FINISHED event
      socket.emit('event', {
        type: EventType.RUN_FINISHED,
        threadId: input.threadId,
        runId: input.runId,
      });
    } catch (error: any) {
      // 6. ❌ On error emit RUN_ERROR event
      console.error('❌ OpenAI Error:', error);
      socket.emit('event', {
        type: EventType.RUN_ERROR,
        message: error?.message,
      });
    }
  });

  socket.on('disconnect', () => {
    console.log('❌ Client disconnected');
  });
});