Added support for reka.ai

Author: splitbrain

Model/Anthropic/ChatModel.php

@@ -51,7 +51,7 @@ public function getAnswer(array $messages): string
     }
 
     /**
-     * Send a request to the OpenAI API
+     * Send a request to the API
      *
      * @param string $endpoint
      * @param array $data Payload to send

Model/Reka/ChatModel.php

@@ -0,0 +1,83 @@
+<?php
+
+namespace dokuwiki\plugin\aichat\Model\Reka;
+
+use dokuwiki\plugin\aichat\Model\AbstractModel;
+use dokuwiki\plugin\aichat\Model\ChatInterface;
+
+class ChatModel extends AbstractModel implements ChatInterface
+{
+    /** @inheritdoc */
+    public function __construct(string $name, array $config)
+    {
+        parent::__construct($name, $config);
+
+        if (empty($config['reka_apikey'])) {
+            throw new \Exception('Reka API key not configured');
+        }
+
+        $this->http->headers['x-api-key'] = $config['reka_apikey'];
+    }
+
+    /** @inheritdoc */
+    public function getAnswer(array $messages): string
+    {
+        $chat = [];
+        foreach ($messages as $message) {
+            if ($message['role'] === 'user') {
+                $chat[] = [
+                    'type' => 'human',
+                    'text' => $message['content'],
+                ];
+            } elseif ($message['role'] === 'assistant') {
+                $chat[] = [
+                    'type' => 'model',
+                    'text' => $message['content'],
+                ];
+            }
+            // system messages are not supported
+        }
+
+        $data = [
+            'conversation_history' => $chat,
+            'model_name' => $this->getModelName(),
+            'temperature' => 0.0,
+        ];
+
+        $response = $this->request('chat', $data);
+        return $response['text'];
+    }
+
+    /**
+     * Send a request to the API
+     *
+     * @param string $endpoint
+     * @param array $data Payload to send
+     * @return array API response
+     * @throws \Exception
+     */
+    protected function request($endpoint, $data)
+    {
+        $url = 'https://api.reka.ai/' . $endpoint;
+        return $this->sendAPIRequest('POST', $url, $data);
+    }
+
+    /** @inheritdoc */
+    protected function parseAPIResponse($response)
+    {
+        if(((int) $this->http->status) !== 200) {
+            if(isset($response['detail'])) {
+                throw new \Exception('Reka API error: ' . $response['detail']);
+            } else {
+                throw new \Exception('Reka API error: ' . $this->http->status . ' ' . $this->http->error);
+            }
+        }
+
+        if (isset($response['metadata'])) {
+            $this->inputTokensUsed += $response['metadata']['input_tokens'];
+            $this->outputTokensUsed += $response['metadata']['generated_tokens'];
+        }
+
+        return $response;
+    }
+}

Model/Reka/models.json

@@ -0,0 +1,25 @@
+{
+    "chat": {
+        "reka-flash": {
+            "description": "Fast and cost-efficient model for most tasks",
+            "inputTokens": 128000,
+            "inputTokenPrice": 0.8,
+            "outputTokens": 8192,
+            "outputTokenPrice": 2.0
+        },
+        "reka-edge": {
+            "description": "Lightweight model for local (i.e., on-hardware) or latency sensitive applications",
+            "inputTokens": 64000,
+            "inputTokenPrice": 0.4,
+            "outputTokens": 8192,
+            "outputTokenPrice": 1.0
+        },
+        "reka-core": {
+            "description": "Superior capabilities for complex tasks",
+            "inputTokens": 128000,
+            "inputTokenPrice": 10.0,
+            "outputTokens": 8192,
+            "outputTokenPrice": 25.0
+        }
+    }
+}

conf/default.php

@@ -20,6 +20,8 @@
 
 $conf['voyageai_apikey'] = '';
 
+$conf['reka_apikey'] = '';
+
 $conf['pinecone_apikey'] = '';
 $conf['pinecone_baseurl'] = '';
 

conf/metadata.php

@@ -27,6 +27,8 @@
 
 $meta['voyageai_apikey'] = array('password');
 
+$meta['reka_apikey'] = array('password');
+
 $meta['pinecone_apikey'] = array('password');
 $meta['pinecone_baseurl'] = array('string');
 

lang/en/settings.php

@@ -17,6 +17,7 @@
 $lang['anthropic_apikey'] = '🧠 <b>Anthropic</b> API key';
 $lang['mistral_apikey'] = '🧠 <b>Mistral</b> API key';
 $lang['voyageai_apikey'] = '🧠 <b>Voyage AI</b> API key';
+$lang['reka_apikey'] = '🧠 <b>Reka</b> API key';
 
 $lang['pinecone_apikey'] = '📥 <b>Pinecone</b> API key';
 $lang['pinecone_baseurl'] = '📥 <b>Pinecone</b> base URL';