feat:add tools

src/do_tool/tool_can_use/fibonacci_sequence_tool.py

@@ -0,0 +1,56 @@
+from src.do_tool.tool_can_use.base_tool import BaseTool, register_tool
+from src.common.logger import get_module_logger
+from typing import Dict, Any
+
+logger = get_module_logger("fibonacci_sequence_tool")
+
+class FibonacciSequenceTool(BaseTool):
+    """生成斐波那契数列的工具"""
+    name = "fibonacci_sequence"
+    description = "生成指定长度的斐波那契数列"
+    parameters = {
+        "type": "object",
+        "properties": {
+            "n": {
+                "type": "integer",
+                "description": "斐波那契数列的长度",
+                "minimum": 1
+            }
+        },
+        "required": ["n"]
+    }
+    
+    async def execute(self, function_args: Dict[str, Any], message_txt: str = "") -> Dict[str, Any]:
+        """执行工具功能
+        
+        Args:
+            function_args: 工具参数
+            message_txt: 原始消息文本
+            
+        Returns:
+            Dict: 工具执行结果
+        """
+        try:
+            n = function_args.get("n")
+            if n <= 0:
+                raise ValueError("参数n必须大于0")
+            
+            sequence = []
+            a, b = 0, 1
+            for _ in range(n):
+                sequence.append(a)
+                a, b = b, a + b
+            
+            return {
+                "name": self.name,
+                "content": sequence
+            }
+        except Exception as e:
+            logger.error(f"fibonacci_sequence工具执行失败: {str(e)}")
+            return {
+                "name": self.name,
+                "content": f"执行失败: {str(e)}"
+            }
+
+# 注册工具
+register_tool(FibonacciSequenceTool)