Răsfoiți Sursa

Enregistrer une seule fois les échanges image après une analyse réussie

Andy Esnard 1 zi în urmă
părinte
comite
7892a7b378
1 a modificat fișierele cu 1 adăugiri și 19 ștergeri
  1. 1 19
      chatbot.py

+ 1 - 19
chatbot.py

@@ -313,8 +313,6 @@ async def call_for_image_analysis(image_data, user_name, user_text=None):
     else:
         messages = [prompt] + [message_to_send]
 
-    await add_to_conversation_history(message_history)
-
     analysis = await openai_client.chat.completions.create(
         model=MODEL,
         stream=False,
@@ -327,6 +325,7 @@ async def call_for_image_analysis(image_data, user_name, user_text=None):
     else:
         logger.info(f"Analyse de l'image par l'API : {analysis.choices[0].message.content}")
 
+    await add_to_conversation_history(message_history)
     await add_to_conversation_history({
         "role": "assistant",
         "content": analysis.choices[0].message.content
@@ -604,23 +603,6 @@ async def on_message(message):
                     for part in split_message(reply):
                         await message.channel.send(part)
 
-                    if has_user_text:
-                        user_message_text = f"{message.author.name} a envoyé une image avec le messsage : \"{user_text}\"."
-                    else:
-                        user_message_text = f"{message.author.name} a envoyé une image."
-
-                    user_message = {
-                        "role": "user",
-                        "content": f"{user_message_text}"
-                    }
-
-                    assistant_message = {
-                        "role": "assistant",
-                        "content": analysis
-                    }
-
-                    await add_to_conversation_history(user_message)
-                    await add_to_conversation_history(assistant_message)
                 else:
                     await temp_msg.delete()
                     await message.channel.send("Désolé, je n'ai pas pu analyser l'image.")