fix generator

Signed-off-by: Matteo Collina <[email protected]>
platformatic · May 10, 2024 · 5b018f2 · 5b018f2
1 parent dbfe084
commit 5b018f2
Showing 1 changed file with 25 additions and 1 deletion.
diff --git a/lib/generator.ts b/lib/generator.ts
@@ -49,6 +49,30 @@ class AiWarpGenerator extends ServiceGenerator {
         default: 'gpt-3.5-turbo',
         type: 'string',
         configValue: 'aiModel'
+      },
+      {
+        var: 'PLT_OPENAPI_API_KEY',
+        label: 'What is your OpenAI API key?',
+        default: '',
+        type: 'string'
+      },
+      {
+        var: 'PLT_MISTRAL_API_KEY',
+        label: 'What is your Mistral API key?',
+        default: '',
+        type: 'string'
+      },
+      {
+        var: 'PLT_AZURE_API_KEY',
+        label: 'What is your AZURE API key?',
+        default: '',
+        type: 'string'
+      },
+      {
+        var: 'PLT_LLAMA2_MODEL',
+        label: 'Where is the llama2 model located?',
+        default: '',
+        type: 'string'
       }
     ]
   }
@@ -102,7 +126,7 @@ class AiWarpGenerator extends ServiceGenerator {
       case 'llama2':
         config.aiProvider = {
           llama2: {
-            modelPath: this.config.aiModel
+            modelPath: `{${this.getEnvVarName('PLT_LLAMA2_MODEL')}}`
           }
         }
         break