wehub-resource-sync
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
"""
|
|
Registry of client-side tool sets.
|
|
|
|
Each tool set is a module in this package that exports:
|
|
|
|
- ``TOOLS``: List of OpenAI-format tool schema dicts.
|
|
- ``execute_tool(name, arguments) -> str``: Local execution function.
|
|
|
|
Usage::
|
|
|
|
from omnigent.client_tools import get_tool_set
|
|
tool_set = get_tool_set("coding")
|
|
schemas = tool_set.TOOLS
|
|
result = tool_set.execute_tool("Read", {"file_path": "/tmp/foo.py"})
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import importlib
|
|
from types import ModuleType
|
|
|
|
|
|
def get_tool_set(name: str) -> ModuleType:
|
|
"""
|
|
Load a tool set module by name.
|
|
|
|
:param name: Tool set name, e.g. ``"coding"``. Must correspond
|
|
to a module in this package
|
|
(``omnigent.client_tools.coding``).
|
|
:returns: The tool set module with ``TOOLS`` and ``execute_tool``.
|
|
:raises SystemExit: If the tool set is not found.
|
|
"""
|
|
try:
|
|
return importlib.import_module(f".{name}", package=__name__)
|
|
except ModuleNotFoundError:
|
|
import sys
|
|
|
|
print(f"Error: unknown tool set {name!r}")
|
|
print("Available tool sets: coding")
|
|
sys.exit(1)
|