Fixed so it checks if the key is set correctly

dxpr · Aug 22, 2024 · 12c1794 · 12c1794
1 parent 4057ba0
commit 12c1794
Showing 1 changed file with 23 additions and 18 deletions.
diff --git a/dxpr_cms_installer/src/Form/ConfigureAPIKeysForm.php b/dxpr_cms_installer/src/Form/ConfigureAPIKeysForm.php
@@ -200,24 +200,29 @@ public function submitForm(array &$form, FormStateInterface $form_state) {
 
     // If the AI provider is set, enable the appropriate modules.
     if ($ai_provider = $form_state->getValue('ai_provider')) {
-      // Setup the key for the provider.
-      $key_id = $ai_provider . '_key';
-      $key = Key::create([
-        'id' => $key_id,
-        'label' => ucfirst($ai_provider) . ' API Key',
-        'description' => 'API Key for ' . ucfirst($ai_provider),
-        'key_type' => 'authentication',
-        'key_provider' => 'config',
-      ]);
-      $key->setKeyValue($form_state->getValue($key_id));
-      $key->save();
-      // Add the key to the config.
-      $this->configFactory->getEditable('provider_' . $ai_provider . '.settings')->set('api_key', $key_id)->save();
-      // Set the default provider.
-      $this->configFactory->getEditable('ai.settings')->set('default_providers.chat', [
-        'provider_id' => $ai_provider,
-        'model_id' => $ai_provider == 'openai' ? 'gpt-4o' : $this->getFirstAiModelId($ai_provider),
-      ])->save();
+      try {
+        // Setup the key for the provider.
+        $key_id = $ai_provider . '_key';
+        $key = Key::create([
+          'id' => $key_id,
+          'label' => ucfirst($ai_provider) . ' API Key',
+          'description' => 'API Key for ' . ucfirst($ai_provider),
+          'key_type' => 'authentication',
+          'key_provider' => 'config',
+        ]);
+        $key->setKeyValue($form_state->getValue($key_id));
+        $key->save();
+        // Add the key to the config.
+        $this->configFactory->getEditable('provider_' . $ai_provider . '.settings')->set('api_key', $key_id)->save();
+        // Set the default provider.
+        $this->configFactory->getEditable('ai.settings')->set('default_providers.chat', [
+          'provider_id' => $ai_provider,
+          'model_id' => $ai_provider == 'openai' ? 'gpt-4o' : $this->getFirstAiModelId($ai_provider),
+        ])->save();
+      }
+      catch (\Exception $e) {
+        $this->messenger()->addError($this->t('An error occurred while saving the AI provider key: @error', ['@error' => $e->getMessage()]));
+      }
     }
   }