Add user config ability to save and fetch queries from db

joshembling · Dec 3, 2023 · a60ec9c · a60ec9c
1 parent 2adee3c
commit a60ec9c
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 10 deletions.
diff --git a/config/laragenie.php b/config/laragenie.php
@@ -7,6 +7,11 @@
         'instructions' => 'Write only in markdown format. Only write factual data that can be pulled from indexed chunks.',
     ],
 
+    'database' => [
+        'fetch' => true,
+        'save' => true,
+    ],
+
     'openai' => [
         'embedding' => [
             'model' => 'text-embedding-ada-002',

diff --git a/src/Commands/LaragenieCommand.php b/src/Commands/LaragenieCommand.php
@@ -62,11 +62,13 @@ public function userQuestion(OpenAI\Client $openai, Pinecone $pinecone)
 
         $formattedQuestion = $ai ? Str::remove('--ai', $question) : $question;
 
-        $laragenie = LaragenieModel::firstOrNew([
-            'question' => $formattedQuestion,
-        ]);
+        if (config('laragenie.database.fetch') || config('laragenie.database.save')) {
+            $laragenie = LaragenieModel::firstOrNew([
+                'question' => $formattedQuestion,
+            ]);
+        }
 
-        if ($laragenie->exists && ! $ai) {
+        if ($laragenie->exists && config('laragenie.database.fetch') && ! $ai) {
             $this->info($laragenie->answer);
         } else {
             $this->question('Asking '.config('laragenie.bot.name').", '{$question}'...");
@@ -80,13 +82,15 @@ public function userQuestion(OpenAI\Client $openai, Pinecone $pinecone)
                 $tokens = $botResponse->usage->totalTokens;
                 $calculatedCost = $this->calculateCost($tokens);
 
-                $laragenie->fill([
-                    'answer' => $answer,
-                    'cost' => $calculatedCost,
-                    'vectors' => $questionResponse['vectors'],
-                ]);
+                if (config('laragenie.database.save')) {
+                    $laragenie->fill([
+                        'answer' => $answer,
+                        'cost' => $calculatedCost,
+                        'vectors' => $questionResponse['vectors'],
+                    ]);
 
-                $laragenie->save();
+                    $laragenie->save();
+                }
 
                 $this->info($answer);
                 $this->costResponse($calculatedCost);