Method handleChat

server/handlers/websocket.go:224–434

handleChat handles chat messages

(wsConn *WebSocketConnection, payload map[string]any)

// handleChat handles chat messages
func (h *WebSocketHandler) handleChat(wsConn *WebSocketConnection, payload map[string]any) {
	// Reset read deadline to prevent timeout during long-running LLM requests
	_ = wsConn.Conn.SetReadDeadline(time.Now().Add(5 * time.Minute))

	// Extract parameters
	templateID, _ := payload["template_id"].(string)
	if templateID == "" {
		templateID = "chat"
	}

	input, _ := payload["input"].(string)
	if input == "" {
		h.sendError(wsConn, "missing_input", "Input message is required")
		return
	}

	var ag *agent.Agent
	var err error

	// Reuse existing agent if available, otherwise create new one
	if wsConn.Agent != nil {
		ag = wsConn.Agent
		logging.Info(wsConn.ctx, "websocket.agent.reused", map[string]any{
			"agent_id": ag.ID(),
		})
	} else {
		// Create agent configuration
		cfg := &types.AgentConfig{
			TemplateID: templateID,
		}

		// Handle model config if provided
		if modelConfigData, ok := payload["model_config"].(map[string]any); ok {
			modelConfig := &types.ModelConfig{}
			if provider, ok := modelConfigData["provider"].(string); ok {
				modelConfig.Provider = provider
			}
			if model, ok := modelConfigData["model"].(string); ok {
				modelConfig.Model = model
			}
			if apiKey, ok := modelConfigData["api_key"].(string); ok {
				modelConfig.APIKey = apiKey
			}

			// Fill API key from environment if missing
			if modelConfig.APIKey == "" && modelConfig.Provider != "" {
				var envKey string
				switch modelConfig.Provider {
				case "deepseek":
					envKey = os.Getenv("DEEPSEEK_API_KEY")
				case "anthropic":
					envKey = os.Getenv("ANTHROPIC_API_KEY")
				case "openai":
					envKey = os.Getenv("OPENAI_API_KEY")
				default:
					envKey = os.Getenv(strings.ToUpper(modelConfig.Provider) + "_API_KEY")
				}
				if envKey != "" {