From 1857247cc3e34b3baf780420041b4dda2952eacd Mon Sep 17 00:00:00 2001 From: MAFTOUH Mohammed Amine <mohammed-amine.maftouh@imt-atlantique.net> Date: Fri, 28 Mar 2025 13:20:02 +0000 Subject: [PATCH] Add new file --- topic generation/generate_topics.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 topic generation/generate_topics.py diff --git a/topic generation/generate_topics.py b/topic generation/generate_topics.py new file mode 100644 index 0000000..a2baa70 --- /dev/null +++ b/topic generation/generate_topics.py @@ -0,0 +1,14 @@ +def generate_topics_for_cluster(messages_cluster): + prompt = f"Donne les principaux sujets ou thèmes pour les messages suivants :\n\n" + "\n".join(messages_cluster) + "\n\nQuels sont les topics principaux de ces messages ?" + # Ancienne syntaxe pour la version 0.29 + response = openai.ChatCompletion.create( + model="gpt-4o", # Vérifiez que ce modèle est disponible avec votre clé + messages=[ + {"role": "system", "content": "Vous êtes un assistant qui aide à générer des topics à partir de textes."}, + {"role": "user", "content": prompt} + ], + max_tokens=150, + temperature=0.7, + ) + topics = response['choices'][0]['message']['content'].strip() + return topics -- GitLab