first update
This commit is contained in:
178
backend/generate_group.py
Normal file
178
backend/generate_group.py
Normal file
@@ -0,0 +1,178 @@
|
||||
from autogen.code_utils import create_virtual_env
|
||||
from autogen.coding import LocalCommandLineCodeExecutor
|
||||
from autogen.agentchat.contrib.capabilities.teachability import Teachability
|
||||
from autogen.agentchat.contrib.capabilities.vision_capability import VisionCapability
|
||||
from autogen.agentchat.contrib.multimodal_conversable_agent import MultimodalConversableAgent
|
||||
from pathlib import Path
|
||||
import autogen
|
||||
from .utils import load_agent_configs
|
||||
import os
|
||||
from .retrieval_group import init_retrieval_group
|
||||
from .constant import config_list, STREAM, SILENT, WORK_DIR
|
||||
|
||||
|
||||
|
||||
agent_configs = load_agent_configs(os.path.join(os.path.dirname(os.path.abspath(__file__)), "config/generate_group.yaml"))
|
||||
venv_context = create_virtual_env(WORK_DIR)
|
||||
llm_config = {"config_list": config_list, "stream": STREAM}
|
||||
|
||||
|
||||
def init_generate_group(outer_retrieval_agent, inner_retrieval_admin):
|
||||
|
||||
outer_generate_agent = autogen.ConversableAgent(
|
||||
name="Outer_Generate_Admin",
|
||||
human_input_mode="NEVER",
|
||||
# human_input_mode="TERMINATE",
|
||||
code_execution_config={
|
||||
"work_dir": WORK_DIR,
|
||||
"use_docker": False,
|
||||
},
|
||||
is_termination_msg=lambda x: x.get("content", "").find("TERMINATE") >= 0,
|
||||
description="Outer_Generate_Admin",
|
||||
default_auto_reply="continue",
|
||||
max_consecutive_auto_reply=1
|
||||
)
|
||||
|
||||
|
||||
structure_scientist_name = "structure_scientist"
|
||||
structure_scientist = MultimodalConversableAgent(
|
||||
name=structure_scientist_name,
|
||||
system_message=agent_configs[structure_scientist_name]['system_message'],
|
||||
llm_config=llm_config,
|
||||
is_termination_msg=lambda x: x.get("content", "")[-1].get("text", "").find("TERMINATE") >= 0,
|
||||
human_input_mode="TERMINATE",
|
||||
description="structure_scientist",
|
||||
max_consecutive_auto_reply=1
|
||||
)
|
||||
|
||||
|
||||
property_scientist_name = "property_scientist"
|
||||
property_scientist = MultimodalConversableAgent(
|
||||
name=property_scientist_name,
|
||||
system_message=agent_configs[property_scientist_name]['system_message'],
|
||||
llm_config=llm_config,
|
||||
is_termination_msg=lambda x: x.get("content", "")[-1].get("text", "").find("TERMINATE") >= 0,
|
||||
human_input_mode="TERMINATE",
|
||||
description="property_scientist",
|
||||
max_consecutive_auto_reply=1
|
||||
)
|
||||
|
||||
|
||||
application_scientist_name = "application_scientist"
|
||||
application_scientist = MultimodalConversableAgent(
|
||||
name=application_scientist_name,
|
||||
system_message=agent_configs[application_scientist_name]['system_message'],
|
||||
llm_config=llm_config,
|
||||
is_termination_msg=lambda x: x.get("content", "")[-1].get("text", "").find("TERMINATE") >= 0,
|
||||
human_input_mode="TERMINATE",
|
||||
description="application_scientist",
|
||||
max_consecutive_auto_reply=1
|
||||
)
|
||||
|
||||
|
||||
synthesis_scientist_name = "synthesis_scientist"
|
||||
synthesis_scientist = autogen.AssistantAgent(
|
||||
name=synthesis_scientist_name,
|
||||
system_message=agent_configs[synthesis_scientist_name]['system_message'],
|
||||
llm_config=llm_config,
|
||||
is_termination_msg=lambda x: x.get("content", "").find("TERMINATE") >= 0,
|
||||
human_input_mode="TERMINATE",
|
||||
description="synthesis_scientist",
|
||||
max_consecutive_auto_reply=3
|
||||
)
|
||||
|
||||
|
||||
|
||||
scheme_critic_name = "scheme_critic"
|
||||
scheme_critic = autogen.AssistantAgent(
|
||||
name=scheme_critic_name,
|
||||
system_message=agent_configs[scheme_critic_name]['system_message'],
|
||||
llm_config=llm_config,
|
||||
# is_termination_msg=lambda x: x.get("content", "").find("TERMINATE") >= 0,
|
||||
human_input_mode="TERMINATE",
|
||||
description="scheme_critic",
|
||||
max_consecutive_auto_reply=3
|
||||
)
|
||||
|
||||
def state_transition(last_speaker, groupchat):
|
||||
messages = groupchat.messages
|
||||
|
||||
if last_speaker is outer_generate_agent:
|
||||
# user -> retrieve
|
||||
if len(messages) <= 1:
|
||||
return outer_retrieval_agent
|
||||
else:
|
||||
return "auto"
|
||||
elif last_speaker is outer_retrieval_agent:
|
||||
# retrieve -> auto select scientist
|
||||
return "auto"
|
||||
|
||||
# 方案问题
|
||||
elif last_speaker is synthesis_scientist:
|
||||
return scheme_critic
|
||||
elif last_speaker is scheme_critic:
|
||||
groupchat.messages[-1]["content"] += "\n\nSYNTHESIS"
|
||||
if "OPTIMIZE" in messages[-1]["content"]:
|
||||
return synthesis_scientist
|
||||
elif ("TERMINATE" not in messages[-1]["content"] and "OPTIMIZE" not in messages[-1]["content"]):
|
||||
groupchat.messages[-1]["content"] += "\n\nTERMINATE"
|
||||
return outer_generate_agent
|
||||
else:
|
||||
return outer_generate_agent
|
||||
|
||||
else:
|
||||
if ("TERMINATE" not in messages[-1]["content"] and "OPTIMIZE" not in messages[-1]["content"]):
|
||||
messages[-1]["content"] += "\n\nTERMINATE"
|
||||
return outer_generate_agent
|
||||
|
||||
|
||||
generate_group = autogen.GroupChat(
|
||||
agents=[outer_generate_agent, outer_retrieval_agent, structure_scientist, property_scientist, application_scientist, synthesis_scientist, scheme_critic],
|
||||
messages=[],
|
||||
speaker_selection_method=state_transition, # custom speaker selection method
|
||||
max_round=10,
|
||||
)
|
||||
|
||||
|
||||
inner_generate_admin = autogen.GroupChatManager(
|
||||
name="Generate_Group_Admin",
|
||||
description="Generate_Group_Admin",
|
||||
groupchat=generate_group,
|
||||
# is_termination_msg=lambda x: x.get("content", "").find("TERMINATE") >= 0,
|
||||
llm_config=llm_config,
|
||||
system_message=agent_configs["admin"]['system_message']
|
||||
)
|
||||
|
||||
|
||||
outer_retrieval_agent.register_nested_chats(
|
||||
[
|
||||
{"recipient": inner_retrieval_admin, "max_turn": 1, "summary_method": "last_msg", "silent": SILENT},
|
||||
],
|
||||
trigger=inner_generate_admin,
|
||||
)
|
||||
|
||||
return inner_generate_admin, outer_generate_agent
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
outer_retrieval_agent, inner_retrieval_admin = init_retrieval_group(WORK_DIR, venv_context)
|
||||
inner_generate_admin, outer_generate_agent = init_generate_group(outer_retrieval_agent, inner_retrieval_admin)
|
||||
|
||||
outer_generate_agent.initiate_chat(
|
||||
inner_generate_admin,
|
||||
# message="如何在常温条件下制备CsPbBr3纳米立方体",
|
||||
# message="how to synthesis of CsPbBr3 Perovskite NCs at room temperature?"
|
||||
# message="how to synthesis CsPbBr3 nanocubes at room temperature?"
|
||||
# message="什么是钙钛矿？"
|
||||
message="钙钛矿有什么作用？"
|
||||
# message="Please prepare few layers graphene from graphite powder.",
|
||||
# message="Can you please prepare black phosphorusene with improved stability from black phosphorus crystals powder?",
|
||||
# message="Can you synthesize gold nanorods by seed-mediated method with absorption peaks at 820 nm?",
|
||||
# message="Please synthesize CsPbBr3 nanocubes with a fluorescence emission wavelength of 520 nm at room temperature?",
|
||||
# message="Please design a new hybridized halide perovskite composite material that is biocompatible and water-stable",
|
||||
# message="please use phospholipid membrane as shell to encapsulate hybrid perovskite"
|
||||
# Now I want a novel bright perovskite composite based CH3NH3PbBr3 and phospholipid membrane(PM) to improve the stability and biocompatibility, please synthesis this materials under room temperature
|
||||
)
|
||||
Reference in New Issue
Block a user