fix format

This commit is contained in:
better629 2024-01-10 19:27:33 +08:00
parent 58c2c55ee9
commit 294b035fb4
2 changed files with 5 additions and 5 deletions

View file

@ -5,7 +5,6 @@
from functools import wraps
action_outcls_registry = dict()
@ -14,6 +13,7 @@ def register_action_outcls(func):
Due to `create_model` return different Class even they have same class name and mapping.
In order to do a comparison, use outcls_id to identify same Class with same class name and field definition
"""
@wraps(func)
def decorater(*args, **kwargs):
"""

View file

@ -3,6 +3,7 @@
# @Desc : unittest of action_outcls_registry
from typing import List
from metagpt.actions.action_node import ActionNode
@ -18,10 +19,9 @@ def test_action_outcls_registry():
outinst1 = outcls1(**out_data)
assert outinst1 == outinst
outcls2 = ActionNode(key="",
expected_type=str,
instruction="",
example="").create_model_class(class_name, out_mapping)
outcls2 = ActionNode(key="", expected_type=str, instruction="", example="").create_model_class(
class_name, out_mapping
)
outinst2 = outcls2(**out_data)
assert outinst2 == outinst