small adjustments

Author: splitbrain

Embeddings.php

@@ -237,11 +237,11 @@ public function getSimilarChunks($query, $lang = '')
 
         $time = microtime(true);
         $chunks = $this->storage->getSimilarChunks($vector, $lang, $fetch);
-        $this->timeSpent = microtime(true) - $time;
+        $this->timeSpent = round(microtime(true) - $time, 2);
         if ($this->logger instanceof CLI) {
             $this->logger->info(
                 'Fetched {count} similar chunks from store in {time} seconds',
-                ['count' => count($chunks), 'time' => round($this->timeSpent, 2)]
+                ['count' => count($chunks), 'time' => $this->timeSpent]
             );
         }
 

Model/Anthropic/ChatModel.php

@@ -23,7 +23,7 @@ public function __construct(string $name, array $config)
     /** @inheritdoc */
     public function getAnswer(array $messages): string
     {
-        // convert OpenAI Style to Anthropic style
+        // system message is separate from the messages array
         $system = '';
         $chat = [];
         foreach ($messages as $message) {
@@ -47,9 +47,6 @@ public function getAnswer(array $messages): string
         }
 
         $response = $this->request('messages', $data);
-
-        print_r($response);
-
         return $response['content'][0]['text'];
     }
 
@@ -71,7 +68,8 @@ protected function request($endpoint, $data)
     protected function parseAPIResponse($response)
     {
         if (isset($response['usage'])) {
-            $this->tokensUsed += $response['usage']['input_tokens'] + $response['usage']['output_tokens'];
+            $this->inputTokensUsed += $response['usage']['input_tokens'];
+            $this->outputTokensUsed += $response['usage']['output_tokens'];
         }
 
         if (isset($response['error'])) {