Dolibarr\Code\Ai\Classes\Ai::generateContent()

Complexity

Conditions	33
Paths	16471

Size

Total Lines	136
Code Lines	82

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

<?php
    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);

The type Dolibarr\Code\Ai\Classes\Exception was not found. Did you mean Exception? If so, make sure to prefix the type with \.

            }

            $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
            );