mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-04 12:56:02 +00:00
40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
# 封装了function calling的一些支持函数
|
|
import logging
|
|
|
|
|
|
from pkg.plugin import host
|
|
|
|
|
|
class ContentFunctionNotFoundError(Exception):
|
|
pass
|
|
|
|
|
|
def get_func_schema_list() -> list:
|
|
"""从plugin包中的函数结构中获取并处理成受GPT支持的格式"""
|
|
if not host.__enable_content_functions__:
|
|
return []
|
|
|
|
schemas = host.__callable_functions__
|
|
|
|
return schemas
|
|
|
|
def get_func(name: str) -> callable:
|
|
if name not in host.__function_inst_map__:
|
|
raise ContentFunctionNotFoundError("没有找到内容函数: {}".format(name))
|
|
|
|
return host.__function_inst_map__[name]
|
|
|
|
def get_func_schema(name: str) -> dict:
|
|
for func in host.__callable_functions__:
|
|
if func['name'] == name:
|
|
return func
|
|
raise ContentFunctionNotFoundError("没有找到内容函数: {}".format(name))
|
|
|
|
def execute_function(name: str, kwargs: dict) -> any:
|
|
"""执行函数调用"""
|
|
|
|
logging.debug("executing function: name='{}', kwargs={}".format(name, kwargs))
|
|
|
|
func = get_func(name)
|
|
return func(**kwargs)
|