|
import inspect |
|
from typing import get_type_hints, List, Dict, Any |
|
|
|
|
|
def doc_to_dict(docstring): |
|
lines = docstring.split("\n") |
|
description = lines[1].strip() |
|
param_dict = {} |
|
|
|
for line in lines: |
|
if ":param" in line: |
|
line = line.replace(":param", "").strip() |
|
param, desc = line.split(":", 1) |
|
param_dict[param.strip()] = desc.strip() |
|
ret_dict = {"description": description, "params": param_dict} |
|
return ret_dict |
|
|
|
|
|
def get_tools_specs(tools) -> List[dict]: |
|
function_list = [ |
|
{"name": func, "function": getattr(tools, func)} |
|
for func in dir(tools) |
|
if callable(getattr(tools, func)) |
|
and not func.startswith("__") |
|
and not inspect.isclass(getattr(tools, func)) |
|
] |
|
|
|
specs = [] |
|
for function_item in function_list: |
|
function_name = function_item["name"] |
|
function = function_item["function"] |
|
|
|
function_doc = doc_to_dict(function.__doc__ or function_name) |
|
specs.append( |
|
{ |
|
"name": function_name, |
|
|
|
"description": function_doc.get("description", function_name), |
|
"parameters": { |
|
"type": "object", |
|
"properties": { |
|
param_name: { |
|
"type": param_annotation.__name__.lower(), |
|
**( |
|
{ |
|
"enum": ( |
|
str(param_annotation.__args__) |
|
if hasattr(param_annotation, "__args__") |
|
else None |
|
) |
|
} |
|
if hasattr(param_annotation, "__args__") |
|
else {} |
|
), |
|
"description": function_doc.get("params", {}).get( |
|
param_name, param_name |
|
), |
|
} |
|
for param_name, param_annotation in get_type_hints( |
|
function |
|
).items() |
|
if param_name != "return" |
|
and not ( |
|
param_name.startswith("__") and param_name.endswith("__") |
|
) |
|
}, |
|
"required": [ |
|
name |
|
for name, param in inspect.signature( |
|
function |
|
).parameters.items() |
|
if param.default is param.empty |
|
and not (name.startswith("__") and name.endswith("__")) |
|
], |
|
}, |
|
} |
|
) |
|
|
|
return specs |
|
|
