refactor keys to config

PlasticStudio · Oct 8, 2024 · b06f46b · b06f46b
1 parent 0ee8209
commit b06f46b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/src/Extensions/SeoAICMSPageEditControllerExtension.php b/src/Extensions/SeoAICMSPageEditControllerExtension.php
@@ -4,6 +4,7 @@
 
 use voku\helper\HtmlDomParser;
 use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Config\Config;
 use SilverStripe\SiteConfig\SiteConfig;
 
 class SeoAICMSPageEditControllerExtension extends DataExtension
@@ -88,11 +89,12 @@ public function generatePrompt()
      */
     public function promptAPICall($prompt)
     {
-        $key = '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';
+        $openai = Config::inst()->get("openai");
+        $key = $openai->key;
         $url = 'https://api.openai.com/v1/chat/completions';
         $data = [
-            "model" => "gpt-4o-mini",
-            "temperature" => 0,
+            "model" => $openai->model,
+            "temperature" => $openai->temperature,
             "messages" => [
                 [
                     "role" => "user",
@@ -162,4 +164,4 @@ public function populateMetaTagsFromAPI($response)
 
         return false;
     }
-}
+}
diff --git a/src/Extensions/SeoAISiteConfigExtension.php b/src/Extensions/SeoAISiteConfigExtension.php
@@ -14,6 +14,11 @@ class SeoAISiteConfigExtension extends DataExtension
 
     public function updateCMSFields(FieldList $fields)
     {
-        $fields->addFieldToTab('Root.SEO', TextareaField::create('ContextPrompt', 'Brand Context Prompt')->setDescription('Additional information to give AI about your brand / content for more accurate metadata generation'), 'UseTitleAsMetaTitle');
+        $fields->addFieldToTab(
+            'Root.SEO',
+            TextareaField::create('ContextPrompt', 'Brand Context Prompt')
+                ->setDescription('Additional information to give AI about your brand / content for more accurate metadata generation'),
+            'UseTitleAsMetaTitle'
+        );
     }
-}
+}