feat: allow any string as model (#367)

* refactor(anthropic.ts, openAi.ts): remove model validation against predefined list to allow any string as model, simplifying configuration and improving flexibility
di-sukharev · Aug 18, 2024 · c425878 · c425878
1 parent 10b031a
commit c425878
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 14 deletions.
diff --git a/src/engine/anthropic.ts b/src/engine/anthropic.ts
@@ -48,15 +48,13 @@ if (
 
 const MODEL = config?.OCO_MODEL;
 if (provider === 'anthropic' &&
-    !MODEL_LIST.anthropic.includes(MODEL) &&
-    command !== 'config' &&
-    mode !== CONFIG_MODES.set) {
-  outro(
-    `${chalk.red('✖')} Unsupported model ${MODEL} for Anthropic. Supported models are: ${MODEL_LIST.anthropic.join(
-      ', '
-    )}`
-  );
-  process.exit(1);
+  MODEL.typeof !== 'string' && 
+  command !== 'config' &&
+  mode !== CONFIG_MODES.set) {
+outro(
+  `${chalk.red('✖')} Unsupported model ${MODEL}. The model can be any string, but the current configuration is not supported.`
+);
+process.exit(1);
 }
 
 export class AnthropicAi implements AiEngine {

diff --git a/src/engine/openAi.ts b/src/engine/openAi.ts
@@ -54,15 +54,12 @@ if (
 
 const MODEL = config?.OCO_MODEL || 'gpt-3.5-turbo';
 if (provider === 'openai' &&
-    !MODEL_LIST.openai.includes(MODEL) &&
+  MODEL.typeof !== 'string' && 
     command !== 'config' &&
     mode !== CONFIG_MODES.set) {
   outro(
-    `${chalk.red('✖')} Unsupported model ${MODEL} for OpenAI. Supported models are: ${MODEL_LIST.openai.join(
-      ', '
-    )}`
+    `${chalk.red('✖')} Unsupported model ${MODEL}. The model can be any string, but the current configuration is not supported.`
   );
-
   process.exit(1);
 }