feat: init repo

Author: zihengli-bytedance

.gitignore

@@ -0,0 +1,5 @@
+*config.yaml
+__pycache__/
+.idea
+.DS_Store
+.venv/

LICENSE

@@ -0,0 +1,24 @@
+MIT License
+
+Copyright (c) Bytedance, Inc. and affiliates.
+
+Permission is hereby granted, free of charge, to any person
+obtaining a copy of this software and associated documentation
+files (the "Software"), to deal in the Software without
+restriction, including without limitation the rights to use,
+copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the
+Software is furnished to do so, subject to the following
+conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
+HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+OTHER DEALINGS IN THE SOFTWARE.

README.md

@@ -0,0 +1,11 @@
+# Model Context Protocol servers and demo agents
+火山引擎 agent / mcp server 共享仓库
+
+## Ref
+### mcp client
+- cline: 支持方舟 豆包/deepseek
+### mcp protocol
+- [Typescript MCP SDK](https://github.com/modelcontextprotocol/typescript-sdk)
+- [Python MCP SDK](https://github.com/modelcontextprotocol/python-sdk)
+- [MCP intro](https://modelcontextprotocol.io/introduction)
+

app/general_client/README.md

@@ -0,0 +1,38 @@
+# General MCP client
+
+## Introduction
+
+A basic implementation of a MCP client working with one mcp server
+
+## Prerequisites
+
+Python >=3.10,<3.12
+
+```
+pip install -r requirement.txt
+```
+
+## Quick Start
+
+1. start a calculator MCP server
+
+```bash
+ARK_API_KEY=<YOUR API KEY> ARK_TOOL_CACULATOR=true PORT=8765 uvx --from git+https://github.com/volcengine/mcp-server.git#subdirectory=server/mcp_server_ark mcp-server-ark --transport sse
+```
+
+2. set up environment variable
+
+   ``` bash
+   export ARK_API_KEY=<YOUR API KEY>
+   ```
+
+3. start your client
+
+   ```bash
+   python3 client.py
+   ```
+
+4. run test query and see your output
+   1. example query to trigger calculator: 计算12341234/2341234+1123
+
+![alt text](assets/image.png)

app/general_client/assets/image.png

@@ -0,0 +1,102 @@
+from typing import Any, Optional
+from arkitect.core.component.context.context import Context
+from arkitect.core.component.context.model import State
+from arkitect.core.component.tool.builder import build_mcp_clients_from_config
+
+from arkitect.core.component.context.hooks import (
+    PreToolCallHook,
+    PostToolCallHook,
+)
+
+CONFIG_FILE_PATH = "./mcp_config.json"
+
+
+class MyHooks(PreToolCallHook, PostToolCallHook):
+    async def pre_tool_call(
+        self,
+        name: str,
+        arguments: str,
+        state: State,
+    ) -> State:
+        print("\n" + "=" * 20 + "Inside pre tool call" + "=" * 20 + "\n")
+        last_assistant_message = state.messages[-1]
+        tool_call_part = last_assistant_message["tool_calls"]
+        for tool_call in tool_call_part:
+            print(
+                f"Tool {tool_call['function']['name']} with {tool_call['function']['arguments']}"
+            )
+            # you may modify this or ask users for approval here
+        return state  # return state no matter if have modified it
+
+    async def post_tool_call(
+        self,
+        name: str,
+        arguments: str,
+        response: Any,
+        exception: Optional[Exception],
+        state: State,
+    ) -> State:
+        print("\n" + "=" * 20 + "Inside post tool call" + "=" * 20 + "\n")
+        print(f"Tool {name} with {arguments} returned {response}")
+        return state  # return state no matter if have modified it
+
+
+class Agent:
+    def __init__(self, mcp_config_file: str):
+        # Initialize session and client objects
+        self.mcp_config_file = mcp_config_file
+
+    async def process_query(self, query: str) -> str:
+        mcp_clients, cleanup = build_mcp_clients_from_config(self.mcp_config_file)
+        messages = [
+            {
+                "role": "user",
+                "content": query,
+            }
+        ]
+
+        # Initialize LLM
+        ctx = Context(
+            model="deepseek-v3-241226",
+            tools=list(
+                mcp_clients.values()
+            ),  # 直接在这个list里传入你的所有的python方法或者MCPClient，可以混着传入
+        )
+        my_hook = MyHooks()
+        ctx.set_pre_tool_call_hook(my_hook)
+        ctx.set_post_tool_call_hook(my_hook)
+        await ctx.init()
+
+        completion = await ctx.completions.create(messages, stream=True)
+        async for chunk in completion:
+            if chunk.choices:
+                print(chunk.choices[0].delta.content, end="")
+        await cleanup()  # 注意cleanup！！！
+
+    async def chat_loop(self):
+        """Run an interactive chat loop"""
+        print("\nMCP Client Started!")
+        print("Type your queries or 'quit' to exit.")
+
+        while True:
+            try:
+                query = input("\nQuery: ").strip()
+
+                if query.lower() == "quit":
+                    break
+
+                await self.process_query(query)
+
+            except Exception as e:
+                print(f"\nError: {str(e)}")
+
+
+async def main():
+    agent = Agent(mcp_config_file=CONFIG_FILE_PATH)
+    await agent.chat_loop()
+
+
+if __name__ == "__main__":
+    import asyncio
+
+    asyncio.run(main())

app/general_client/client.py

@@ -0,0 +1,7 @@
+{
+  "mcpServers": {
+    "calculator": {
+      "url": "http://localhost:8765/sse"
+    }
+  }
+}

app/general_client/mcp_config.json

@@ -0,0 +1 @@
+git+https://github.com/luminghao-bytedance/ai-app-lab.git@feat/deepresearch_agent