Change try another model

agent.py

@@ -1,42 +1,46 @@
 from huggingface_hub import InferenceClient
 import os
 
-client = InferenceClient(
-    model= "mistralai/Mistral-7B-Instruct-v0.1", # "HuggingFaceH4/zephyr-7b-beta",  # ou mistralai/Mistral-7B-Instruct-v0.1
-    token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
-)
+# Initialise le client avec le modèle open-source gratuit
+client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
 
-def build_email_agent():
-    return client
-
-def generate_email(client, to_email: str, context: str, tone: str) -> tuple[str, str]:
-    prompt = f"""Tu es un assistant virtuel. Rédige un email destiné à {to_email}.
+def generate_email(model, to: str, context: str, tone: str) -> tuple[str, str]:
+    """
+    Génère un objet et un corps d'email à partir du destinataire, du contexte et du ton.
+    """
+    prompt = f"""
+Tu es un assistant virtuel. Rédige un email destiné à {to}.
 
 Contexte : {context}
 Ton : {tone}
 
-Réponds de manière courte en deux parties :
+Réponds en deux parties :
 1. Objet : ligne d'objet de l'email
 2. Corps : contenu de l'email bien rédigé, en respectant le ton demandé.
 Commence chaque partie par son étiquette.
 """
 
-    response = client.text_generation(
-        prompt=prompt,
-        max_new_tokens=512,
+    response = client.chat_completion(
+        model="mistralai/Mistral-7B-Instruct-v0.1",
+        messages=[
+            {"role": "system", "content": "Tu es un assistant qui rédige des emails."},
+            {"role": "user", "content": prompt}
+        ],
         temperature=0.7,
-        do_sample=True,
-        return_full_text=False
+        max_tokens=800,
     )
 
-    print("🔍 Résultat brut :\n", response)
+    content = response.choices[0].message["content"]
+
+    # Extraire les deux parties de la réponse (objet + corps)
+    lines = content.strip().split("\n", 1)
+    subject = ""
+    body = ""
 
-    subject = "Sans objet"
-    lines = response.splitlines()
-    for i, line in enumerate(lines):
+    for line in lines:
         if line.lower().startswith("objet"):
             subject = line.split(":", 1)[-1].strip()
-            body = "\n".join(lines[i + 1:]).strip()
-            return subject, body
+        else:
+            body += line.strip() + "\n"
 
-    return subject, response.strip()
+    return subject, body.strip()