Intlayer

import { readAsset } from 'utils:asset'; import { type AIConfig, type AIOptions, AIProvider, generateText, } from '@intlayer/ai'; import { logger } from '@logger'; export type AutocompleteOptions = { text: string; aiConfig: AIConfig; applicationContext?: string; contextBefore?: string; currentLine?: string; contextAfter?: string; }; export type AutocompleteFileResultData = { autocompletion: string; tokenUsed: number; }; // The prompt template to send to the AI model const CHAT_GPT_PROMPT = readAsset('./PROMPT.md'); export const aiDefaultOptions: AIOptions = { provider: AIProvider.OPENAI, model: 'gpt-4o-mini', temperature: 0.7, }; /** * Autocompletes a content declaration file by constructing a prompt for AI models. * The prompt includes details about the project's locales, file paths of content declarations, * and requests for identifying issues or inconsistencies. */ export const autocomplete = async ({ text, aiConfig, applicationContext, contextBefore, currentLine, contextAfter, }: AutocompleteOptions): Promise<AutocompleteFileResultData | undefined> => { // Prepare the prompt for AI by replacing placeholders with actual values. const prompt = CHAT_GPT_PROMPT.replace( '{{applicationContext}}', applicationContext ?? '' ) .replace('{{contextBefore}}', contextBefore ?? '') .replace('{{currentLine}}', currentLine ?? '') .replace('{{contextAfter}}', contextAfter ?? ''); // Use the AI SDK to generate the completion const { text: newContent, usage } = await generateText({ ...aiConfig, messages: [ { role: 'system', content: prompt }, { role: 'assistant', content: text }, ], }); logger.info(`${usage?.totalTokens ?? 0} tokens used in the request`); return { autocompletion: newContent, tokenUsed: usage?.totalTokens ?? 0, }; };