import { AiEngine } from '../engine/Engine';
import { api } from '../engine/openAi';
import { getConfig } from '../commands/config';
import { ollamaAi } from '../engine/ollama';
import { testAi } from '../engine/testAi';

export function getEngine(): AiEngine {
  const config = getConfig();
  if (config?.OCO_AI_PROVIDER == 'ollama') {
    return ollamaAi;
  } else if (config?.OCO_AI_PROVIDER == 'test') {
    return testAi;
  }
  //open ai gpt by default
  return api;
}