✨(ai) add backend for message generation

Author: MathisRouget

docs/env.md

@@ -276,6 +276,8 @@ The application uses a new environment file structure with `.defaults` and `.loc
 | `AI_API_KEY` | None| API Key used for AI features | Optional |
 | `AI_MODEL` | None | Default model used for AI features | Optional |
 | `AI_FEATURE_SUMMARY_ENABLED` | `False` | Default enabled mode for summary AI features | Required |
+| `AI_FEATURE_MESSAGES_GENERATION_ENABLED` | `False` | Default enabled mode for message generation AI features | Required |
+
 
 ### External Services
 

env.d/development/backend.defaults

@@ -79,6 +79,7 @@ AI_API_KEY=
 AI_MODEL=
 
 AI_FEATURE_SUMMARY_ENABLED=False
+AI_FEATURE_MESSAGES_GENERATION_ENABLED=False
 
 # Interoperability
 # Drive - https://github.com/suitenumerique/drive

src/backend/core/ai/ai_prompts.json

@@ -1,8 +1,12 @@
 {
     "en-us": {
-            "summary_query": "You are an intelligent assistant that summarizes email threads. You should summarize the content of the conversation in one or two lines maximum WITHOUT stating 'Summary:'. If important links appear in the emails, you must mention them clearly in Markdown format and integrate them into the summary. If there are no important links, the summary should not contain any information about links. \n Here is the conversation:\n\n{messages}\n\n. Summarize the above emails in Markdown in the language '{language}' :"
+            "summary_query": "You are an intelligent assistant that summarizes email threads. You should summarize the content of the conversation in one or two lines maximum WITHOUT stating 'Summary:'. If important links appear in the emails, you must mention them clearly in Markdown format and integrate them into the summary. If there are no important links, the summary should not contain any information about links. \n Here is the conversation:\n\n{messages}\n\n. Summarize the above emails in Markdown in the language '{language}' :",
+            "new_message_generation_query" : "You are an assistant integrated into the email writing area of an email client. THE EXPECTED RESPONSE IS AN EMAIL BODY IN THE LANGUAGE {language}. Your response should ONLY be the body of the message (do not mention the subject or recipients) DO NOT put quotes around your response. If your response contains fields to be filled by the user, MAKE THEM ALL IN BOLD MARKDOWN AND IN BRACKETS so that the user notices them (e.g., **[Your Name]**, **[Date]**, **[Signature]**, **[Number of Users]**). Here is an example of generation: 'User prompt: 'Request an API key for Mistral'. Your response: 'Hello,\nCould you please provide me with an API key for Mistral?\nBest regards,\n[Signature]' Here is the current email draft:\n\n{draft}\n\n. Here is the user prompt : {user_prompt}\nIf the draft does not contain a signature and only in this case, then sign with {name}. NEVER CHANGE ANYTHING ELSE IN THE DRAFT OTHER THAN WHAT THE USER REQUESTS!!!",
+            "thread_response_generation_query" : "You are an assistant integrated into the email writing area of an email client. THE EXPECTED RESPONSE IS AN EMAIL BODY IN THE LANGUAGE {language}. Your response should ONLY be the body of the message (do not mention the subject or recipients) DO NOT put quotes around your response. If your response contains fields to be filled by the user, MAKE THEM ALL IN BOLD MARKDOWN AND IN BRACKETS so that the user notices them (e.g., **[Your Name]**, **[Date]**, **[Signature]**, **[Number of Users]**). Here is an example of generation: 'User prompt: 'Request an API key for Mistral'. Previous messages: 'Hello Thomas,\nI recommend using Mistral for AI.\nBest regards,\nNicolas' User prompt: 'request the key'. Your response: Hello Nicolas,\nThank you for your response, could you please provide me with an API key for Mistral?\nBest regards,\nThomas' Your response: 'Hello,\nCould you please provide me with an API key for Mistral?\nBest regards,\n[Signature]' Here is the current email draft:\n\n{draft}\n\n. Here are the previous messages in the thread:\n\n{messages}\n\n. Here is the user prompt : {user_prompt}\nIf the draft does not contain a signature and only in this case, then sign with {name}. NEVER CHANGE ANYTHING ELSE IN THE DRAFT OTHER THAN WHAT THE USER REQUESTS!!!"
     },
     "fr-fr": {
-            "summary_query": "Tu es un assistant intelligent qui résume des boucles de mails. Tu dois résumer le contenu de la conversation en une ou deux lignes maximum SANS préciser 'Résumé:'. Si des liens importants apparaissent dans les emails, tu dois les mentionner dans le résumé de façon claire en Markdown et les intégrer au résumé. Si les liens ne sont pas importants ou qu'il n'y en a pas, le résumé ne doit pas contenir d'information à ce sujet ni même le mentionner. Voici la conversation:\n\n{messages}\n\n. Résumé des emails ci-dessus en Markdown dans la langue '{language}' :"
+            "summary_query": "Tu es un assistant intelligent qui résume des boucles de mails. Tu dois résumer le contenu de la conversation en une ou deux lignes maximum SANS préciser 'Résumé:'. Si des liens importants apparaissent dans les emails, tu dois les mentionner dans le résumé de façon claire en Markdown et les intégrer au résumé. Si les liens ne sont pas importants ou qu'il n'y en a pas, le résumé ne doit pas contenir d'information à ce sujet ni même le mentionner. Voici la conversation:\n\n{messages}\n\n. Résumé des emails ci-dessus en Markdown dans la langue '{language}' :",
+            "new_message_generation_query" : "Tu es un assistant intégré dans la zone d'écriture des mails dans une boîte mail. LA REPONSE ATTENDUE EST UN CORPS DE MAIL dans la langue {language}. Ta réponse doit être UNIQUEMENT le corps du message (ne mentionne pas l'objet ni les destinataires) NE mets PAS de guillemets autour de ta réponse. Si ta réponse contient des champs à remplir par l'utilisateur, METS LES TOUS EN GRAS MARKDOWN ET ENTRE CROCHETS pour que l'utilisateur le remarque. (ex : **[Votre nom]**, **[Date]**, **[Signature]**, **[Nombre d'utilisateurs]**). Voici un exemple de génération : 'Prompt de l'utilisateur : 'Demande une clé API pour Mistral'. Ta réponse : 'Bonjour,\nPourriez-vous me fournir une clé API pour Mistral ?\nBien à vous,\n[Signature]' Voici le brouillon de mail actuel :\n\n{draft}\n\n. Voici le prompt de l'utilisateur : {user_prompt}\nSi le brouillon ne contient pas de signature et uniquement dans ce cas, alors signe avec {name}. NE CHANGE JAMAIS RIEN D'AUTRE PAR RAPPORT AU BROUILLON QUE CE QUE L'UTILISATEUR TE DEMANDE !!!",
+            "thread_response_generation_query" : "Tu es un assistant intégré dans la zone d'écriture des mails dans une boîte mail. LA REPONSE ATTENDUE EST UN CORPS DE MAIL dans la langue {language}. Ta réponse doit être UNIQUEMENT le corps du message (ne mentionne pas l'objet ni les destinataires) NE mets PAS de guillemets autour de ta réponse. Si ta réponse contient des champs à remplir par l'utilisateur, METS LES TOUS EN GRAS MARKDOWN ET ENTRE CROCHETS pour que l'utilisateur le remarque. (ex : **[Votre nom]**, **[Date]**, **[Signature]**, **[Nombre d'utilisateurs]**). Voici un exemple de génération : 'Prompt de l'utilisateur : 'Demande une clé API pour Mistral'. Messages précédents: 'Bonjour Thomas,\nJe vous conseille d'utiliser Mistral pour l'IA.\nBien à vous,\nNicolas' Prompt de l'utilisateur : 'demande la clé'. Ta réponse : Bonjour Nicolas,\nMerci pour votre réponse, pourriez-vous me fournir une clé API pour Mistral ?\nBien à vous,\nThomas' Ta réponse : 'Bonjour,\nPourriez-vous me fournir une clé API pour Mistral ?\nBien à vous,\n[Signature]' Voici le brouillon de mail actuel :\n\n{draft}\n\n. Voici les anciens messages du thread : \n\n{messages}\n\n. Voici le prompt de l'utilisateur : {user_prompt}\nSi le brouillon ne contient pas de signature et uniquement dans ce cas, alors signe avec {name}. NE CHANGE JAMAIS RIEN D'AUTRE PAR RAPPORT AU BROUILLON QUE CE QUE L'UTILISATEUR TE DEMANDE !!!"
     }
 }

src/backend/core/ai/message_generator.py

@@ -0,0 +1,51 @@
+from django.conf import settings
+from django.utils import translation
+
+from core.ai.utils import get_messages_from_thread, load_ai_prompts
+from core.models import Thread
+from core.services.ai_service import AIService
+
+
+def generate_new_message(draft: str, user_prompt: str, name: str) -> str:
+    """Generates a new mail using the AI model base on user prompt."""
+
+    # Determine the active or fallback language
+    active_language = translation.get_language() or settings.LANGUAGE_CODE
+
+    # Get the prompt for the active language
+    prompts = load_ai_prompts()
+    prompt_template = prompts.get(active_language)
+    prompt_query = prompt_template["new_message_generation_query"]
+    prompt = prompt_query.format(
+        draft=draft, language=active_language, user_prompt=user_prompt, name=name
+    )
+
+    answer = AIService().call_ai_api(prompt)
+
+    return answer
+
+
+def generate_reply_message(draft: str, thread: Thread, user_prompt: str) -> str:
+    """Generates a reply message using the AI model based on the thread context and user prompt."""
+
+    # Determine the active or fallback language
+    active_language = translation.get_language() or settings.LANGUAGE_CODE
+
+    # Extract messages from the thread
+    messages = get_messages_from_thread(thread)
+    messages_as_text = "\n\n".join([message.get_as_text() for message in messages])
+
+    # Get the prompt for the active language
+    prompts = load_ai_prompts()
+    prompt_template = prompts.get(active_language)
+    prompt_query = prompt_template["reply_message_generation_query"]
+    prompt = prompt_query.format(
+        draft=draft,
+        messages=messages_as_text,
+        language=active_language,
+        user_prompt=user_prompt,
+    )
+
+    answer = AIService().call_ai_api(prompt)
+
+    return answer

src/backend/core/ai/thread_summarizer.py

@@ -1,10 +1,6 @@
-import json
-from pathlib import Path
-
-from django.conf import settings
 from django.utils import translation
 
-from core.ai.utils import get_messages_from_thread
+from core.ai.utils import get_active_language, get_messages_from_thread, load_ai_prompts
 from core.models import Thread
 from core.services.ai_service import AIService
 
@@ -13,18 +9,14 @@ def summarize_thread(thread: Thread) -> str:
     """Summarizes a thread using the OpenAI client based on the active Django language."""
 
     # Determine the active or fallback language
-    active_language = translation.get_language() or settings.LANGUAGE_CODE
+    active_language = get_active_language()
 
     # Extract messages from the thread
     messages = get_messages_from_thread(thread)
     messages_as_text = "\n\n".join([message.get_as_text() for message in messages])
 
-    # Load prompt templates from ai_prompts.json
-    prompts_path = Path(__file__).parent / "ai_prompts.json"
-    with open(prompts_path, encoding="utf-8") as f:
-        prompts = json.load(f)
-
     # Get the prompt for the active language
+    prompts = load_ai_prompts()
     prompt_template = prompts.get(active_language)
     prompt_query = prompt_template["summary_query"]
     prompt = prompt_query.format(messages=messages_as_text, language=active_language)

src/backend/core/ai/utils.py

@@ -1,10 +1,25 @@
+import json
+from pathlib import Path
 from typing import List
 
 from django.conf import settings
+from django.utils import translation
 
 from core.models import Message, Thread
 
 
+def get_active_language() -> str:
+    """Get the active language or fallback to the default language code."""
+    return translation.get_language() or settings.LANGUAGE_CODE
+
+
+def load_ai_prompts() -> dict:
+    """Load AI prompts from the ai_prompts.json file."""
+    prompts_path = Path(__file__).parent / "ai_prompts.json"
+    with open(prompts_path, encoding="utf-8") as f:
+        return json.load(f)
+
+
 def get_messages_from_thread(thread: Thread) -> List[Message]:
     """
     Extract messages from a thread and return them as a list of text representations using Message.get_as_text().
@@ -32,9 +47,19 @@ def is_ai_enabled() -> bool:
 
 def is_ai_summary_enabled() -> bool:
     """
-    Check if AI summary features are enabled.
+    Check if AI summary feature is enabled.
     This is determined by the presence of the AI settings and if AI_FEATURE_SUMMARY_ENABLED is set to 1.
     """
     return all(
         [is_ai_enabled(), getattr(settings, "AI_FEATURE_SUMMARY_ENABLED", False)]
     )
+
+
+def is_ai_messages_generation_enabled() -> bool:
+    """
+    Check if AI messages generation feature is enabled.
+    This is determined by the presence of the AI settings and if AI_FEATURE_MESSAGES_GENERATION_ENABLED is set to 1.
+    """
+    return all(
+        [is_ai_enabled(), getattr(settings, "AI_FEATURE_MESSAGES_GENERATION_ENABLED", False)]
+    )