Merge pull request #7 from arnebr/main

Allow to customize OpenAI Model
spatie · Jul 22, 2024 · a014da1 · a014da1
2 parents f139ef7 + 1677ae7
commit a014da1
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/src/SolutionProviders/Laravel/OpenAiSolutionProvider.php b/src/SolutionProviders/Laravel/OpenAiSolutionProvider.php
@@ -31,6 +31,7 @@ public function getSolutions(Throwable $throwable): array
             cacheTtlInSeconds: 60,
             applicationType: 'Laravel ' . Str::before(app()->version(), '.'),
             applicationPath: base_path(),
+            openAiModel: config('error-solutions.open_ai_model', 'gpt-3.5-turbo'),
         );
 
         return $solutionProvider->getSolutions($throwable);

diff --git a/src/Solutions/OpenAi/OpenAiSolution.php b/src/Solutions/OpenAi/OpenAiSolution.php
@@ -28,6 +28,7 @@ public function __construct(
         protected int|null            $cacheTtlInSeconds = 60,
         protected string|null         $applicationType = null,
         protected string|null         $applicationPath = null,
+        protected string|null         $openAiModel = null,
     ) {
         $this->prompt = $this->generatePrompt();
 
@@ -99,7 +100,7 @@ protected function generatePrompt(): string
 
     protected function getModel(): string
     {
-        return 'gpt-3.5-turbo';
+        return $this->openAiModel ?? 'gpt-3.5-turbo';
     }
 
     protected function getApplicationFrame(Throwable $throwable): ?Frame

diff --git a/src/Solutions/OpenAi/OpenAiSolutionProvider.php b/src/Solutions/OpenAi/OpenAiSolutionProvider.php
@@ -9,11 +9,12 @@
 class OpenAiSolutionProvider implements HasSolutionsForThrowable
 {
     public function __construct(
-        protected string $openAiKey,
-        protected ?CacheInterface $cache = null,
-        protected int $cacheTtlInSeconds = 60 * 60,
-        protected string|null $applicationType = null,
-        protected string|null $applicationPath = null,
+        protected string            $openAiKey,
+        protected ?CacheInterface   $cache = null,
+        protected int               $cacheTtlInSeconds = 60 * 60,
+        protected string|null       $applicationType = null,
+        protected string|null       $applicationPath = null,
+        protected string            $openAiModel = 'gpt-3.5-turbo',
     ) {
         $this->cache ??= new DummyCache();
     }
@@ -33,6 +34,7 @@ public function getSolutions(Throwable $throwable): array
                 $this->cacheTtlInSeconds,
                 $this->applicationType,
                 $this->applicationPath,
+                $this->openAiModel
             ),
         ];
     }