refactor action_output and action_node

This commit is contained in:
geekan 2023-12-19 21:24:08 +08:00 committed by better629
parent a75ab7971f
commit d159bfc4e1
7 changed files with 18 additions and 39 deletions

View file

@ -5,7 +5,12 @@
import copy
import pickle
<<<<<<< HEAD
from metagpt.utils.common import import_class
=======
from metagpt.actions.action_node import ActionNode
from metagpt.schema import Message
>>>>>>> 09e2f05 (refactor action_output and action_node)
def actionoutout_schema_to_mapping(schema: dict) -> dict:
@ -104,13 +109,11 @@ def deserialize_general_message(message_dict: dict) -> "Message":
return message
def deserialize_message(message_ser: str) -> "Message":
def deserialize_message(message_ser: str) -> Message:
message = pickle.loads(message_ser)
if message.instruct_content:
ic = message.instruct_content
actionoutput_class = import_class("ActionOutput", "metagpt.actions.action_output")
ic_obj = actionoutput_class.create_model_class(class_name=ic["class"], mapping=ic["mapping"])
ic_obj = ActionNode.create_model_class(class_name=ic["class"], mapping=ic["mapping"])
ic_new = ic_obj(**ic["value"])
message.instruct_content = ic_new