Added additional codellama models

Added support for codellama:python codellama:code codellama:7b-python codellama:7b-code codellama:13b-python codellama:13b-code codellama:34b-code codellama:34b-python
jorge-menjivar · Dec 15, 2023 · a153b19 · a153b19
1 parent f89ffb0
commit a153b19
Showing 1 changed file with 64 additions and 0 deletions.
diff --git a/apps/unsaged/types/ai-models.ts b/apps/unsaged/types/ai-models.ts
@@ -174,6 +174,38 @@ export const PossibleAiModels: PossibleAiModelsInterface = {
     vendor: 'Ollama',
     type: 'text',
   },
+  'codellama:python': {
+    id: 'codellama:python',
+    maxLength: 32000,
+    tokenLimit: 4096,
+    requestLimit: 3000,
+    vendor: 'Ollama',
+    type: 'text',
+  },
+  'codellama:code': {
+    id: 'codellama:code',
+    maxLength: 32000,
+    tokenLimit: 4096,
+    requestLimit: 3000,
+    vendor: 'Ollama',
+    type: 'text',
+  },
+  'codellama:7b-python': {
+    id: 'codellama:7b-python',
+    maxLength: 32000,
+    tokenLimit: 4096,
+    requestLimit: 3000,
+    vendor: 'Ollama',
+    type: 'text',
+  },
+  'codellama:7b-code': {
+    id: 'codellama:7b-code',
+    maxLength: 32000,
+    tokenLimit: 4096,
+    requestLimit: 3000,
+    vendor: 'Ollama',
+    type: 'text',
+  },
   'codellama:7b': {
     id: 'codellama:7b',
     maxLength: 32000,
@@ -182,6 +214,22 @@ export const PossibleAiModels: PossibleAiModelsInterface = {
     vendor: 'Ollama',
     type: 'text',
   },
+  'codellama:13b-python': {
+    id: 'codellama:13b-python',
+    maxLength: 32000,
+    tokenLimit: 4096,
+    requestLimit: 3000,
+    vendor: 'Ollama',
+    type: 'text',
+  },
+  'codellama:13b-code': {
+    id: 'codellama:13b-code',
+    maxLength: 32000,
+    tokenLimit: 4096,
+    requestLimit: 3000,
+    vendor: 'Ollama',
+    type: 'text',
+  },
   'codellama:13b': {
     id: 'codellama:13b',
     maxLength: 32000,
@@ -198,6 +246,22 @@ export const PossibleAiModels: PossibleAiModelsInterface = {
     vendor: 'Ollama',
     type: 'text',
   },
+  'codellama:34b-code': {
+    id: 'codellama:34b-code',
+    maxLength: 32000,
+    tokenLimit: 4096,
+    requestLimit: 3000,
+    vendor: 'Ollama',
+    type: 'text',
+  },
+  'codellama:34b-python': {
+    id: 'codellama:34b-python',
+    maxLength: 32000,
+    tokenLimit: 4096,
+    requestLimit: 3000,
+    vendor: 'Ollama',
+    type: 'text',
+  },
   'wizardcoder:latest': {
     id: 'wizardcoder:latest',
     maxLength: 32000,