bstraehle commited on
Commit
719b83b
1 Parent(s): 0b41fd1

Update multi_agent.py

Browse files
Files changed (1) hide show
  1. 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
- chat += f"**{message['role'].replace('assistant', 'Code Executor').replace('user', 'Code Writer')}**\n{message['content']}\n\n"
 
 
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)