File size: 3,023 Bytes
8e11aba cac4e7b 4d969ce 9ea371a 6141f1b 9ea371a 6141f1b 0e9c24d 5bcc2da 9ea371a 1923ab6 722929d ed306c8 4d969ce ed306c8 cac4e7b 4d969ce d8a9c24 cac4e7b ed306c8 cac4e7b 4d969ce 722929d 4d969ce 722929d d56f6b1 05c2b03 b1c069d 6141f1b 9ea371a d8f2386 da61fc4 9ea371a da61fc4 24dad07 11d87fa da61fc4 e90039b 24dad07 e90039b e5c6880 39a08ef bcef287 f4baca0 bcef287 e5c6880 39a08ef 576e7cf c65ef57 10ea0e6 c65ef57 cac4e7b 576e7cf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 |
import base64, json, os
from autogen import ConversableAgent, AssistantAgent
from autogen.coding import LocalCommandLineCodeExecutor
def read_file(file_path: str) -> str:
with open(file_path, "r", encoding="utf-8") as file:
return file.read()
def read_image_file(image_file_path: str) -> str:
with open(image_file_path, "rb") as image_file:
image_data = image_file.read()
return base64.b64encode(image_data).decode("utf-8")
def generate_markdown_image(image_data: str) -> str:
return f""
def format_as_markdown(code: str) -> str:
markdown_code = '```\n'
markdown_code += code
markdown_code += '\n```'
return markdown_code
def run_multi_agent(llm, message):
llm_config = {"model": llm}
executor = LocalCommandLineCodeExecutor(
timeout=60,
work_dir="coding",
)
code_executor_agent = ConversableAgent(
name="code_executor_agent",
llm_config=False,
code_execution_config={"executor": executor},
human_input_mode="NEVER",
default_auto_reply="Please continue. If everything is done, reply 'TERMINATE'.",
)
code_writer_agent = AssistantAgent(
name="code_writer_agent",
llm_config=llm_config,
code_execution_config=False,
human_input_mode="NEVER",
)
chat_result = code_executor_agent.initiate_chat(
code_writer_agent,
message=message,
max_turns=10
)
image_data = read_image_file("/home/user/app/coding/ytd_stock_gains.png")
markdown_code_png = generate_markdown_image(image_data)
file_name_py = ""
file_name_sh = ""
for file in os.listdir("coding"):
if file:
_, file_extension = os.path.splitext(file)
if file_extension == ".py":
file_name_py = file
if file_extension == ".sh":
file_name_sh = file
try:
file_path_py = "coding/" + file_name_py
code_py = read_file(file_path_py)
markdown_code_py = format_as_markdown(code_py)
file_path_sh = "coding/" + file_name_sh
code_sh = read_file(file_path_sh)
markdown_code_sh = format_as_markdown(code_sh)
except FileNotFoundError:
print(f"Error: File '{file_path_sh}' not found.")
except IOError as e:
print(f"Error reading file '{file_path_sh}': {e.strerror}")
except Exception as e:
print(f"An unexpected error occurred: {e}")
#
#print("===")
#print(chat_result)
#print("===")
#print(chat_result.chat_history)
#print("===")
#print(chat_result.chat_history[0])
#print("===")
chat = ""
for message in chat_result.chat_history:
chat += f"{message['role']}\n{message['content']}\n"
print("===")
#
result = f"{markdown_code_png}\n{markdown_code_sh}\n{markdown_code_py}\n{chat}"
#print("===")
#print(result)
#print("===")
return result |