githubmirror/AgentPilot

src/system/tools.py

19 строк412 B

jbexta

prep

04 мар 2024, 07:53

04 мар 2024, 07:535d8394b

import json
from src.utils import sql


class ToolManager:
    def __init__(self):
        self.tools = {}

    def load(self):
        self.tools = sql.get_results("""
            SELECT
                name,
                config
            FROM tools""", return_type='dict')
        for k, v in self.tools.items():
            self.tools[k] = json.loads(v)

    def to_dict(self):
        return self.tools