Merge pull request #154 from wgzesg-bd/feat/builtin-tools

Feat/builtin tools

add calculator and link reader as built in tools

Author: wgzesg-bd

arkitect/core/component/asr/asr_client.py

@@ -90,7 +90,7 @@ def __init__(
         self.user = user
         self.model_name = model_name
 
-        self.conn: Optional[websockets.WebSocketClientProtocol] = None
+        self.conn: Optional[websockets.WebSocketClientProtocol] = None  # type: ignore
         self.session_id: Optional[str] = None
         self.inited = False
 

arkitect/core/component/tool/__init__.py

@@ -13,14 +13,17 @@
 # limitations under the License.
 
 from .builder import build_mcp_clients_from_config
+from .builtin_tools import calculator, link_reader
 from .mcp_client import MCPClient
-from .tool_pool import ToolPool, build_tool_pool
 from .mcp_server import ArkFastMCP
+from .tool_pool import ToolPool, build_tool_pool
 
 __all__ = [
     "MCPClient",
     "ToolPool",
     "build_tool_pool",
     "build_mcp_clients_from_config",
     "ArkFastMCP",
+    "link_reader",
+    "calculator",
 ]

arkitect/core/component/tool/builtin_tools.py

@@ -0,0 +1,56 @@
+# Copyright 2025 Bytedance Ltd. and/or its affiliates
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from pydantic import BaseModel
+
+from arkitect.core.client.http import default_ark_client
+
+
+async def link_reader(url_list: list[str]) -> dict:
+    """
+    当你需要获取网页、pdf、抖音视频内容时，使用此工具。可以获取url链接下的标题和内容。
+
+
+    examples: {"url_list":["abc.com", "xyz.com"]}
+    Args:
+        url_list (list[str]): 需要解析网页链接,最多3个,以列表返回
+    """
+    client = default_ark_client()
+    body = {
+        "action_name": "LinkReader",
+        "tool_name": "LinkReader",
+        "parameters": {"url_list": url_list},
+    }
+
+    response = await client.post(path="/tools/execute", body=body, cast_to=BaseModel)
+    return response
+
+
+async def calculator(input: str) -> dict:
+    """Evaluate a given mathematical expression
+
+    Args:
+    input(str): The mathematical expression in WolframLanguage InputForm
+
+
+    """
+    client = default_ark_client()
+    body = {
+        "action_name": "Calculator",
+        "tool_name": "Calculator",
+        "parameters": {"input": input},
+    }
+
+    response = await client.post(path="/tools/execute", body=body, cast_to=BaseModel)
+    return response

arkitect/core/component/tool/mcp_server.py

@@ -14,6 +14,7 @@
 
 from typing import Any, Literal, Sequence
 
+from arkitect.telemetry.trace import setup_tracing, task
 from mcp.server.fastmcp import FastMCP
 from mcp.types import (
     EmbeddedResource,
@@ -24,8 +25,6 @@
     Tool as MCPTool,
 )
 
-from arkitect.telemetry.trace import setup_tracing, task
-
 
 class ArkFastMCP(FastMCP):
     def __init__(self, *args, **kwargs):  # type: ignore

arkitect/core/component/tts/tts_client.py

@@ -61,7 +61,7 @@ def __init__(
         self.log_id = log_id
         self.base_url = base_url
 
-        self.conn: Optional[websockets.WebSocketClientProtocol] = None
+        self.conn: Optional[websockets.WebSocketClientProtocol] = None  # type: ignore
         self.session_id: Optional[str] = None
         self.connection_params: ConnectionParams = connection_params
         self.inited = False

examples/quick-start/agent.py renamed to examples/quick_start/agent.py

@@ -75,6 +75,7 @@ markers = [
 ]
 asyncio_mode = "auto"
 
+
 [dependency-groups]
 lint = [
     "ruff<1.0.0,>=0.1.5",