ai.class.php

Go to the documentation of this file.

<?php
/* Copyright (C) 2024  Laurent Destailleur     <eldy@users.sourceforge.net>
 * Copyright (C) 2024       Frédéric France             <frederic.france@free.fr>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <https://www.gnu.org/licenses/>.
 * or see https://www.gnu.org/
 */
 
require_once DOL_DOCUMENT_ROOT."/core/lib/admin.lib.php";
require_once DOL_DOCUMENT_ROOT.'/core/lib/geturl.lib.php';
 
 
class Ai
{
  protected $db;
 
  private $apiService;
 
  private $apiKey;
 
  private $apiEndpoint;
 
 
  public function __construct($db)
  {
    $this->db = $db;
 
    // Get API key according to enabled AI
    $this->apiService = getDolGlobalString('AI_API_SERVICE', 'chatgpt');
    $this->apiKey = getDolGlobalString('AI_API_'.strtoupper($this->apiService).'_KEY');
  }
 
  public function generateContent($instructions, $model = 'auto', $function = 'textgeneration', $format = '')
  {
    if (empty($this->apiKey)) {
      return array('error' => true, 'message' => 'API key is not defined for the AI enabled service '.$this->apiService);
    }
 
    if (empty($this->apiEndpoint)) {
      if ($function == 'imagegeneration') {
        if ($this->apiService == 'chatgpt') {
          $this->apiEndpoint = 'https://api.openai.com/v1/images/generations';
          if ($model == 'auto') {
            $model = getDolGlobalString('AI_API_CHATGPT_MODEL_IMAGE', 'dall-e-3');
          }
        }
      } elseif ($function == 'audiotext') {
        if ($this->apiService == 'chatgpt') {
          $this->apiEndpoint = 'https://api.openai.com/v1/audio/speech';
          if ($model == 'auto') {
            $model = getDolGlobalString('AI_API_CHATGPT_MODEL_AUDIO', 'tts-1');
          }
        }
      } elseif ($function == 'transcription') {
        if ($this->apiService == 'chatgpt') {
          $this->apiEndpoint = 'https://api.openai.com/v1/audio/transcriptions';
          if ($model == 'auto') {
            $model = getDolGlobalString('AI_API_CHATGPT_MODEL_TRANSCRIPT', 'whisper-1');
          }
        }
      } elseif ($function == 'translation') {
        if ($this->apiService == 'chatgpt') {
          $this->apiEndpoint = 'https://api.openai.com/v1/audio/translations';
          if ($model == 'auto') {
            $model = getDolGlobalString('AI_API_CHATGPT_MODEL_TRANSLATE', 'whisper-1');
          }
        }
      } else {  // else textgeneration...
        if ($this->apiService == 'groq') {
          $this->apiEndpoint = 'https://api.groq.com/openai/v1/chat/completions';
          if ($model == 'auto') {
            $model = getDolGlobalString('AI_API_GROK_MODEL_TEXT', 'mixtral-8x7b-32768');  // 'llama3-8b-8192', 'gemma-7b-it'
          }
        } elseif ($this->apiService == 'chatgpt') {
          $this->apiEndpoint = 'https://api.openai.com/v1/chat/completions';
          if ($model == 'auto') {
            $model = getDolGlobalString('AI_API_CHATGPT_MODEL_TEXT', 'gpt-3.5-turbo');
          }
        }
      }
    }
 
    dol_syslog("Call API for apiEndpoint=".$this->apiEndpoint." apiKey=".substr($this->apiKey, 0, 3).'***********, model='.$model);
 
    try {
      if (empty($this->apiEndpoint)) {
        throw new Exception('The AI service '.$this->apiService.' is not yet supported for the type of request '.$function);
      }
 
      $configurationsJson = getDolGlobalString('AI_CONFIGURATIONS_PROMPT');
      $configurations = json_decode($configurationsJson, true);
 
      $prePrompt = '';
      $postPrompt = '';
 
      if (isset($configurations[$function])) {
        if (isset($configurations[$function]['prePrompt'])) {
          $prePrompt = $configurations[$function]['prePrompt'];
        }
 
        if (isset($configurations[$function]['postPrompt'])) {
          $postPrompt = $configurations[$function]['postPrompt'];
        }
      }
      $fullInstructions = $prePrompt.' '.$instructions.' .'.$postPrompt;
 
 
      $payload = json_encode([
        'messages' => [
          ['role' => 'user', 'content' => $fullInstructions]
        ],
        'model' => $model
      ]);
 
      $headers = ([
        'Authorization: Bearer ' . $this->apiKey,
        'Content-Type: application/json'
      ]);
      $response = getURLContent($this->apiEndpoint, 'POST', $payload, 1, $headers);
 
      if (empty($response['http_code'])) {
        throw new Exception('API request failed. No http received');
      }
      if (!empty($response['http_code']) && $response['http_code'] != 200) {
        throw new Exception('API request failed with status code ' . $response['http_code']);
      }
      // Decode JSON response
      $decodedResponse = json_decode($response['content'], true);
 
      // Extraction content
      $generatedContent = $decodedResponse['choices'][0]['message']['content'];
 
      dol_syslog("generatedContent=".$generatedContent);
 
      // If content is not HTML, we convert it into HTML
      if ($format == 'html') {
        if (!dol_textishtml($generatedContent)) {
          dol_syslog("Result was detected as not HTML so we convert it into HTML.");
          $generatedContent = dol_nl2br($generatedContent);
        } else {
          dol_syslog("Result was detected as already HTML. Do nothing.");
        }
 
        // TODO If content is for website module, we must
        // - clan html header, keep body only and remove ``` ticks added by AI
        // - add tags <section contenEditable="true"> </section>
      }
 
      return $generatedContent;
    } catch (Exception $e) {
      $errormessage = $e->getMessage();
      if (!empty($response['content'])) {
        $decodedResponse = json_decode($response['content'], true);
 
        // With OpenAI, error is into an object error into the content
        if (!empty($decodedResponse['error']['message'])) {
          $errormessage .= ' - '.$decodedResponse['error']['message'];
        }
      }
 
      return array(
        'error' => true,
        'message' => $errormessage,
        'code' => (empty($response['http_code']) ? 0 : $response['http_code']),
        'curl_error_no' => (empty($response['curl_error_no']) ? $response['curl_error_no'] : ''),
        'format' => $format,
        'service' => $this->apiService,
        'function'=>$function
      );
    }
  }
}