roadmap 2.2 + add 单测用例

metagpt/actions/action_output.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+# coding: utf-8
+"""
+@Time    : 2023/7/11 10:03
+@Author  : chengmaoyu
+@File    : action_output
+"""
+
+from pydantic import create_model, validator, root_validator, BaseModel
+from typing import Dict, Type
+
+
+class ActionOutput:
+    content: str
+    instruct_content: BaseModel
+
+    def __init__(self, content: str, instruct_content: BaseModel):
+        self.content = content
+        self.instruct_content = instruct_content
+
+    @classmethod
+    def create_model_class(cls, class_name: str, mapping: Dict[str, Type]):
+        new_class = create_model(class_name, **mapping)
+
+        @validator('*', allow_reuse=True)
+        def check_name(v, field):
+            if field.name not in mapping.keys():
+                raise ValueError(f'Unrecognized block: {field.name}')
+            return v
+
+        @root_validator(pre=True, allow_reuse=True)
+        def check_missing_fields(values):
+            required_fields = set(mapping.keys())
+            missing_fields = required_fields - set(values.keys())
+            if missing_fields:
+                raise ValueError(f'Missing fields: {missing_fields}')
+            return values
+
+        new_class.__validator_check_name = classmethod(check_name)
+        new_class.__root_validator_check_missing_fields = classmethod(check_missing_fields)
+        return new_class