tool_interface.py

from abc import ABC, abstractmethod

class BaseToolkit(ABC):
    @abstractmethod
    def get_tools(self):
        """获取所有可用工具"""
        pass
    
    @abstractmethod
    def validate_tool(self, tool_name):
        """验证工具是否可用"""
        pass
    
    @abstractmethod
    def execute_tool(self, tool_name, params):
        """执行指定工具"""
        pass

class CustomToolkit(BaseToolkit):
    def __init__(self):
        self.tools = {
            # 示例工具
            "search": {
                "description": "搜索知识库",
                "params": {"query": "搜索关键词"}
            },
            "calculate": {
                "description": "执行计算",
                "params": {"expression": "数学表达式"}
            }
        }
    
    def get_tools(self):
        return self.tools
    
    def validate_tool(self, tool_name):
        return tool_name in self.tools
    
    def execute_tool(self, tool_name, params):
        if not self.validate_tool(tool_name):
            raise ValueError(f"工具{tool_name}不存在")
        
        # 实际工具执行逻辑
        if tool_name == "search":
            return self._search(params["query"])
        elif tool_name == "calculate":
            return self._calculate(params["expression"])
    
    def _search(self, query):
        # 实现搜索逻辑
        return {"result": f"搜索结果: {query}"}
    
    def _calculate(self, expression):
        # 实现计算逻辑
        try:
            from ast import literal_eval
            return {"result": str(literal_eval(expression))}
        except (ValueError, SyntaxError) as e:
            return {"error": f"计算错误: {str(e)}"}