bstraehle commited on
Commit
96e76f7
·
verified ·
1 Parent(s): 10ea0e6

Update multi_agent.py

Browse files
Files changed (1) hide show
  1. multi_agent.py +7 -2
multi_agent.py CHANGED
@@ -50,6 +50,11 @@ def run_multi_agent(llm, message):
50
  max_turns=10
51
  )
52
 
 
 
 
 
 
53
  image_data = read_image_file("/home/user/app/coding/ytd_stock_gains.png")
54
  markdown_code_png = generate_markdown_image(image_data)
55
 
@@ -79,10 +84,10 @@ def run_multi_agent(llm, message):
79
  except Exception as e:
80
  print(f"An unexpected error occurred: {e}")
81
 
82
- result = markdown_code_png + "\n" + markdown_code_sh + "\n" + markdown_code_py
83
 
84
  print("===")
85
- print(chat_result)
86
  #print(result)
87
  print("===")
88
 
 
50
  max_turns=10
51
  )
52
 
53
+ chat_history = ""
54
+
55
+ for message in chat_result["messages"]:
56
+ chat_history += f"Role: {message['role']}, Content: {message['content']}\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)
60
 
 
84
  except Exception as e:
85
  print(f"An unexpected error occurred: {e}")
86
 
87
+ result = markdown_code_png + "\n" + markdown_code_sh + "\n" + markdown_code_py + "\n" + chat
88
 
89
  print("===")
90
+ print(chat_history)
91
  #print(result)
92
  print("===")
93