apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-05 13:52:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add async fn type to tool schema

Browse source
This commit is contained in:
yzlin 2024-02-06 15:49:14 +08:00
parent 3852e05195
commit 5abc5c3812
3 changed files with 36 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

13
metagpt/tools/tool_convert.py
View file

@ -15,18 +15,19 @@ def convert_code_to_tool_schema(obj, include: list[str] = []):
# method_doc = inspect.getdoc(method)
method_doc = get_class_method_docstring(obj, name)
if method_doc:
function_type = "function" if not inspect.iscoroutinefunction(method) else "async_function"
schema["methods"][name] = {"type": function_type, **docstring_to_schema(method_doc)}
schema["methods"][name] = function_docstring_to_schema(method, method_doc)
elif inspect.isfunction(obj):
schema = {
"type": "function",
**docstring_to_schema(docstring),
}
schema = function_docstring_to_schema(obj, docstring)
return schema
def function_docstring_to_schema(fn_obj, docstring):
function_type = "function" if not inspect.iscoroutinefunction(fn_obj) else "async_function"
return {"type": function_type, **docstring_to_schema(docstring)}
def docstring_to_schema(docstring: str):
if docstring is None:
return {}

22
tests/data/rsp_cache.json
View file

File diff suppressed because one or more lines are too long

21
tests/metagpt/tools/test_tool_convert.py
View file

@ -95,12 +95,26 @@ def dummy_fn(df: pd.DataFrame) -> dict:
pass
async def dummy_async_fn(df: pd.DataFrame) -> dict:
"""
A dummy async function for test
Args:
df (pd.DataFrame): test args.
Returns:
dict: test returns.
"""
pass
def test_convert_code_to_tool_schema_class():
expected = {
"type": "class",
"description": "Completing missing values with simple strategies.",
"methods": {
"__init__": {
"type": "function",
"description": "Initialize self.",
"parameters": {
"properties": {
@ -121,6 +135,7 @@ def test_convert_code_to_tool_schema_class():
},
},
"fit": {
"type": "function",
"description": "Fit the FillMissingValue model.",
"parameters": {
"properties": {"df": {"type": "pd.DataFrame", "description": "The input DataFrame."}},
@ -128,6 +143,7 @@ def test_convert_code_to_tool_schema_class():
},
},
"transform": {
"type": "function",
"description": "Transform the input DataFrame with the fitted model.",
"parameters": {
"properties": {"df": {"type": "pd.DataFrame", "description": "The input DataFrame."}},
@ -152,3 +168,8 @@ def test_convert_code_to_tool_schema_function():
}
schema = convert_code_to_tool_schema(dummy_fn)
assert schema == expected
def test_convert_code_to_tool_schema_async_function():
schema = convert_code_to_tool_schema(dummy_async_fn)
assert schema.get("type") == "async_function"