import { ReasoningChatModel } from '../reasoningChatModel';
import { ChatOpenAI } from '@langchain/openai';
import logger from '../../utils/logger';
import { getDeepseekApiKey } from '../../config';
import axios from 'axios';

interface DeepSeekModel {
  id: string;
  object: string;
  owned_by: string;
}

interface ModelListResponse {
  object: 'list';
  data: DeepSeekModel[];
}

interface ChatModelConfig {
  displayName: string;
  model: ReasoningChatModel | ChatOpenAI;
}

// Define which models require reasoning capabilities
const REASONING_MODELS = ['deepseek-reasoner'];

const MODEL_DISPLAY_NAMES: Record<string, string> = {
  'deepseek-reasoner': 'DeepSeek R1',
  'deepseek-chat': 'DeepSeek V3'
};

export const loadDeepSeekChatModels = async (): Promise<Record<string, ChatModelConfig>> => {
  const deepSeekEndpoint = 'https://api.deepseek.com'; 

  const apiKey = getDeepseekApiKey();
  if (!apiKey) return {}; 

  if (!deepSeekEndpoint || !apiKey) {
    logger.debug('DeepSeek endpoint or API key not configured, skipping');
    return {};
  }

  try {
    const response = await axios.get<{ data: DeepSeekModel[] }>(`${deepSeekEndpoint}/models`, {
      headers: {
        'Content-Type': 'application/json',
        'Authorization': `Bearer ${apiKey}`,
      },
    });

    const deepSeekModels = response.data.data;

    const chatModels = deepSeekModels.reduce<Record<string, ChatModelConfig>>((acc, model) => {
      // Only include models we have display names for
      if (model.id in MODEL_DISPLAY_NAMES) {
        // Use ReasoningChatModel for models that need reasoning capabilities
        if (REASONING_MODELS.includes(model.id)) {
          acc[model.id] = {
            displayName: MODEL_DISPLAY_NAMES[model.id],
            model: new ReasoningChatModel({
              apiKey,
              baseURL: deepSeekEndpoint,
              modelName: model.id,
              temperature: 0.7,
              streamDelay: 50, // Add a small delay to control streaming speed
            }),
          };
        } else {
          // Use standard ChatOpenAI for other models
          acc[model.id] = {
            displayName: MODEL_DISPLAY_NAMES[model.id],
            model: new ChatOpenAI({
              openAIApiKey: apiKey,
              configuration: {
                baseURL: deepSeekEndpoint,
              },
              modelName: model.id,
              temperature: 0.7,
            }),
          };
        }
      }
      return acc;
    }, {});

    return chatModels;
  } catch (err) {
    logger.error(`Error loading DeepSeek models: ${String(err)}`);
    return {};
  }
};