automatic stylefixes

Author: splitbrain

Chunk.php

@@ -22,11 +22,10 @@ public function __construct(
         protected $id,
         protected $text,
         protected $embedding,
-                  $lang = '',
-                  $created = '',
+        $lang = '',
+        $created = '',
         protected $score = 0
-    )
-    {
+    ) {
         $this->language = $lang ?: $this->determineLanguage();
         $this->created = $created ?: time();
     }

Embeddings.php

@@ -54,12 +54,11 @@ class Embeddings
      * @param array $config The plugin configuration
      */
     public function __construct(
-        ChatInterface      $chatModel,
+        ChatInterface $chatModel,
         EmbeddingInterface $embedModel,
-        AbstractStorage    $storage,
-                           $config
-    )
-    {
+        AbstractStorage $storage,
+        $config
+    ) {
         $this->chatModel = $chatModel;
         $this->embedModel = $embedModel;
         $this->storage = $storage;

Model/AbstractModel.php

@@ -93,7 +93,8 @@ public function getModelName()
      */
     public function resetUsageStats()
     {
-        $this->tokensUsed = 0;
+        $this->inputTokensUsed = 0;
+        $this->outputTokensUsed = 0;
         $this->timeUsed = 0;
         $this->requestsMade = 0;
     }

Model/Mistral/AbstractMistralModel.php

@@ -15,7 +15,7 @@ abstract class AbstractMistralModel extends AbstractModel
     public function __construct(string $name, array $config)
     {
         parent::__construct($name, $config);
-        if(empty($config['mistral_apikey'])) {
+        if (empty($config['mistral_apikey'])) {
             throw new \Exception('Mistral API key not configured');
         }
         $this->http->headers['Authorization'] = 'Bearer ' . $config['mistral_apikey'];

Model/ModelInterface.php

@@ -50,5 +50,4 @@ public function getMaxInputTokenLength(): int;
      * The price for 1,000,000 input tokens in USD
      */
     public function getInputTokenPrice(): float;
-
 }

Model/VoyageAI/EmbeddingModel.php

@@ -58,5 +58,4 @@ protected function parseAPIResponse($response)
 
         return $response;
     }
-
 }

ModelFactory.php

@@ -29,7 +29,7 @@ public function __construct(array $config)
      *
      * @param bool $debug
      */
-    public function setDebug(bool $debug=true)
+    public function setDebug(bool $debug = true)
     {
         $this->debug = $debug;
         $this->getChatModel()->setDebug($debug);
@@ -155,5 +155,4 @@ public function loadModel(string $type, string $name)
 
         return $instance;
     }
-
 }

ModelSetting.php

@@ -2,11 +2,10 @@
 
 namespace dokuwiki\plugin\aichat;
 
-
 use dokuwiki\plugin\config\core\Setting\SettingMultichoice;
 
-class ModelSetting extends SettingMultichoice {
-
+class ModelSetting extends SettingMultichoice
+{
     /** @inheritdoc */
     public function __construct($key, $params = null)
     {
@@ -17,7 +16,7 @@ public function __construct($key, $params = null)
         $jsons = glob(__DIR__ . '/Model/*/models.json');
         foreach ($jsons as $json) {
             $models = json_decode(file_get_contents($json), true);
-            if(!isset($models[$type])) continue;
+            if (!isset($models[$type])) continue;
 
             $namespace = basename(dirname($json));
             foreach (array_keys($models[$type]) as $model) {

Storage/PineconeStorage.php

@@ -54,7 +54,7 @@ protected function runQuery($endpoint, mixed $data, $method = 'POST')
         }
 
         try {
-        $result = json_decode((string)$response, true, 512, JSON_THROW_ON_ERROR);
+            $result = json_decode((string)$response, true, 512, JSON_THROW_ON_ERROR);
         } catch (\JsonException $e) {
             throw new \Exception('Pinecone API returned invalid JSON. ' . $response, 0, $e);
         }

cli/simulate.php

@@ -1,5 +1,6 @@
 <?php
 
+use dokuwiki\Extension\CLIPlugin;
 use dokuwiki\plugin\aichat\ModelFactory;
 use splitbrain\phpcli\Colors;
 use splitbrain\phpcli\Options;
@@ -10,7 +11,7 @@
  * @license GPL 2 http://www.gnu.org/licenses/gpl-2.0.html
  * @author  Andreas Gohr <gohr@cosmocode.de>
  */
-class cli_plugin_aichat_simulate extends \dokuwiki\Extension\CLIPlugin
+class cli_plugin_aichat_simulate extends CLIPlugin
 {
     /** @var helper_plugin_aichat */
     protected $helper;
@@ -50,7 +51,7 @@ protected function main(Options $options)
         [$input, $output] = $options->getArgs();
         $questions = $this->readInputFile($input);
         $outfh = @fopen($output, 'w');
-        if(!$outfh) throw new \Exception("Could not open $output for writing");
+        if (!$outfh) throw new \Exception("Could not open $output for writing");
 
         $models = $this->helper->factory->getModels(true, 'chat');
 
@@ -92,7 +93,7 @@ protected function simulate($questions, $model)
                 'question' => $q,
                 'rephrased' => $result['question'],
                 'answer' => $result['answer'],
-                'source.list' => join("\n", $result['sources']),
+                'source.list' => implode("\n", $result['sources']),
                 'source.time' => $this->helper->getEmbeddings()->timeSpent,
                 ...$this->flattenStats('stats.embedding', $this->helper->getEmbeddingModel()->getUsageStats()),
                 ...$this->flattenStats('stats.rephrase', $this->helper->getRephraseModel()->getUsageStats()),
@@ -132,13 +133,13 @@ protected function records2rows(array $result): array
         $rows[] = $row;
 
         // write rows
-        for($i=0; $i<$numberOfRecords; $i++) {
-            foreach($rowkeys as $type => $keys) {
+        for ($i = 0; $i < $numberOfRecords; $i++) {
+            foreach ($rowkeys as $type => $keys) {
                 $row = [];
                 $row[] = $type;
-                foreach($models as $model) {
+                foreach ($models as $model) {
                     foreach ($keys as $key) {
-                        if($key) {
+                        if ($key) {
                             $row[] = $result[$model][$i][$key];
                         } else {
                             $row[] = '';
@@ -161,9 +162,10 @@ protected function records2rows(array $result): array
      * @param array $stats
      * @return array
      */
-    protected function flattenStats(string $prefix, array $stats) {
+    protected function flattenStats(string $prefix, array $stats)
+    {
         $result = [];
-        foreach($stats as $key => $value) {
+        foreach ($stats as $key => $value) {
             $result["$prefix.$key"] = $value;
         }
         return $result;
@@ -183,4 +185,3 @@ protected function readInputFile(string $file): array
         return $questions;
     }
 }
-