Update multi_agent.py
Browse files- multi_agent.py +4 -1
multi_agent.py
CHANGED
@@ -51,9 +51,12 @@ def run_multi_agent(llm, message):
|
|
51 |
)
|
52 |
|
53 |
chat = ""
|
|
|
54 |
|
55 |
for message in chat_result.chat_history:
|
56 |
-
|
|
|
|
|
57 |
|
58 |
image_data = read_image_file("/home/user/app/coding/ytd_stock_gains.png")
|
59 |
markdown_code_png = generate_markdown_image(image_data)
|
|
|
51 |
)
|
52 |
|
53 |
chat = ""
|
54 |
+
first_message = True
|
55 |
|
56 |
for message in chat_result.chat_history:
|
57 |
+
if not first_message:
|
58 |
+
chat += f"**{message['role'].replace('assistant', 'Code Executor').replace('user', 'Code Writer')}**\n{message['content']}\n\n"
|
59 |
+
first_message = False
|
60 |
|
61 |
image_data = read_image_file("/home/user/app/coding/ytd_stock_gains.png")
|
62 |
markdown_code_png = generate_markdown_image(image_data)
|