Refactored gemini, openai and claude into one file and removed logic from the base class, also made helper class definable from command line

2025-06-03 13:04:42 -05:00
parent bd0ce3e340
commit f15228fa58
36 changed files with 487 additions and 3847 deletions
@@ -0,0 +1,24 @@
+# models_config.yaml
+
+GEMINI:
+  api_key_env: GEMINI_API_KEY
+  base_url: https://generativelanguage.googleapis.com/v1beta
+  supports_switching: true
+  switch_options:
+    small:
+      name: gemini-pro
+      max_tokens: 2048
+    large:
+      name: gemini-1.5-pro-latest
+      max_tokens: 8192
+OPENAI:
+  api_key_env: OPENAI_API_KEY
+  base_url: null # Indicates to use the default OpenAI API base URL
+  supports_switching: true
+  switch_options:
+    small:
+      name: gpt-3.5-turbo
+      max_tokens: null
+    large:
+      name: gpt-4
+      max_tokens: null